def test_disconnect(self):
conn = Connection()
mock_sock = mock.Mock()
conn._sock = mock_sock
conn.disconnect()
mock_sock.shutdown.assert_called_once()
mock_sock.close.assert_called_once()
assert conn._sock is None
def test_disconnect__close_OSError(self):
"""An OSError on socket close will still clear out the socket."""
conn = Connection()
mock_sock = mock.Mock()
conn._sock = mock_sock
conn._sock.close.side_effect = OSError
conn.disconnect()
mock_sock.shutdown.assert_called_once()
mock_sock.close.assert_called_once()
assert conn._sock is None
def to_blocking_connection(self, socket_read_size=65536):
""" Convert asynchronous connection to blocking socket connection
"""
conn = Connection(
self.host, self.port, self.db, self.password, self.socket_timeout,
self.socket_connect_timeout, self.socket_keepalive,
self.socket_keepalive_options, self.retry_on_timeout, self.encoding,
self.encoding_errors, self.decode_responses, DefaultParser, socket_read_size)
conn._sock = self._stream.socket
return conn
def to_blocking_connection(self, socket_read_size=65536):
""" Convert asynchronous connection to blocking socket connection
"""
conn = Connection(
self.host, self.port, self.db, self.password, self.socket_timeout,
self.socket_connect_timeout, self.socket_keepalive,
self.socket_keepalive_options, self.retry_on_timeout,
parser_class=DefaultParser, socket_read_size=socket_read_size)
conn._sock = self._stream.socket
return conn
