From 3117b0b0e547a5e6bd6019f4a599a94e129f97bb Mon Sep 17 00:00:00 2001
From: Carl George <carl@george.computer>
Date: Mon, 13 Feb 2023 18:13:49 -0600
Subject: [PATCH] Use standard library mock when possible
---
requirements-dev.txt | 2 +-
tests/conftest.py | 5 ++++-
tests/formatter/to_python_test.py | 5 ++++-
tests/formatter/to_wire_test.py | 5 ++++-
tests/model/bless_models_test.py | 5 ++++-
tests/model/create_model_type_test.py | 5 ++++-
tests/model/get_model_name_test.py | 5 ++++-
tests/model/model_discovery_test.py | 5 ++++-
tests/model/model_test.py | 5 ++++-
tests/model/post_process_spec_test.py | 5 ++++-
tests/model/tag_models_test.py | 5 ++++-
tests/operation/security_object_test.py | 5 ++++-
tests/param/add_file_test.py | 5 ++++-
tests/param/cast_request_param_test.py | 5 ++++-
tests/param/get_param_type_spec_test.py | 5 ++++-
tests/param/marshal_param_test.py | 6 ++++--
tests/param/unmarshal_param_test.py | 6 ++++--
tests/request/unmarshal_request_test.py | 5 ++++-
tests/response/unmarshal_response_test.py | 6 ++++--
tests/response/validate_response_body_test.py | 5 ++++-
tests/response/validate_response_headers_test.py | 5 ++++-
tests/response/validate_response_test.py | 6 ++++--
tests/spec/Spec/build_test.py | 6 ++++--
tests/spec/Spec/flattened_spec_test.py | 5 ++++-
tests/spec/build_http_handlers_test.py | 5 ++++-
tests/spec/pickling_test.py | 5 ++++-
tests/swagger20_validator/enum_validator_test.py | 6 ++++--
tests/swagger20_validator/format_validator_test.py | 5 ++++-
tests/swagger20_validator/ref_validator_test.py | 6 ++++--
tests/swagger20_validator/required_validator_test.py | 6 ++++--
tests/swagger20_validator/type_validator_test.py | 5 ++++-
tests/util_test.py | 5 ++++-
32 files changed, 125 insertions(+), 40 deletions(-)
diff --git a/requirements-dev.txt b/requirements-dev.txt
index bc696d5..71f6d12 100644
--- a/requirements-dev.txt
+++ b/requirements-dev.txt
@@ -1,5 +1,5 @@
# Unit test dependencies
-mock
+mock; python_version<'3.3'
mypy-extensions; python_version>='3.5'
mypy; python_version>='3.5'
pre-commit
diff --git a/tests/conftest.py b/tests/conftest.py
index 8126277..b1ad43b 100644
--- a/tests/conftest.py
+++ b/tests/conftest.py
@@ -6,7 +6,10 @@ from itertools import chain
import pytest
import simplejson as json
import yaml
-from mock import Mock
+try:
+ from unittest.mock import Mock
+except ImportError:
+ from mock import Mock
from six import iteritems
from six import iterkeys
from six.moves.urllib import parse as urlparse
diff --git a/tests/formatter/to_python_test.py b/tests/formatter/to_python_test.py
index 9aa6fea..48bde85 100644
--- a/tests/formatter/to_python_test.py
+++ b/tests/formatter/to_python_test.py
@@ -3,7 +3,10 @@ from datetime import date
from datetime import datetime
import six
-from mock import patch
+try:
+ from unittest.mock import patch
+except ImportError:
+ from mock import patch
from bravado_core.formatter import SwaggerFormat
from bravado_core.formatter import to_python
diff --git a/tests/formatter/to_wire_test.py b/tests/formatter/to_wire_test.py
index 803de4e..e6df8cb 100644
--- a/tests/formatter/to_wire_test.py
+++ b/tests/formatter/to_wire_test.py
@@ -4,7 +4,10 @@ from datetime import datetime
import pytest
import six
-from mock import patch
+try:
+ from unittest.mock import patch
+except ImportError:
+ from mock import patch
from pytz import timezone
from bravado_core.exception import SwaggerMappingError
diff --git a/tests/model/bless_models_test.py b/tests/model/bless_models_test.py
index a596a6b..c928d55 100644
--- a/tests/model/bless_models_test.py
+++ b/tests/model/bless_models_test.py
@@ -1,5 +1,8 @@
# -*- coding: utf-8 -*-
-import mock
+try:
+ from unittest import mock
+except ImportError:
+ import mock
import pytest
from bravado_core.model import _bless_models
diff --git a/tests/model/create_model_type_test.py b/tests/model/create_model_type_test.py
index a9ac13f..c5dd359 100644
--- a/tests/model/create_model_type_test.py
+++ b/tests/model/create_model_type_test.py
@@ -1,5 +1,8 @@
# -*- coding: utf-8 -*-
-import mock
+try:
+ from unittest import mock
+except ImportError:
+ import mock
import pytest
from bravado_core.model import create_model_type
diff --git a/tests/model/get_model_name_test.py b/tests/model/get_model_name_test.py
index bdf7578..e49e1c4 100644
--- a/tests/model/get_model_name_test.py
+++ b/tests/model/get_model_name_test.py
@@ -1,5 +1,8 @@
# -*- coding: utf-8 -*-
-import mock
+try:
+ from unittest import mock
+except ImportError:
+ import mock
import pytest
from bravado_core.model import _get_model_name
diff --git a/tests/model/model_discovery_test.py b/tests/model/model_discovery_test.py
index f72b8b4..11469f4 100644
--- a/tests/model/model_discovery_test.py
+++ b/tests/model/model_discovery_test.py
@@ -1,5 +1,8 @@
# -*- coding: utf-8 -*-
-import mock
+try:
+ from unittest import mock
+except ImportError:
+ import mock
import pytest
from bravado_core.model import _run_post_processing
diff --git a/tests/model/model_test.py b/tests/model/model_test.py
index 387ed39..1268ad4 100644
--- a/tests/model/model_test.py
+++ b/tests/model/model_test.py
@@ -3,7 +3,10 @@ import datetime
from copy import deepcopy
import pytest
-from mock import Mock
+try:
+ from unittest.mock import Mock
+except ImportError:
+ from mock import Mock
from six import add_metaclass
from bravado_core.content_type import APP_JSON
diff --git a/tests/model/post_process_spec_test.py b/tests/model/post_process_spec_test.py
index 44e66c5..c45667c 100644
--- a/tests/model/post_process_spec_test.py
+++ b/tests/model/post_process_spec_test.py
@@ -1,7 +1,10 @@
# -*- coding: utf-8 -*-
import functools
-from mock import Mock
+try:
+ from unittest.mock import Mock
+except ImportError:
+ from mock import Mock
from bravado_core.model import _post_process_spec
from bravado_core.spec import Spec
diff --git a/tests/model/tag_models_test.py b/tests/model/tag_models_test.py
index 7af50ed..7046443 100644
--- a/tests/model/tag_models_test.py
+++ b/tests/model/tag_models_test.py
@@ -1,7 +1,10 @@
# -*- coding: utf-8 -*-
import copy
-import mock
+try:
+ from unittest import mock
+except ImportError:
+ import mock
import pytest
from bravado_core import model
diff --git a/tests/operation/security_object_test.py b/tests/operation/security_object_test.py
index f2f602d..c11ce59 100644
--- a/tests/operation/security_object_test.py
+++ b/tests/operation/security_object_test.py
@@ -1,6 +1,9 @@
# -*- coding: utf-8 -*-
import pytest
-from mock import Mock
+try:
+ from unittest.mock import Mock
+except ImportError:
+ from mock import Mock
from six import iteritems
from bravado_core.exception import SwaggerSchemaError
diff --git a/tests/param/add_file_test.py b/tests/param/add_file_test.py
index 0612bf0..85dfa7e 100644
--- a/tests/param/add_file_test.py
+++ b/tests/param/add_file_test.py
@@ -1,6 +1,9 @@
# -*- coding: utf-8 -*-
import pytest
-from mock import Mock
+try:
+ from unittest.mock import Mock
+except ImportError:
+ from mock import Mock
from bravado_core.exception import SwaggerMappingError
from bravado_core.operation import Operation
diff --git a/tests/param/cast_request_param_test.py b/tests/param/cast_request_param_test.py
index 4cb1534..1cd560f 100644
--- a/tests/param/cast_request_param_test.py
+++ b/tests/param/cast_request_param_test.py
@@ -1,5 +1,8 @@
# -*- coding: utf-8 -*-
-from mock import patch
+try:
+ from unittest.mock import patch
+except ImportError:
+ from mock import patch
from bravado_core.param import cast_request_param
diff --git a/tests/param/get_param_type_spec_test.py b/tests/param/get_param_type_spec_test.py
index 1c2af8e..2184ca6 100644
--- a/tests/param/get_param_type_spec_test.py
+++ b/tests/param/get_param_type_spec_test.py
@@ -1,6 +1,9 @@
# -*- coding: utf-8 -*-
import pytest
-from mock import Mock
+try:
+ from unittest.mock import Mock
+except ImportError:
+ from mock import Mock
from bravado_core.exception import SwaggerMappingError
from bravado_core.operation import Operation
diff --git a/tests/param/marshal_param_test.py b/tests/param/marshal_param_test.py
index b39000b..6e9329e 100644
--- a/tests/param/marshal_param_test.py
+++ b/tests/param/marshal_param_test.py
@@ -5,8 +5,10 @@ from json import loads
import pytest
from jsonschema import ValidationError
-from mock import Mock
-from mock import patch
+try:
+ from unittest.mock import Mock, patch
+except ImportError:
+ from mock import Mock, patch
from bravado_core.content_type import APP_JSON
from bravado_core.operation import Operation
diff --git a/tests/param/unmarshal_param_test.py b/tests/param/unmarshal_param_test.py
index a10476b..d306caa 100644
--- a/tests/param/unmarshal_param_test.py
+++ b/tests/param/unmarshal_param_test.py
@@ -2,8 +2,10 @@
import datetime
import pytest
-from mock import Mock
-from mock import patch
+try:
+ from unittest.mock import Mock, patch
+except ImportError:
+ from mock import Mock, patch
from bravado_core.operation import Operation
from bravado_core.param import Param
diff --git a/tests/request/unmarshal_request_test.py b/tests/request/unmarshal_request_test.py
index 7b1af3a..92a0f03 100644
--- a/tests/request/unmarshal_request_test.py
+++ b/tests/request/unmarshal_request_test.py
@@ -1,6 +1,9 @@
# -*- coding: utf-8 -*-
import pytest
-from mock import Mock
+try:
+ from unittest.mock import Mock
+except ImportError:
+ from mock import Mock
from bravado_core.exception import SwaggerMappingError
from bravado_core.operation import Operation
diff --git a/tests/response/unmarshal_response_test.py b/tests/response/unmarshal_response_test.py
index cc302ea..9b28f27 100644
--- a/tests/response/unmarshal_response_test.py
+++ b/tests/response/unmarshal_response_test.py
@@ -2,8 +2,10 @@
import msgpack
import pytest
from jsonschema import ValidationError
-from mock import Mock
-from mock import patch
+try:
+ from unittest.mock import Mock, patch
+except ImportError:
+ from mock import Mock, patch
from bravado_core.content_type import APP_JSON
from bravado_core.content_type import APP_MSGPACK
diff --git a/tests/response/validate_response_body_test.py b/tests/response/validate_response_body_test.py
index d7e2bd6..4faf37b 100644
--- a/tests/response/validate_response_body_test.py
+++ b/tests/response/validate_response_body_test.py
@@ -1,7 +1,10 @@
# -*- coding: utf-8 -*-
import msgpack
import pytest
-from mock import Mock
+try:
+ from unittest.mock import Mock
+except ImportError:
+ from mock import Mock
from bravado_core.content_type import APP_MSGPACK
from bravado_core.exception import SwaggerMappingError
diff --git a/tests/response/validate_response_headers_test.py b/tests/response/validate_response_headers_test.py
index 4b9612c..b5ddf9f 100644
--- a/tests/response/validate_response_headers_test.py
+++ b/tests/response/validate_response_headers_test.py
@@ -1,7 +1,10 @@
# -*- coding: utf-8 -*-
import pytest
from jsonschema.exceptions import ValidationError
-from mock import Mock
+try:
+ from unittest.mock import Mock
+except ImportError:
+ from mock import Mock
from bravado_core.operation import Operation
from bravado_core.response import OutgoingResponse
diff --git a/tests/response/validate_response_test.py b/tests/response/validate_response_test.py
index b339a72..d88e50d 100644
--- a/tests/response/validate_response_test.py
+++ b/tests/response/validate_response_test.py
@@ -1,6 +1,8 @@
# -*- coding: utf-8 -*-
-from mock import Mock
-from mock import patch
+try:
+ from unittest.mock import Mock, patch
+except ImportError:
+ from mock import Mock, patch
from bravado_core.operation import Operation
from bravado_core.response import OutgoingResponse
diff --git a/tests/spec/Spec/build_test.py b/tests/spec/Spec/build_test.py
index 86f53ac..5fb66e3 100644
--- a/tests/spec/Spec/build_test.py
+++ b/tests/spec/Spec/build_test.py
@@ -1,7 +1,9 @@
# -*- coding: utf-8 -*-
import pytest
-from mock import Mock
-from mock import patch
+try:
+ from unittest.mock import Mock, patch
+except ImportError:
+ from mock import Mock, patch
from six import iterkeys
from six.moves.urllib.request import pathname2url
from swagger_spec_validator.common import SwaggerValidationError
diff --git a/tests/spec/Spec/flattened_spec_test.py b/tests/spec/Spec/flattened_spec_test.py
index 8c0e948..7d44252 100644
--- a/tests/spec/Spec/flattened_spec_test.py
+++ b/tests/spec/Spec/flattened_spec_test.py
@@ -3,7 +3,10 @@ import copy
import functools
import os
-import mock
+try:
+ from unittest import mock
+except ImportError:
+ import mock
import pytest
from six.moves.urllib.parse import urlparse
from swagger_spec_validator import validator20
diff --git a/tests/spec/build_http_handlers_test.py b/tests/spec/build_http_handlers_test.py
index 55d9ca8..76b9804 100644
--- a/tests/spec/build_http_handlers_test.py
+++ b/tests/spec/build_http_handlers_test.py
@@ -2,7 +2,10 @@
import json
from io import StringIO
-import mock
+try:
+ from unittest import mock
+except ImportError:
+ import mock
import pytest
import yaml
diff --git a/tests/spec/pickling_test.py b/tests/spec/pickling_test.py
index 34a3407..b59fbb5 100644
--- a/tests/spec/pickling_test.py
+++ b/tests/spec/pickling_test.py
@@ -1,5 +1,8 @@
# -*- coding: utf-8 -*-
-import mock
+try:
+ from unittest import mock
+except ImportError:
+ import mock
import pytest
from six.moves.cPickle import dumps
from six.moves.cPickle import loads
diff --git a/tests/swagger20_validator/enum_validator_test.py b/tests/swagger20_validator/enum_validator_test.py
index f163a87..f149206 100644
--- a/tests/swagger20_validator/enum_validator_test.py
+++ b/tests/swagger20_validator/enum_validator_test.py
@@ -1,8 +1,10 @@
# -*- coding: utf-8 -*-
import pytest
from jsonschema import ValidationError
-from mock import Mock
-from mock import patch
+try:
+ from unittest.mock import Mock, patch
+except ImportError:
+ from mock import Mock, patch
from bravado_core.spec import Spec
from bravado_core.swagger20_validator import enum_validator
diff --git a/tests/swagger20_validator/format_validator_test.py b/tests/swagger20_validator/format_validator_test.py
index 8e6bd8f..2ff0e84 100644
--- a/tests/swagger20_validator/format_validator_test.py
+++ b/tests/swagger20_validator/format_validator_test.py
@@ -1,7 +1,10 @@
# -*- coding: utf-8 -*-
import pytest
from jsonschema import ValidationError
-from mock import patch
+try:
+ from unittest.mock import patch
+except ImportError:
+ from mock import patch
from bravado_core.exception import SwaggerValidationError
from bravado_core.formatter import SwaggerFormat
diff --git a/tests/swagger20_validator/ref_validator_test.py b/tests/swagger20_validator/ref_validator_test.py
index b9c71a2..844cc1f 100644
--- a/tests/swagger20_validator/ref_validator_test.py
+++ b/tests/swagger20_validator/ref_validator_test.py
@@ -1,8 +1,10 @@
# -*- coding: utf-8 -*-
import pytest
from jsonschema.validators import RefResolver
-from mock import MagicMock
-from mock import Mock
+try:
+ from unittest.mock import MagicMock, Mock
+except ImportError:
+ from mock import MagicMock, Mock
from bravado_core.swagger20_validator import ref_validator
diff --git a/tests/swagger20_validator/required_validator_test.py b/tests/swagger20_validator/required_validator_test.py
index 13e33bf..8b07ac8 100644
--- a/tests/swagger20_validator/required_validator_test.py
+++ b/tests/swagger20_validator/required_validator_test.py
@@ -1,8 +1,10 @@
# -*- coding: utf-8 -*-
import pytest
from jsonschema.exceptions import ValidationError
-from mock import Mock
-from mock import patch
+try:
+ from unittest.mock import Mock, patch
+except ImportError:
+ from mock import Mock, patch
from bravado_core.swagger20_validator import required_validator
diff --git a/tests/swagger20_validator/type_validator_test.py b/tests/swagger20_validator/type_validator_test.py
index 01375aa..7d6d749 100644
--- a/tests/swagger20_validator/type_validator_test.py
+++ b/tests/swagger20_validator/type_validator_test.py
@@ -1,5 +1,8 @@
# -*- coding: utf-8 -*-
-from mock import patch
+try:
+ from unittest.mock import patch
+except ImportError:
+ from mock import patch
from bravado_core.swagger20_validator import type_validator
diff --git a/tests/util_test.py b/tests/util_test.py
index 6ee8fb1..d3e7b82 100644
--- a/tests/util_test.py
+++ b/tests/util_test.py
@@ -1,7 +1,10 @@
# -*- coding: utf-8 -*-
from inspect import getcallargs
-import mock
+try:
+ from unittest import mock
+except ImportError:
+ import mock
import pytest
from bravado_core.util import AliasKeyDict
--
2.39.1