forked from pool/python-geventhttpclient
19 lines
559 B
Diff
19 lines
559 B
Diff
|
--- a/src/geventhttpclient/tests/test_ssl.py
|
||
|
|
+++ b/src/geventhttpclient/tests/test_ssl.py
|
||
|
|
@@ -1,5 +1,4 @@
|
||
|
|
-import mock
|
||
|
|
-from mock import patch, Mock
|
||
|
|
+import unittest.mock as mock
|
||
|
|
|
||
|
|
import dpkt.ssl
|
||
|
|
import six
|
||
|
|
@@ -117,7 +116,7 @@ def _get_sni_sent_from_client(**addition
|
||
|
|
('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'
|