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)])
def test_encode_3_args(self):
expected = b'*3\r\n$3\r\nSET\r\n$8\r\nKey/Name\r\n$19\r\nSomeValue_?#!\xc3\x84\xc3\x9c\xc3\x96\r\n'
self.assertEqual(writer('SET', 'Key/Name', 'SomeValue_?#!ÄÜÖ'),
expected)
def test_encode_2_args(self):
expected = b'*2\r\n$4\r\nECHO\r\n$14\r\nTest!!!!111elf\r\n'
self.assertEqual(writer('ECHO', 'Test!!!!111elf'), expected)
def test_encode_1_args(self):
expected = b'*1\r\n$4\r\nPING\r\n'
self.assertEqual(writer('PING'), expected)
def test_encode_0_args(self):
expected = b'*0\r\n'
self.assertEqual(writer(), expected)
def test_encode_3_args(self):
expected = b'*3\r\n$3\r\nSET\r\n$8\r\nKey/Name\r\n$19\r\nSomeValue_?#!\xc3\x84\xc3\x9c\xc3\x96\r\n'
self.assertEqual(
writer('SET', 'Key/Name', 'SomeValue_?#!ÄÜÖ'),
expected)
def test_encode_2_args(self):
expected = b'*2\r\n$4\r\nECHO\r\n$14\r\nTest!!!!111elf\r\n'
self.assertEqual(
writer('ECHO', 'Test!!!!111elf'),
expected)
def test_encode_1_args(self):
expected = b'*1\r\n$4\r\nPING\r\n'
self.assertEqual(
writer('PING'),
expected)
def test_encode_0_args(self):
expected = b'*0\r\n'
self.assertEqual(
writer(),
expected)