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)])
