Compatibility with ckan 2.2.

2015-12-10 16:33:09 +00:00 · 2015-12-10 16:33:09 +00:00 · 77f98d5b18
parent 8f54421c99
commit 77f98d5b18
1 changed files with 7 additions and 1 deletions
--- a/ckanext/harvest/tests/test_action.py
+++ b/ckanext/harvest/tests/test_action.py
@ -11,7 +11,13 @@ try:
 except ImportError:
    from ckan.new_tests import factories as ckan_factories
    from ckan.new_tests.helpers import (_get_test_app, reset_db,
-                                        FunctionalTestBase, assert_in)
+                                        FunctionalTestBase)
+    try:
+        from ckan.new_tests.helpers import assert_in
+    except ImportError:
+        # ckan 2.2 only has it in the legacy test helpers
+        from ckan.tests.helpers import assert_in
+
 from ckan import plugins as p
 from ckan.plugins import toolkit
 from ckan import model