Cross python version mock import

ckan · Nov 30, 2022 · 0b1fca6 · 0b1fca6
1 parent 8d68f3b
commit 0b1fca6
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 3 deletions.
diff --git a/ckanext/xloader/tests/test_action.py b/ckanext/xloader/tests/test_action.py
@@ -1,5 +1,8 @@
 import pytest
-import mock
+try:
+    from unittest import mock
+except ImportError:
+    import mock
 
 from ckan.tests import helpers, factories
 

diff --git a/ckanext/xloader/tests/test_jobs.py b/ckanext/xloader/tests/test_jobs.py
@@ -13,7 +13,10 @@
 import pytest
 
 from nose.tools import make_decorator
-import mock
+try:
+    from unittest import mock
+except ImportError:
+    import mock
 import responses
 from sqlalchemy import MetaData, Table
 from sqlalchemy.sql import select

diff --git a/ckanext/xloader/tests/test_plugin.py b/ckanext/xloader/tests/test_plugin.py
@@ -2,7 +2,10 @@
 
 import datetime
 import pytest
-import mock
+try:
+    from unittest import mock
+except ImportError:
+    import mock
 from six import text_type as str
 from ckan.tests import helpers, factories
 from ckan.logic import _actions