class TestMockRedis(TestCase):
def setUp(self):
self.r = MockRedis()
self.r.zadd('myzlist', 'one', 1)
self.r.zadd('myzlist', 'two', 2)
self.r.zadd('myzlist', 'three', 3)
def tearDown(self):
self.r.data.clear()
def test_zrange(self):
assert (self.r.zrange('myzlist', 0, -1) ==
[(1, 'one'), (2, 'two'), (3, 'three')])
assert self.r.zrange('myzlist', 2, 3) == [(3, 'three')]
assert self.r.zrange('myzlist', -2, -1) == [(2, 'two'), (3, 'three')]
def test_zrevrange(self):
assert (self.r.zrevrange('myzlist', 0, -1) ==
[(3, 'three'), (2, 'two'), (1, 'one')])
assert self.r.zrevrange('myzlist', 2, 3) == [(1, 'one')]
assert self.r.zrevrange('myzlist', -2, -1) == [(2, 'two'), (1, 'one')]
class TestRedisUtil(TestCase):
def setUp(self):
self.r = MockRedis()
self.r_util = Redis('localhost', 6300)
self.r.lpush('test_key', 'something worth keeping')
def tearDown(self):
self.r.data.clear()
def test_pop(self):
"""Test that the redis_util pop wrapper works."""
result = self.r_util.pop('test_key')
assert result == 'something worth keeping'
def test_push(self):
"""Test that the redis_util push wrapper works."""
test_val = 'another thing of note'
self.r_util.push('test_key', test_val)
result = self.r.lpop('test_key')
assert result == test_val
def setUp(self):
self.r = MockRedis()
self.r.zadd('myzlist', 'one', 1)
self.r.zadd('myzlist', 'two', 2)
self.r.zadd('myzlist', 'three', 3)
def setUp(self):
self.r = MockRedis()
self.r_util = Redis('localhost', 6300)
self.r.lpush('test_key', 'something worth keeping')
