def setUp(self):
if redis_client is None:
raise SkipTest()
self.client = BatchRedisClient(redis_client)
self.key_prefix = '%s|' % (time.time(),)
class RedisClientTests(TestCase):
def setUp(self):
if redis_client is None:
raise SkipTest()
self.client = BatchRedisClient(redis_client)
self.key_prefix = '%s|' % (time.time(),)
def test_simple_get(self):
@batch_context
def get_thing(t, v):
a = spawn(self.client.delete, self.key_prefix + 'hi' + t)
b = spawn(self.client.set, self.key_prefix + 'hi' + t, v)
c = spawn(self.client.get, self.key_prefix + 'hi' + t)
_, _, result = a.get(), b.get(), c.get()
return int(result)
@batch_context
def test():
a, b = spawn(get_thing, 'a', 1), spawn(get_thing, 'b', 2)
return a.get() + b.get()
self.assertEquals(3, test())
def test_scan(self):
@batch_context
def do_thing(t, v):
k = self.key_prefix + 'hi' + t + '2'
self.client.sadd(k, v)
self.assertEquals([k], self.client.scan(match=self.key_prefix + 'hi' + t + '*', count=1000)[1])
self.assertEquals([v], self.client.sscan(k)[1])
@batch_context
def test():
a = spawn(do_thing, 'a', '1')
b = spawn(do_thing, 'b', '1')
a.get(), b.get()
test()
def test_pipeline(self):
@batch_context
def test():
with self.client.pipeline() as p:
p.set(self.key_prefix + 'yes', '1')
p.set(self.key_prefix + 'yes', '2')
p.get(self.key_prefix + 'yes')
self.assertIsNone(self.client.get(self.key_prefix + 'yes'))
s1, s2, g = p.execute()
self.assertEquals('2', self.client.get(self.key_prefix + 'yes'))
self.assertEquals(1, s1)
self.assertEquals(1, s2)
self.assertEquals('2', g)
test()
