Fix unit test for Python 3.7

apache · Jul 6, 2023 · d873c4c · d873c4c
1 parent 095f0cf
commit d873c4c
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git a/setup.cfg b/setup.cfg
@@ -118,6 +118,7 @@ install_requires =
     markupsafe>=1.1.1
     marshmallow-oneofschema>=2.0.1
     mdit-py-plugins>=0.3.0
+    mock>=5.0.2;python_version<="3.7"
     # Pip can not find a version that satisfies constraints if opentelemetry-api is not pinned.
     opentelemetry-api==1.15.0
     opentelemetry-exporter-otlp

diff --git a/tests/providers/http/hooks/test_http.py b/tests/providers/http/hooks/test_http.py
@@ -21,9 +21,14 @@
 import json
 import logging
 import os
+import sys
 from collections import OrderedDict
 from http import HTTPStatus
-from unittest import mock
+
+if sys.version_info.major == 3 and sys.version_info.minor >= 8:
+    from unittest import mock
+else:
+    from unittest import mock
 
 import pytest
 import requests