class TestRedis(TestCase):
def setUp(self):
self.redis = MockRedis()
self.redis.flushdb()
def test_get(self):
self.assertEqual(None, self.redis.get('key'))
self.redis.redis['key'] = 'value'
self.assertEqual('value', self.redis.get('key'))
def test_set(self):
self.assertEqual(None, self.redis.redis.get('key'))
self.redis.set('key', 'value')
self.assertEqual('value', self.redis.redis.get('key'))
def test_get_types(self):
'''
Python bools, lists, dicts are returned as strings by
redis-py/redis.
'''
values = list([
True,
False,
[1, '2'],
{
'a': 1,
'b': 'c'
},
])
self.assertEqual(None, self.redis.get('key'))
for value in values:
self.redis.set('key', value)
self.assertEqual(str(value),
self.redis.get('key'),
"redis.get")
self.redis.hset('hkey', 'item', value)
self.assertEqual(str(value),
self.redis.hget('hkey', 'item'))
self.redis.sadd('skey', value)
self.assertEqual(set([str(value)]),
self.redis.smembers('skey'))
self.redis.flushdb()
def test_incr(self):
'''
incr, hincr when keys exist
'''
values = list([
(1, '2'),
('1', '2'),
])
for value in values:
self.redis.set('key', value[0])
self.redis.incr('key')
self.assertEqual(value[1],
self.redis.get('key'),
"redis.incr")
self.redis.hset('hkey', 'attr', value[0])
self.redis.hincrby('hkey', 'attr')
self.assertEqual(value[1],
self.redis.hget('hkey', 'attr'),
"redis.hincrby")
self.redis.flushdb()
def test_incr_init(self):
'''
incr, hincr, decr when keys do NOT exist
'''
self.redis.incr('key')
self.assertEqual('1', self.redis.get('key'))
self.redis.hincrby('hkey', 'attr')
self.assertEqual('1', self.redis.hget('hkey', 'attr'))
self.redis.decr('dkey')
self.assertEqual('-1', self.redis.get('dkey'))
def test_ttl(self):
self.redis.set('key', 'key')
self.redis.expire('key', 30)
assert self.redis.ttl('key') <= 30
self.assertEqual(self.redis.ttl('invalid_key'), -1)
def test_push_pop_returns_str(self):
key = 'l'
values = ['5', 5, [], {}]
for v in values:
self.redis.rpush(key, v)
self.assertEquals(self.redis.lpop(key),
str(v))
def test_llen(self):
self.assertEqual(self.redis.llen('key'), 0)
self.assertFalse(self.redis.exists('key'))
self.redis.rpush('key', 'value')
self.assertEqual(self.redis.llen('key'), 1)
def test_lrange(self):
self.assertEqual(self.redis.lrange('key', 0, 100), [])
self.assertFalse(self.redis.exists('key'))
for i in range(3):
self.redis.rpush('key', i)
self.assertEqual(self.redis.lrange('key', 2, 0), [])
self.assertEqual(self.redis.lrange('key', 0, 100), ['0', '1', '2'])
self.assertEqual(self.redis.lrange('key', 0, 2), ['0', '1', '2'])
self.assertEqual(self.redis.lrange('key', 1, 1), ['1'])
self.assertEqual(self.redis.lrange('key', 1, -1), ['1', '2'])
self.assertEqual(self.redis.lrange('key', 1, -2), ['1'])
self.assertEqual(self.redis.lrange('key', -3, -2), ['0', '1'])
self.assertEqual(self.redis.lrange('key', -1, -1), ['2'])
class TestRedisHash(TestCase):
"""hash tests"""
def setUp(self):
self.redis = MockRedis()
self.redis.flushdb()
def test_hexists(self):
hashkey = "hash"
self.assertFalse(self.redis.hexists(hashkey, "key"))
self.redis.hset(hashkey, "key", "value")
self.assertTrue(self.redis.hexists(hashkey, "key"))
self.assertFalse(self.redis.hexists(hashkey, "key2"))
def test_hgetall(self):
hashkey = "hash"
self.assertEquals({}, self.redis.hgetall(hashkey))
self.redis.hset(hashkey, "key", "value")
self.assertEquals({"key": "value"}, self.redis.hgetall(hashkey))
def test_hdel(self):
hashkey = "hash"
self.redis.hmset(hashkey, {1: 1, 2: 2, 3: 3})
self.assertEquals(0, self.redis.hdel(hashkey, "foo"))
self.assertEquals({
"1": "1",
"2": "2",
"3": "3"
}, self.redis.hgetall(hashkey))
self.assertEquals(2, self.redis.hdel(hashkey, "1", 2))
self.assertEquals({"3": "3"}, self.redis.hgetall(hashkey))
self.assertEquals(1, self.redis.hdel(hashkey, "3", 4))
self.assertEquals({}, self.redis.hgetall(hashkey))
self.assertFalse(self.redis.exists(hashkey))
def test_hlen(self):
hashkey = "hash"
self.assertEquals(0, self.redis.hlen(hashkey))
self.redis.hset(hashkey, "key", "value")
self.assertEquals(1, self.redis.hlen(hashkey))
def test_hset(self):
hashkey = "hash"
self.redis.hset(hashkey, "key", "value")
self.assertEquals("value", self.redis.hget(hashkey, "key"))
def test_hget(self):
hashkey = "hash"
self.assertEquals(None, self.redis.hget(hashkey, "key"))
def test_hset_integral(self):
hashkey = "hash"
self.redis.hset(hashkey, 1, 2)
self.assertEquals("2", self.redis.hget(hashkey, 1))
self.assertEquals("2", self.redis.hget(hashkey, "1"))
def test_hsetnx(self):
hashkey = "hash"
self.assertEquals(1, self.redis.hsetnx(hashkey, "key", "value1"))
self.assertEquals("value1", self.redis.hget(hashkey, "key"))
self.assertEquals(0, self.redis.hsetnx(hashkey, "key", "value2"))
self.assertEquals("value1", self.redis.hget(hashkey, "key"))
def test_hmset(self):
hashkey = "hash"
self.redis.hmset(hashkey, {"key1": "value1", "key2": "value2"})
self.assertEquals("value1", self.redis.hget(hashkey, "key1"))
self.assertEquals("value2", self.redis.hget(hashkey, "key2"))
def test_hmset_integral(self):
hashkey = "hash"
self.redis.hmset(hashkey, {1: 2, 3: 4})
self.assertEquals("2", self.redis.hget(hashkey, "1"))
self.assertEquals("2", self.redis.hget(hashkey, 1))
self.assertEquals("4", self.redis.hget(hashkey, "3"))
self.assertEquals("4", self.redis.hget(hashkey, 3))
def test_hmget(self):
hashkey = "hash"
self.redis.hmset(hashkey, {1: 2, 3: 4})
self.assertEquals(["2", None, "4"],
self.redis.hmget(hashkey, "1", "2", "3"))
self.assertEquals(["2", None, "4"],
self.redis.hmget(hashkey, ["1", "2", "3"]))
self.assertEquals(["2", None, "4"],
self.redis.hmget(hashkey, [1, 2, 3]))
def test_hincrby(self):
hashkey = "hash"
self.assertEquals(1, self.redis.hincrby(hashkey, "key", 1))
self.assertEquals(3, self.redis.hincrby(hashkey, "key", 2))
self.assertEquals("3", self.redis.hget(hashkey, "key"))
def test_hincrbyfloat(self):
hashkey = "hash"
self.assertEquals(1.2, self.redis.hincrbyfloat(hashkey, "key", 1.2))
self.assertEquals(3.5, self.redis.hincrbyfloat(hashkey, "key", 2.3))
self.assertEquals("3.5", self.redis.hget(hashkey, "key"))
def test_hkeys(self):
hashkey = "hash"
self.redis.hmset(hashkey, {1: 2, 3: 4})
self.assertEquals(["1", "3"], sorted(self.redis.hkeys(hashkey)))
def test_hvals(self):
hashkey = "hash"
self.redis.hmset(hashkey, {1: 2, 3: 4})
self.assertEquals(["2", "4"], sorted(self.redis.hvals(hashkey)))
class TestRedisHash(TestCase):
"""hash tests"""
def setUp(self):
self.redis = MockRedis()
self.redis.flushdb()
def test_hexists(self):
hashkey = "hash"
self.assertFalse(self.redis.hexists(hashkey, "key"))
self.redis.hset(hashkey, "key", "value")
self.assertTrue(self.redis.hexists(hashkey, "key"))
self.assertFalse(self.redis.hexists(hashkey, "key2"))
def test_hgetall(self):
hashkey = "hash"
self.assertEquals({}, self.redis.hgetall(hashkey))
self.redis.hset(hashkey, "key", "value")
self.assertEquals({"key": "value"}, self.redis.hgetall(hashkey))
def test_hdel(self):
hashkey = "hash"
self.redis.hmset(hashkey, {1: 1, 2: 2, 3: 3})
self.assertEquals(0, self.redis.hdel(hashkey, "foo"))
self.assertEquals({"1": "1", "2": "2", "3": "3"}, self.redis.hgetall(hashkey))
self.assertEquals(2, self.redis.hdel(hashkey, "1", 2))
self.assertEquals({"3": "3"}, self.redis.hgetall(hashkey))
self.assertEquals(1, self.redis.hdel(hashkey, "3", 4))
self.assertEquals({}, self.redis.hgetall(hashkey))
self.assertFalse(self.redis.exists(hashkey))
def test_hlen(self):
hashkey = "hash"
self.assertEquals(0, self.redis.hlen(hashkey))
self.redis.hset(hashkey, "key", "value")
self.assertEquals(1, self.redis.hlen(hashkey))
def test_hset(self):
hashkey = "hash"
self.redis.hset(hashkey, "key", "value")
self.assertEquals("value", self.redis.hget(hashkey, "key"))
def test_hget(self):
hashkey = "hash"
self.assertEquals(None, self.redis.hget(hashkey, "key"))
def test_hset_integral(self):
hashkey = "hash"
self.redis.hset(hashkey, 1, 2)
self.assertEquals("2", self.redis.hget(hashkey, 1))
self.assertEquals("2", self.redis.hget(hashkey, "1"))
def test_hsetnx(self):
hashkey = "hash"
self.assertEquals(1, self.redis.hsetnx(hashkey, "key", "value1"))
self.assertEquals("value1", self.redis.hget(hashkey, "key"))
self.assertEquals(0, self.redis.hsetnx(hashkey, "key", "value2"))
self.assertEquals("value1", self.redis.hget(hashkey, "key"))
def test_hmset(self):
hashkey = "hash"
self.redis.hmset(hashkey, {"key1": "value1", "key2": "value2"})
self.assertEquals("value1", self.redis.hget(hashkey, "key1"))
self.assertEquals("value2", self.redis.hget(hashkey, "key2"))
def test_hmset_integral(self):
hashkey = "hash"
self.redis.hmset(hashkey, {1: 2, 3: 4})
self.assertEquals("2", self.redis.hget(hashkey, "1"))
self.assertEquals("2", self.redis.hget(hashkey, 1))
self.assertEquals("4", self.redis.hget(hashkey, "3"))
self.assertEquals("4", self.redis.hget(hashkey, 3))
def test_hmget(self):
hashkey = "hash"
self.redis.hmset(hashkey, {1: 2, 3: 4})
self.assertEquals(["2", None, "4"], self.redis.hmget(hashkey, "1", "2", "3"))
self.assertEquals(["2", None, "4"], self.redis.hmget(hashkey, ["1", "2", "3"]))
self.assertEquals(["2", None, "4"], self.redis.hmget(hashkey, [1, 2, 3]))
def test_hincrby(self):
hashkey = "hash"
self.assertEquals(1, self.redis.hincrby(hashkey, "key", 1))
self.assertEquals(3, self.redis.hincrby(hashkey, "key", 2))
self.assertEquals("3", self.redis.hget(hashkey, "key"))
def test_hincrbyfloat(self):
hashkey = "hash"
self.assertEquals(1.2, self.redis.hincrbyfloat(hashkey, "key", 1.2))
self.assertEquals(3.5, self.redis.hincrbyfloat(hashkey, "key", 2.3))
self.assertEquals("3.5", self.redis.hget(hashkey, "key"))
def test_hkeys(self):
hashkey = "hash"
self.redis.hmset(hashkey, {1: 2, 3: 4})
self.assertEquals(["1", "3"], sorted(self.redis.hkeys(hashkey)))
def test_hvals(self):
hashkey = "hash"
self.redis.hmset(hashkey, {1: 2, 3: 4})
self.assertEquals(["2", "4"], sorted(self.redis.hvals(hashkey)))
