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