diff --git a/ckan/new_tests/controllers/test_package.py b/ckan/new_tests/controllers/test_package.py index b05bb202008..f534a33bcee 100644 --- a/ckan/new_tests/controllers/test_package.py +++ b/ckan/new_tests/controllers/test_package.py @@ -2,13 +2,13 @@ from routes import url_for -from ckan.new_tests.controllers import WsgiAppCase -from ckan.new_tests.factories import Sysadmin +import ckan.new_tests.controllers as controllers +import ckan.new_tests.factories as factories -class TestPackageController(WsgiAppCase): +class TestPackageController(controllers.WsgiAppCase): def test_create_form_renders(self): - user = Sysadmin() + user = factories.Sysadmin() extra_environ = { 'REMOTE_USER': user['name'].encode('ascii'), } diff --git a/ckan/new_tests/controllers/test_util.py b/ckan/new_tests/controllers/test_util.py index 2f3bdb0ccb3..458cf05d8d7 100644 --- a/ckan/new_tests/controllers/test_util.py +++ b/ckan/new_tests/controllers/test_util.py @@ -4,10 +4,10 @@ from routes import url_for as url_for -from ckan.new_tests.controllers import WsgiAppCase +import ckan.new_tests.controllers as controllers -class TestUtil(WsgiAppCase): +class TestUtil(controllers.WsgiAppCase): def test_redirect_ok(self): response = self.app.get( url=url_for(controller='util', action='redirect'),