def test__execute_bulk_bulk_size_reached(self):
conn_mock_1 = Mock()
conn_mock_1.read.return_value = 'PONG1'
conn_mock_2 = Mock()
conn_mock_2.read.return_value = 'PONG2'
conn_mock_3 = Mock()
conn_mock_3.read.return_value = 'PONG3'
self.connection_mock.side_effect = [
conn_mock_1, conn_mock_2, conn_mock_3
]
client = pyredis.client.HashClient(buckets=self.buckets)
client.bulk_start(3)
result = client._execute_bulk(u'Ping', conn=conn_mock_1)
self.assertIsNone(result)
self.assertEqual(client._bulk_size_current, 1)
self.assertEqual(client._bulk_results, [])
result = client._execute_bulk(u'Ping', conn=conn_mock_2)
self.assertIsNone(result)
self.assertEqual(client._bulk_size_current, 2)
self.assertEqual(client._bulk_results, [])
result = client._execute_bulk(u'Ping', conn=conn_mock_3)
self.assertIsNone(result)
self.assertEqual(client._bulk_size_current, 0)
self.assertEqual(client._bulk_results, ['PONG1', 'PONG2', 'PONG3'])
def test_bulk_start_no_keep_results(self):
client = pyredis.client.Client(host='127.0.0.1')
client.bulk_start(keep_results=False)
self.assertTrue(client.bulk)
self.assertEqual(client._bulk_size, 5000)
self.assertEqual(client._bulk_size_current, 0)
self.assertIsNone(client._bulk_results)
self.assertFalse(client._bulk_keep)
def test_bulk_start(self):
client = pyredis.client.Client(host='127.0.0.1')
client.bulk_start()
self.assertTrue(client.bulk)
self.assertEqual(client._bulk_size, 5000)
self.assertEqual(client._bulk_size_current, 0)
self.assertEqual(client._bulk_results, [])
self.assertTrue(client._bulk_keep)
def test_bulk_start_no_keep_results(self):
client = pyredis.client.Client(host='127.0.0.1')
client.bulk_start(keep_results=False)
self.assertTrue(client.bulk)
self.assertEqual(client._bulk_size, 5000)
self.assertEqual(client._bulk_size_current, 0)
self.assertIsNone(client._bulk_results)
self.assertFalse(client._bulk_keep)
def test_bulk_start(self):
client = pyredis.client.Client(host='127.0.0.1')
client.bulk_start()
self.assertTrue(client.bulk)
self.assertEqual(client._bulk_size, 5000)
self.assertEqual(client._bulk_size_current, 0)
self.assertEqual(client._bulk_results, [])
self.assertTrue(client._bulk_keep)
def test__execute_bulk_bulk_size_not_reached(self):
conn_mock = Mock()
self.connection_mock.return_value = conn_mock
client = pyredis.client.Client(host='127.0.0.1')
client.bulk_start()
result = client._execute_bulk('Ping')
conn_mock.write.assert_called_with('Ping')
self.assertIsNone(result)
def test__execute_bulk_bulk_size_not_reached(self):
conn_mock = Mock()
self.connection_mock.return_value = conn_mock
client = pyredis.client.Client(host='127.0.0.1')
client.bulk_start()
result = client._execute_bulk('Ping')
conn_mock.write.assert_called_with('Ping')
self.assertIsNone(result)
def test_bulk_start_raise_already_open(self):
conn_mock_1 = Mock()
conn_mock_2 = Mock()
conn_mock_3 = Mock()
self.connection_mock.side_effect = [conn_mock_1, conn_mock_2, conn_mock_3]
client = pyredis.client.HashClient(buckets=self.buckets)
client.bulk_start()
self.assertRaises(PyRedisError, client.bulk_start)
def test__execute_bulk_bulk_size_not_reached(self):
conn_mock_1 = Mock()
conn_mock_2 = Mock()
conn_mock_3 = Mock()
self.connection_mock.side_effect = [conn_mock_1, conn_mock_2, conn_mock_3]
client = pyredis.client.HashClient(buckets=self.buckets)
client.bulk_start()
result = client._execute_bulk('Ping', conn=conn_mock_1)
conn_mock_1.write.assert_called_with('Ping')
self.assertIsNone(result)
def test_bulk_start_bulk_size_42(self):
conn_mock_1 = Mock()
conn_mock_2 = Mock()
conn_mock_3 = Mock()
self.connection_mock.side_effect = [conn_mock_1, conn_mock_2, conn_mock_3]
client = pyredis.client.HashClient(buckets=self.buckets)
client.bulk_start(bulk_size=42)
self.assertTrue(client.bulk)
self.assertEqual(client._bulk_size, 42)
self.assertEqual(client._bulk_size_current, 0)
def test_bulk_start_no_keep_results(self):
conn_mock_1 = Mock()
conn_mock_2 = Mock()
conn_mock_3 = Mock()
self.connection_mock.side_effect = [conn_mock_1, conn_mock_2, conn_mock_3]
client = pyredis.client.HashClient(buckets=self.buckets)
client.bulk_start(keep_results=False)
self.assertTrue(client.bulk)
self.assertEqual(client._bulk_size, 5000)
self.assertEqual(client._bulk_size_current, 0)
self.assertIsNone(client._bulk_results)
self.assertFalse(client._bulk_keep)
def test__execute_bulk_bulk_size_reached(self):
conn_mock = Mock()
conn_mock.read.return_value = b'PONG'
self.connection_mock.return_value = conn_mock
client = pyredis.client.Client(host='127.0.0.1')
client.bulk_start(3)
result = client._execute_bulk('Ping')
self.assertIsNone(result)
self.assertEqual(client._bulk_size_current, 1)
self.assertEqual(client._bulk_results, [])
result = client._execute_bulk('Ping')
self.assertIsNone(result)
self.assertEqual(client._bulk_size_current, 2)
self.assertEqual(client._bulk_results, [])
result = client._execute_bulk('Ping')
self.assertIsNone(result)
self.assertEqual(client._bulk_size_current, 0)
self.assertEqual(client._bulk_results, [b'PONG', b'PONG', b'PONG'])
def test__execute_bulk_bulk_size_reached(self):
conn_mock = Mock()
conn_mock.read.return_value = b'PONG'
self.connection_mock.return_value = conn_mock
client = pyredis.client.Client(host='127.0.0.1')
client.bulk_start(3)
result = client._execute_bulk('Ping')
self.assertIsNone(result)
self.assertEqual(client._bulk_size_current, 1)
self.assertEqual(client._bulk_results, [])
result = client._execute_bulk('Ping')
self.assertIsNone(result)
self.assertEqual(client._bulk_size_current, 2)
self.assertEqual(client._bulk_results, [])
result = client._execute_bulk('Ping')
self.assertIsNone(result)
self.assertEqual(client._bulk_size_current, 0)
self.assertEqual(client._bulk_results, [b'PONG', b'PONG', b'PONG'])
def test_bulk_start_bulk_size_42(self):
client = pyredis.client.Client(host='127.0.0.1')
client.bulk_start(bulk_size=42)
self.assertTrue(client.bulk)
self.assertEqual(client._bulk_size, 42)
self.assertEqual(client._bulk_size_current, 0)
def test_bulk_start_raise_already_open(self):
client = pyredis.client.Client(host='127.0.0.1')
client.bulk_start()
self.assertRaises(PyRedisError, client.bulk_start)
def test_bulk_start_bulk_size_42(self):
client = pyredis.client.Client(host='127.0.0.1')
client.bulk_start(bulk_size=42)
self.assertTrue(client.bulk)
self.assertEqual(client._bulk_size, 42)
self.assertEqual(client._bulk_size_current, 0)
def test_bulk_start_raise_already_open(self):
client = pyredis.client.Client(host='127.0.0.1')
client.bulk_start()
self.assertRaises(PyRedisError, client.bulk_start)
