def test_write_two_chunks(self): cmd = 'ECHO' payload = "x" * 512 msg = writer(cmd, payload) sock_mock = Mock() self.socket_mock.socket.return_value = sock_mock reader_mock = Mock() self.reader_mock.return_value = reader_mock connection = pyredis.connection.Connection(host='127.0.0.1', encoding='utf-8') connection._authenticate = Mock() connection._setdb = Mock() connection.write(cmd, payload) self.assertEqual(sock_mock.sendall.call_args_list, [call(msg)])
def test_write_two_chunks(self): cmd = 'ECHO' payload = "x" * 512 msg = writer(cmd, payload) sock_mock = Mock() sock_mock.send.side_effect = [500, 34] self.socket_mock.socket.return_value = sock_mock reader_mock = Mock() self.reader_mock.return_value = reader_mock connection = pyredis.connection.Connection(host='127.0.0.1', encoding='utf-8') connection._authenticate = Mock() connection._setdb = Mock() connection.write(cmd, payload) self.assertEqual(sock_mock.send.call_args_list, [call(msg), call(msg[500:])])
def test_write_one_chunk(self): cmd = u'ECHO' payload = u"x" * 512 msg = writer(cmd, payload) length = 534 sock_mock = Mock() sock_mock.send.side_effect = [length] self.socket_mock.socket.return_value = sock_mock reader_mock = Mock() self.reader_mock.return_value = reader_mock connection = pyredis.connection.Connection(host=u'127.0.0.1', encoding=u'utf-8') connection._authenticate = Mock() connection._setdb = Mock() connection.write(cmd, payload) self.assertEqual(sock_mock.send.call_args_list, [call(msg)])