kallithea Changeset - 7d23f2491ce8

Changeset - 7d23f2491ce8

Parent rev.

Child rev.

[Not reviewed]

default

0 1 0

Konstantin Veretennicov - 10 years ago 2016-05-01 23:29:09
kveretennicov@gmail.com

tests: add set_test_settings fixture

Uses https://pytest.org/latest/yieldfixture.html#yieldfixture to restore
settings after tests. Takes into account current specifics of list-valued
settings.

1 file changed with 35 insertions and 0 deletions:

kallithea/tests/conftest.py

0 comments (0 inline, 0 general)

kallithea/tests/conftest.py

➞

Show inline comments

@@ @@ -6,12 +6,13 @@ import pkg_resources @@
 from paste.deploy import loadapp
 import pylons.test
 from pylons.i18n.translation import _get_translator
 import pytest
 from kallithea.model.user import UserModel
 from kallithea.model.meta import Session
 from kallithea.model.db import Setting
 def pytest_configure():
     path = os.getcwd()
     sys.path.insert(0, path)
     pkg_resources.working_set.add_entry(path)
@@ @@ -46,6 +47,40 @@ def create_test_user(): @@
         test_users.append(user)
         return user
     yield _create_test_user
     for user in test_users:
         UserModel().delete(user)
     Session().commit()
 def _set_settings(*kvtseq):
     session = Session()
     for kvt in kvtseq:
         assert len(kvt) in (2, 3)
         k = kvt[0]
         v = kvt[1]
         t = kvt[2] if len(kvt) == 3 else 'unicode'
         setting = Setting.create_or_update(k, v, t)
         session.add(setting)
     session.flush()
 @pytest.yield_fixture
 def set_test_settings():
     """Restore settings after test is over."""
     # Save settings.
     settings_snapshot = [
         (s.app_settings_name, s.app_settings_value, s.app_settings_type)
         for s in Setting.query().all()]
     yield _set_settings
     # Restore settings.
     session = Session()
     keys = frozenset(k for (k, v, t) in settings_snapshot)
     for s in Setting.query().all():
         if s.app_settings_name not in keys:
             session.delete(s)
     for k, v, t in settings_snapshot:
         if t == 'list' and hasattr(v, '__iter__'):
             v = ','.join(v) # Quirk: must format list value manually.
         s = Setting.create_or_update(k, v, t)
         session.add(s)
     session.commit()

0 comments (0 inline, 0 general)