Secure your code as it's written. Use Snyk Code to scan source code in minutes - no build needed - and fix issues immediately.
def mocketize_teardown(self):
self.assertEqual(len(Mocket._requests), 0)
def test_register(self):
Entry.register(('localhost', 6379), 'SET snowman "is ☃!"', OK)
self.assertEqual(
Mocket._entries[('localhost', 6379)][0].command,
[b'*3', b'$3', b'SET', b'$7', b'snowman', b'$7', b'is \xe2\x98\x83!']
)
self.assertEqual(
Mocket._entries[('localhost', 6379)][0].responses[0].data,
b'+OK\r\n'
)
def mocketize_setup(self):
Entry.register_response('FLUSHDB', OK)
self.rclient.flushdb()
self.assertEqual(len(Mocket._requests), 1)
Mocket.reset()
def enable(namespace=None, truesocket_recording_dir=None):
Mocket._namespace = namespace
Mocket._truesocket_recording_dir = truesocket_recording_dir
if truesocket_recording_dir:
# JSON dumps will be saved here
assert os.path.isdir(truesocket_recording_dir)
socket.socket = socket.__dict__["socket"] = MocketSocket
socket._socketobject = socket.__dict__["_socketobject"] = MocketSocket
socket.SocketType = socket.__dict__["SocketType"] = MocketSocket
socket.create_connection = socket.__dict__[
"create_connection"
] = create_connection
socket.gethostname = socket.__dict__["gethostname"] = lambda: "localhost"
socket.gethostbyname = socket.__dict__[
"gethostbyname"
] = lambda host: "127.0.0.1"
socket.getaddrinfo = socket.__dict__[
def get_entry(self, data):
return Mocket.get_entry(self._host, self._port, data)
def enable(namespace=None, truesocket_recording_dir=None):
Mocket._namespace = namespace
Mocket._truesocket_recording_dir = truesocket_recording_dir
if truesocket_recording_dir:
# JSON dumps will be saved here
assert os.path.isdir(truesocket_recording_dir)
socket.socket = socket.__dict__["socket"] = MocketSocket
socket._socketobject = socket.__dict__["_socketobject"] = MocketSocket
socket.SocketType = socket.__dict__["SocketType"] = MocketSocket
socket.create_connection = socket.__dict__[
"create_connection"
] = create_connection
socket.gethostname = socket.__dict__["gethostname"] = lambda: "localhost"
socket.gethostbyname = socket.__dict__[
"gethostbyname"
] = lambda host: "127.0.0.1"
socket.gethostbyname = socket.__dict__["gethostbyname"] = true_gethostbyname
socket.getaddrinfo = socket.__dict__["getaddrinfo"] = true_getaddrinfo
ssl.wrap_socket = ssl.__dict__["wrap_socket"] = true_ssl_wrap_socket
# ssl.SSLSocket = ssl.__dict__["SSLSocket"] = true_ssl_socket
ssl.SSLContext = ssl.__dict__["SSLContext"] = true_ssl_context
socket.inet_pton = socket.__dict__["inet_pton"] = true_inet_pton
urllib3.util.ssl_.wrap_socket = urllib3.util.ssl_.__dict__[
"wrap_socket"
] = true_urllib3_wrap_socket
urllib3.util.ssl_.ssl_wrap_socket = urllib3.util.ssl_.__dict__[
"ssl_wrap_socket"
] = true_urllib3_ssl_wrap_socket
urllib3.connection.ssl_wrap_socket = urllib3.connection.__dict__[
"ssl_wrap_socket"
] = true_urllib3_ssl_wrap_socket
Mocket.reset()
if pyopenssl_override: # pragma: no cover
# Put the pyopenssl version back in place
inject_into_urllib3()