diff --git a/test/test_rosdep_alpine.py b/test/test_rosdep_alpine.py index 5ab0bf433..d719f22d8 100644 --- a/test/test_rosdep_alpine.py +++ b/test/test_rosdep_alpine.py @@ -29,7 +29,10 @@ import os import traceback -from mock import Mock, patch +try: + from unittest.mock import Mock, patch +except ImportError: + from mock import Mock, patch import rospkg.os_detect diff --git a/test/test_rosdep_arch.py b/test/test_rosdep_arch.py index 857aa58cb..5c7ccc03c 100644 --- a/test/test_rosdep_arch.py +++ b/test/test_rosdep_arch.py @@ -29,7 +29,10 @@ import os import traceback -from mock import Mock, patch +try: + from unittest.mock import Mock, patch +except ImportError: + from mock import Mock, patch def get_test_dir(): diff --git a/test/test_rosdep_cygwin.py b/test/test_rosdep_cygwin.py index b277ea51d..1919e240d 100644 --- a/test/test_rosdep_cygwin.py +++ b/test/test_rosdep_cygwin.py @@ -29,7 +29,10 @@ import os import traceback -from mock import patch +try: + from unittest.mock import patch +except ImportError: + from mock import patch def get_test_dir(): diff --git a/test/test_rosdep_debian.py b/test/test_rosdep_debian.py index 75cb60455..428b301f4 100644 --- a/test/test_rosdep_debian.py +++ b/test/test_rosdep_debian.py @@ -29,7 +29,10 @@ import os import traceback -from mock import Mock, patch, call +try: + from unittest.mock import Mock, patch, call +except ImportError: + from mock import Mock, patch, call def get_test_dir(): diff --git a/test/test_rosdep_freebsd.py b/test/test_rosdep_freebsd.py index d8ee607a2..e2db8c6ad 100644 --- a/test/test_rosdep_freebsd.py +++ b/test/test_rosdep_freebsd.py @@ -30,7 +30,10 @@ import os import traceback -from mock import patch, Mock +try: + from unittest.mock import patch, Mock +except ImportError: + from mock import patch, Mock def get_test_dir(): diff --git a/test/test_rosdep_gem.py b/test/test_rosdep_gem.py index b722161e9..1b6f9a43b 100644 --- a/test/test_rosdep_gem.py +++ b/test/test_rosdep_gem.py @@ -31,7 +31,10 @@ import os import traceback -from mock import Mock, patch +try: + from unittest.mock import Mock, patch +except ImportError: + from mock import Mock, patch def get_test_dir(): diff --git a/test/test_rosdep_gentoo.py b/test/test_rosdep_gentoo.py index a44b84b63..015753dfc 100644 --- a/test/test_rosdep_gentoo.py +++ b/test/test_rosdep_gentoo.py @@ -29,7 +29,10 @@ import os import traceback -from mock import Mock, patch +try: + from unittest.mock import Mock, patch +except ImportError: + from mock import Mock, patch import rospkg.os_detect diff --git a/test/test_rosdep_installers.py b/test/test_rosdep_installers.py index 1d3186638..4b91a3f22 100644 --- a/test/test_rosdep_installers.py +++ b/test/test_rosdep_installers.py @@ -28,7 +28,10 @@ from __future__ import print_function from contextlib import contextmanager -from mock import patch +try: + from unittest.mock import Mock, patch +except ImportError: + from mock import Mock, patch import os import sys try: @@ -117,7 +120,6 @@ def test_InstallerContext_os_version_and_name(): context.set_os_override(*val) assert val == context.get_os_name_and_version() - from mock import Mock os_detect_mock = Mock(spec=OsDetect) os_detect_mock.get_name.return_value = 'fakeos' os_detect_mock.get_version.return_value = 'fakeos-version' @@ -559,7 +561,6 @@ def get_packages_to_install(*args): assert 'apt' in str(e) # annoying mock to test generally impossible error condition - from mock import Mock lookup = Mock(spec=RosdepLookup) lookup.resolve_all.return_value = ([('bad-key', ['stuff'])], []) diff --git a/test/test_rosdep_loader.py b/test/test_rosdep_loader.py index b935b2966..b5f2cefb0 100644 --- a/test/test_rosdep_loader.py +++ b/test/test_rosdep_loader.py @@ -25,7 +25,10 @@ # ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE # POSSIBILITY OF SUCH DAMAGE. -from mock import Mock +try: + from unittest.mock import Mock +except ImportError: + from mock import Mock def test_RosdepLoader(): diff --git a/test/test_rosdep_main.py b/test/test_rosdep_main.py index 58c0bb286..c8a73219e 100644 --- a/test/test_rosdep_main.py +++ b/test/test_rosdep_main.py @@ -38,8 +38,10 @@ import unittest -from mock import patch -from mock import DEFAULT +try: + from unittest.mock import DEFAULT, patch +except ImportError: + from mock import DEFAULT, patch from rosdep2 import main from rosdep2.ament_packages import AMENT_PREFIX_PATH_ENV_VAR diff --git a/test/test_rosdep_npm.py b/test/test_rosdep_npm.py index 023005d58..b89ce824f 100644 --- a/test/test_rosdep_npm.py +++ b/test/test_rosdep_npm.py @@ -31,7 +31,10 @@ import os import traceback -from mock import Mock, patch +try: + from unittest.mock import Mock, patch +except ImportError: + from mock import Mock, patch def get_test_dir(): diff --git a/test/test_rosdep_opensuse.py b/test/test_rosdep_opensuse.py index 608474c4c..fb0817654 100644 --- a/test/test_rosdep_opensuse.py +++ b/test/test_rosdep_opensuse.py @@ -29,7 +29,10 @@ import os import traceback -from mock import patch +try: + from unittest.mock import patch +except ImportError: + from mock import patch def get_test_dir(): diff --git a/test/test_rosdep_osx.py b/test/test_rosdep_osx.py index 14cedf0bf..e3c1c59cf 100644 --- a/test/test_rosdep_osx.py +++ b/test/test_rosdep_osx.py @@ -32,9 +32,10 @@ import os import traceback -from mock import call -from mock import Mock -from mock import patch +try: + from unittest.mock import call, Mock, patch +except ImportError: + from mock import call, Mock, patch def get_test_dir(): diff --git a/test/test_rosdep_pip.py b/test/test_rosdep_pip.py index 628481b48..34840ce1c 100644 --- a/test/test_rosdep_pip.py +++ b/test/test_rosdep_pip.py @@ -30,7 +30,10 @@ import os import sys import traceback -from mock import Mock, patch +try: + from unittest.mock import Mock, patch +except ImportError: + from mock import Mock, patch def get_test_dir(): diff --git a/test/test_rosdep_redhat.py b/test/test_rosdep_redhat.py index e9dab6cdd..8a40da4ae 100644 --- a/test/test_rosdep_redhat.py +++ b/test/test_rosdep_redhat.py @@ -29,7 +29,10 @@ import os import traceback -from mock import patch, Mock +try: + from unittest.mock import patch, Mock +except ImportError: + from mock import patch, Mock def get_test_dir(): diff --git a/test/test_rosdep_rospkg_loader.py b/test/test_rosdep_rospkg_loader.py index c397b640a..aa38b9dcc 100644 --- a/test/test_rosdep_rospkg_loader.py +++ b/test/test_rosdep_rospkg_loader.py @@ -30,7 +30,10 @@ import os import yaml -from mock import Mock +try: + from unittest.mock import Mock +except ImportError: + from mock import Mock from rospkg import RosPack, RosStack diff --git a/test/test_rosdep_slackware.py b/test/test_rosdep_slackware.py index 8f6857c72..303d7e1e1 100644 --- a/test/test_rosdep_slackware.py +++ b/test/test_rosdep_slackware.py @@ -29,7 +29,10 @@ import os import traceback -from mock import Mock, patch +try: + from unittest.mock import Mock, patch +except ImportError: + from mock import Mock, patch import rospkg.os_detect