[1767][tests] bug fix: only use custom select class for TestWUI

ckan · Feb 15, 2012 · 66993a2 · 66993a2
1 parent d4c5870
commit 66993a2
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/ckan/tests/functional/test_tag_vocab.py b/ckan/tests/functional/test_tag_vocab.py
@@ -131,7 +131,6 @@ def value__get(self):
 
     value = property(value__get, value__set)
 
-paste.fixture.Field.classes['select'] = Select
 
 
 class TestWUI(WsgiAppCase):
@@ -152,6 +151,10 @@ def setup_class(cls):
         cls.old_default_controller = package_controller._default_controller_behaviour
         package_controller._default_controller_behaviour = cls.plugin
 
+        # use our custom select class for this test suite
+        cls.old_select = paste.fixture.Field.classes['select']
+        paste.fixture.Field.classes['select'] = Select
+
         # create a test vocab
         params = json.dumps({'name': TEST_VOCAB_NAME})
         extra_environ = {'Authorization' : str(cls.sysadmin_user.apikey)}
@@ -174,9 +177,10 @@ def setup_class(cls):
     @classmethod
     def teardown_class(cls):
         plugins.unload(cls.plugin)
-        model.repo.rebuild_db()
         from ckan.controllers import package as package_controller
         package_controller._default_controller_behaviour = cls.old_default_controller
+        paste.fixture.Field.classes['select'] = cls.old_select
+        model.repo.rebuild_db()
 
     def _get_vocab_id(self, vocab_name):
         params = json.dumps({'name': vocab_name})