2020-12-12 12:12:15 +00:00
|
|
|
--- a/src/geventhttpclient/tests/test_ssl.py
|
|
|
|
+++ b/src/geventhttpclient/tests/test_ssl.py
|
2020-12-13 21:28:49 +00:00
|
|
|
@@ -1,5 +1,7 @@
|
2020-12-12 12:12:15 +00:00
|
|
|
-import mock
|
|
|
|
-from mock import patch, Mock
|
2020-12-13 21:28:49 +00:00
|
|
|
+try:
|
|
|
|
+ import unittest.mock as mock
|
|
|
|
+except ImportError:
|
|
|
|
+ import mock
|
2020-12-12 12:12:15 +00:00
|
|
|
|
|
|
|
import dpkt.ssl
|
|
|
|
import six
|
2020-12-13 21:28:49 +00:00
|
|
|
@@ -117,7 +119,7 @@ def _get_sni_sent_from_client(**addition
|
2020-12-12 12:12:15 +00:00
|
|
|
('127.0.0.1', server_port)
|
|
|
|
)
|
|
|
|
with mock.patch(
|
|
|
|
- 'gevent.socket.getaddrinfo', Mock(return_value=[mock_addrinfo])
|
|
|
|
+ 'gevent.socket.getaddrinfo', mock.Mock(return_value=[mock_addrinfo])
|
|
|
|
):
|
|
|
|
|
|
|
|
server_host = 'some_foo'
|