def test_execute_bulk(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._execute_bulk = Mock()
client._bulk = True
client.execute(b'PING', shard_key='blarg')
client._execute_bulk.assert_called_with(b'PING', conn=conn_mock_3)
def test_execute(self):
client = pyredis.client.Client(host='127.0.0.1')
client._execute_basic = Mock()
client._execute_basic.return_value = b'PONG'
result = client.execute(b'PING')
client._execute_basic.assert_called_with(b'PING')
self.assertEqual(result, b'PONG')
def test_execute(self):
client = pyredis.client.Client(host='127.0.0.1')
client._execute_basic = Mock()
client._execute_basic.return_value = b'PONG'
result = client.execute(b'PING')
client._execute_basic.assert_called_with(b'PING')
self.assertEqual(result, b'PONG')
def test_execute_non_bulk(self):
client = pyredis.client.Client(host=u'127.0.0.1')
client._execute_basic = Mock()
client._execute_basic.return_value = 'PONG'
result = client.execute(u'Ping')
client._execute_basic.assert_called_with(u'Ping')
self.assertEqual(result, 'PONG')
def test_execute(self):
sentinels = [('host1', 12345), ('host2', 12345), ('host3', 12345)]
client = pyredis.client.SentinelClient(sentinels=sentinels)
client._conn = Mock()
client._conn.read.return_value = b'PONG'
result = client.execute('PING')
client._conn.write.assert_called_with('PING')
client._conn.read.assert_called_with()
self.assertEqual(result, b'PONG')
def test_execute(self):
sentinels = [('host1', 12345), ('host2', 12345), ('host3', 12345)]
client = pyredis.client.SentinelClient(sentinels=sentinels)
client._conn = Mock()
client._conn.read.return_value = b'PONG'
result = client.execute('PING')
client._conn.write.assert_called_with('PING')
client._conn.read.assert_called_with()
self.assertEqual(result, b'PONG')
def test_execute(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._execute_basic = Mock()
client._execute_basic.return_value = b'PONG'
result = client.execute(b'PING', shard_key='blarg')
client._execute_basic.assert_called_with(b'PING', conn=conn_mock_3)
self.assertEqual(result, b'PONG')
def test_execute_non_bulk_shard_key(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)
print client._conns
client._execute_basic = Mock()
client._execute_basic.return_value = 'PONG'
result = client.execute(u'Ping', shard_key=u'blarg')
client._execute_basic.assert_called_with(u'Ping', conn=conn_mock_3)
self.assertEqual(result, 'PONG')
def test_execute_bulk(self):
client = pyredis.client.Client(host='127.0.0.1')
client._execute_bulk = Mock()
client._bulk = True
client.execute(b'PING')
client._execute_bulk.assert_called_with(b'PING')
def test_execute_bulk(self):
client = pyredis.client.Client(host='127.0.0.1')
client._execute_bulk = Mock()
client._bulk = True
client.execute(b'PING')
client._execute_bulk.assert_called_with(b'PING')
