def test_client_instantiates_with_kwargs(self, StrictRedisClient):
client = StrictRedis(num=0)
client.connect()
self.assertEquals(StrictRedisClient.call_count, 1)
StrictRedisClient.assert_any_call(host='localhost', port=6379, db=0, socket_timeout=None,
password=None, unix_socket_path=None)
def setUp(self):
self.redis = StrictRedis(num=0, db=1)
self.redis.flushdb()
class StrictRedisTest(BaseTest):
def setUp(self):
self.redis = StrictRedis(num=0, db=1)
self.redis.flushdb()
def test_proxy(self):
self.assertEquals(self.redis.incr('RedisTest_proxy'), 1)
def test_with_cluster(self):
p = BaseCluster(
backend=StrictRedis,
hosts={0: {'db': 1}},
)
self.assertEquals(p.incr('RedisTest_with_cluster'), 1)
def test_provides_retryable_exceptions(self):
self.assertEquals(StrictRedis.retryable_exceptions, frozenset([redis.RedisError]))
def test_provides_identifier(self):
self.assertEquals(self.redis.identifier, str(self.redis.identifier))
@mock.patch('nydus.db.backends.redis.StrictRedisClient')
def test_client_instantiates_with_kwargs(self, StrictRedisClient):
client = StrictRedis(num=0)
client.connect()
self.assertEquals(StrictRedisClient.call_count, 1)
StrictRedisClient.assert_any_call(host='localhost', port=6379, db=0, socket_timeout=None,
password=None, unix_socket_path=None)
@mock.patch('nydus.db.backends.redis.StrictRedisClient')
def test_map_does_pipeline(self, StrictRedisClient):
redis = create_cluster({
'backend': 'nydus.db.backends.redis.StrictRedis',
'router': 'nydus.db.routers.keyvalue.PartitionRouter',
'hosts': {
0: {'db': 0},
1: {'db': 1},
}
})
with redis.map() as conn:
conn.set('a', 0)
conn.set('d', 1)
# ensure this was actually called through the pipeline
self.assertFalse(StrictRedisClient().set.called)
self.assertEquals(StrictRedisClient().pipeline.call_count, 2)
StrictRedisClient().pipeline.assert_called_with()
self.assertEquals(StrictRedisClient().pipeline().set.call_count, 2)
StrictRedisClient().pipeline().set.assert_any_call('a', 0)
StrictRedisClient().pipeline().set.assert_any_call('d', 1)
self.assertEquals(StrictRedisClient().pipeline().execute.call_count, 2)
StrictRedisClient().pipeline().execute.assert_called_with()
@mock.patch('nydus.db.backends.redis.StrictRedisClient')
def test_map_only_runs_on_required_nodes(self, StrictRedisClient):
redis = create_cluster({
'engine': 'nydus.db.backends.redis.StrictRedis',
'router': 'nydus.db.routers.keyvalue.PartitionRouter',
'hosts': {
0: {'db': 0},
1: {'db': 1},
}
})
with redis.map() as conn:
conn.set('a', 0)
conn.set('b', 1)
# ensure this was actually called through the pipeline
self.assertFalse(StrictRedisClient().set.called)
self.assertEquals(StrictRedisClient().pipeline.call_count, 1)
StrictRedisClient().pipeline.assert_called_with()
self.assertEquals(StrictRedisClient().pipeline().set.call_count, 2)
StrictRedisClient().pipeline().set.assert_any_call('a', 0)
StrictRedisClient().pipeline().set.assert_any_call('b', 1)
self.assertEquals(StrictRedisClient().pipeline().execute.call_count, 1)
StrictRedisClient().pipeline().execute.assert_called_with()