1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
|
#!/usr/bin/env python
#coding: utf-8
from common import *
def test_linsert():
r = getconn()
r.rpush('mylist', 'Hello')
r.rpush('mylist', 'World')
r.linsert('mylist', 'BEFORE', 'World', 'There')
rst = r.lrange('mylist', 0, -1)
assert(rst == ['Hello', 'There', 'World'])
def test_lpush_lrange():
r = getconn()
vals = ['vvv-%s' % i for i in range(10) ]
assert([] == r.lrange('mylist', 0, -1))
r.lpush('mylist', *vals)
rst = r.lrange('mylist', 0, -1)
assert(10 == len(rst))
def test_hscan():
r = getconn()
kv = {'kkk-%s' % i : 'vvv-%s' % i for i in range(10)}
r.hmset('a', kv)
cursor, dic = r.hscan('a')
assert(str(cursor) == '0')
assert(dic == kv)
cursor, dic = r.hscan('a', match='kkk-5')
assert(str(cursor) == '0')
assert(dic == {'kkk-5': 'vvv-5'})
def test_hscan_large():
r = getconn()
kv = {'x'* 100 + 'kkk-%s' % i : 'vvv-%s' % i for i in range(1000)}
r.hmset('a', kv)
cursor = '0'
dic = {}
while True:
cursor, t = r.hscan('a', cursor, count=10)
for k, v in t.items():
dic[k] = v
if '0' == str(cursor):
break
assert(dic == kv)
cursor, dic = r.hscan('a', '0', match='*kkk-5*', count=1000)
if str(cursor) == '0':
assert(len(dic) == 111)
else:
assert(len(dic) == 111)
#again.
cursor, dic = r.hscan('a', cursor, match='*kkk-5*', count=1000)
assert(str(cursor) == '0')
assert(len(dic) == 0)
def test_zscan():
r = getconn()
r.zadd('a', 'a', 1, 'b', 2, 'c', 3)
cursor, pairs = r.zscan('a')
assert(str(cursor) == '0')
assert(set(pairs) == set([('a', 1), ('b', 2), ('c', 3)]))
cursor, pairs = r.zscan('a', match='a')
assert(str(cursor) == '0')
assert(set(pairs) == set([('a', 1)]))
def test_sscan():
r = getconn()
r.sadd('a', 1, 2, 3)
cursor, members = r.sscan('a')
assert(str(cursor) == '0')
assert(set(members) == set(['1', '2', '3']))
cursor, members = r.sscan('a', match='1')
assert(str(cursor) == '0')
assert(set(members) == set(['1']))
|
