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_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): 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')