]>
Commit | Line | Data |
---|---|---|
7f7499ee | 1 | start_server {tags {"zset"}} { |
4774a53b PN |
2 | proc create_zset {key items} { |
3 | r del $key | |
4 | foreach {score entry} $items { | |
5 | r zadd $key $score $entry | |
6 | } | |
7 | } | |
8 | ||
9ec4ea20 PN |
9 | proc basics {encoding} { |
10 | if {$encoding == "ziplist"} { | |
11 | r config set zset-max-ziplist-entries 128 | |
12 | r config set zset-max-ziplist-value 64 | |
100ed062 | 13 | } elseif {$encoding == "skiplist"} { |
9ec4ea20 PN |
14 | r config set zset-max-ziplist-entries 0 |
15 | r config set zset-max-ziplist-value 0 | |
16 | } else { | |
17 | puts "Unknown sorted set encoding" | |
18 | exit | |
19 | } | |
184d74ab | 20 | |
9ec4ea20 PN |
21 | test "Check encoding - $encoding" { |
22 | r del ztmp | |
23 | r zadd ztmp 10 x | |
24 | assert_encoding $encoding ztmp | |
25 | } | |
184d74ab | 26 | |
9ec4ea20 PN |
27 | test "ZSET basic ZADD and score update - $encoding" { |
28 | r del ztmp | |
29 | r zadd ztmp 10 x | |
30 | r zadd ztmp 20 y | |
31 | r zadd ztmp 30 z | |
32 | assert_equal {x y z} [r zrange ztmp 0 -1] | |
33 | ||
34 | r zadd ztmp 1 y | |
35 | assert_equal {y x z} [r zrange ztmp 0 -1] | |
98578b57 | 36 | } |
9ec4ea20 PN |
37 | |
38 | test "ZSET element can't be set to NaN with ZADD - $encoding" { | |
39 | assert_error "*not a double*" {r zadd myzset nan abc} | |
98578b57 | 40 | } |
98578b57 | 41 | |
9ec4ea20 PN |
42 | test "ZSET element can't be set to NaN with ZINCRBY" { |
43 | assert_error "*not a double*" {r zadd myzset nan abc} | |
44 | } | |
45 | ||
46 | test "ZINCRBY calls leading to NaN result in error" { | |
47 | r zincrby myzset +inf abc | |
48 | assert_error "*NaN*" {r zincrby myzset -inf abc} | |
49 | } | |
50 | ||
51 | test "ZCARD basics - $encoding" { | |
52 | assert_equal 3 [r zcard ztmp] | |
53 | assert_equal 0 [r zcard zdoesntexist] | |
54 | } | |
55 | ||
04a10b1a PN |
56 | test "ZREM removes key after last element is removed" { |
57 | r del ztmp | |
58 | r zadd ztmp 10 x | |
59 | r zadd ztmp 20 y | |
60 | ||
61 | assert_equal 1 [r exists ztmp] | |
62 | assert_equal 0 [r zrem ztmp z] | |
63 | assert_equal 1 [r zrem ztmp y] | |
64 | assert_equal 1 [r zrem ztmp x] | |
65 | assert_equal 0 [r exists ztmp] | |
66 | } | |
67 | ||
9ec4ea20 PN |
68 | test "ZRANGE basics - $encoding" { |
69 | r del ztmp | |
70 | r zadd ztmp 1 a | |
71 | r zadd ztmp 2 b | |
72 | r zadd ztmp 3 c | |
73 | r zadd ztmp 4 d | |
74 | ||
75 | assert_equal {a b c d} [r zrange ztmp 0 -1] | |
76 | assert_equal {a b c} [r zrange ztmp 0 -2] | |
77 | assert_equal {b c d} [r zrange ztmp 1 -1] | |
78 | assert_equal {b c} [r zrange ztmp 1 -2] | |
79 | assert_equal {c d} [r zrange ztmp -2 -1] | |
80 | assert_equal {c} [r zrange ztmp -2 -2] | |
81 | ||
82 | # out of range start index | |
83 | assert_equal {a b c} [r zrange ztmp -5 2] | |
84 | assert_equal {a b} [r zrange ztmp -5 1] | |
85 | assert_equal {} [r zrange ztmp 5 -1] | |
86 | assert_equal {} [r zrange ztmp 5 -2] | |
87 | ||
88 | # out of range end index | |
89 | assert_equal {a b c d} [r zrange ztmp 0 5] | |
90 | assert_equal {b c d} [r zrange ztmp 1 5] | |
91 | assert_equal {} [r zrange ztmp 0 -5] | |
92 | assert_equal {} [r zrange ztmp 1 -5] | |
93 | ||
94 | # withscores | |
95 | assert_equal {a 1 b 2 c 3 d 4} [r zrange ztmp 0 -1 withscores] | |
96 | } | |
97 | ||
98 | test "ZREVRANGE basics - $encoding" { | |
99 | r del ztmp | |
100 | r zadd ztmp 1 a | |
101 | r zadd ztmp 2 b | |
102 | r zadd ztmp 3 c | |
103 | r zadd ztmp 4 d | |
104 | ||
105 | assert_equal {d c b a} [r zrevrange ztmp 0 -1] | |
106 | assert_equal {d c b} [r zrevrange ztmp 0 -2] | |
107 | assert_equal {c b a} [r zrevrange ztmp 1 -1] | |
108 | assert_equal {c b} [r zrevrange ztmp 1 -2] | |
109 | assert_equal {b a} [r zrevrange ztmp -2 -1] | |
110 | assert_equal {b} [r zrevrange ztmp -2 -2] | |
111 | ||
112 | # out of range start index | |
113 | assert_equal {d c b} [r zrevrange ztmp -5 2] | |
114 | assert_equal {d c} [r zrevrange ztmp -5 1] | |
115 | assert_equal {} [r zrevrange ztmp 5 -1] | |
116 | assert_equal {} [r zrevrange ztmp 5 -2] | |
117 | ||
118 | # out of range end index | |
119 | assert_equal {d c b a} [r zrevrange ztmp 0 5] | |
120 | assert_equal {c b a} [r zrevrange ztmp 1 5] | |
121 | assert_equal {} [r zrevrange ztmp 0 -5] | |
122 | assert_equal {} [r zrevrange ztmp 1 -5] | |
123 | ||
124 | # withscores | |
125 | assert_equal {d 4 c 3 b 2 a 1} [r zrevrange ztmp 0 -1 withscores] | |
126 | } | |
127 | ||
128 | test "ZRANK/ZREVRANK basics - $encoding" { | |
129 | r del zranktmp | |
130 | r zadd zranktmp 10 x | |
131 | r zadd zranktmp 20 y | |
132 | r zadd zranktmp 30 z | |
133 | assert_equal 0 [r zrank zranktmp x] | |
134 | assert_equal 1 [r zrank zranktmp y] | |
135 | assert_equal 2 [r zrank zranktmp z] | |
136 | assert_equal "" [r zrank zranktmp foo] | |
137 | assert_equal 2 [r zrevrank zranktmp x] | |
138 | assert_equal 1 [r zrevrank zranktmp y] | |
139 | assert_equal 0 [r zrevrank zranktmp z] | |
140 | assert_equal "" [r zrevrank zranktmp foo] | |
141 | } | |
142 | ||
143 | test "ZRANK - after deletion - $encoding" { | |
144 | r zrem zranktmp y | |
145 | assert_equal 0 [r zrank zranktmp x] | |
146 | assert_equal 1 [r zrank zranktmp z] | |
147 | } | |
148 | ||
149 | test "ZINCRBY - can create a new sorted set - $encoding" { | |
150 | r del zset | |
151 | r zincrby zset 1 foo | |
152 | assert_equal {foo} [r zrange zset 0 -1] | |
153 | assert_equal 1 [r zscore zset foo] | |
154 | } | |
155 | ||
156 | test "ZINCRBY - increment and decrement - $encoding" { | |
157 | r zincrby zset 2 foo | |
158 | r zincrby zset 1 bar | |
159 | assert_equal {bar foo} [r zrange zset 0 -1] | |
160 | ||
161 | r zincrby zset 10 bar | |
162 | r zincrby zset -5 foo | |
163 | r zincrby zset -5 bar | |
164 | assert_equal {foo bar} [r zrange zset 0 -1] | |
165 | ||
166 | assert_equal -2 [r zscore zset foo] | |
167 | assert_equal 6 [r zscore zset bar] | |
168 | } | |
169 | ||
170 | proc create_default_zset {} { | |
171 | create_zset zset {-inf a 1 b 2 c 3 d 4 e 5 f +inf g} | |
172 | } | |
173 | ||
174 | test "ZRANGEBYSCORE/ZREVRANGEBYSCORE/ZCOUNT basics" { | |
175 | create_default_zset | |
176 | ||
177 | # inclusive range | |
178 | assert_equal {a b c} [r zrangebyscore zset -inf 2] | |
179 | assert_equal {b c d} [r zrangebyscore zset 0 3] | |
180 | assert_equal {d e f} [r zrangebyscore zset 3 6] | |
181 | assert_equal {e f g} [r zrangebyscore zset 4 +inf] | |
182 | assert_equal {c b a} [r zrevrangebyscore zset 2 -inf] | |
183 | assert_equal {d c b} [r zrevrangebyscore zset 3 0] | |
184 | assert_equal {f e d} [r zrevrangebyscore zset 6 3] | |
185 | assert_equal {g f e} [r zrevrangebyscore zset +inf 4] | |
186 | assert_equal 3 [r zcount zset 0 3] | |
187 | ||
188 | # exclusive range | |
189 | assert_equal {b} [r zrangebyscore zset (-inf (2] | |
190 | assert_equal {b c} [r zrangebyscore zset (0 (3] | |
191 | assert_equal {e f} [r zrangebyscore zset (3 (6] | |
192 | assert_equal {f} [r zrangebyscore zset (4 (+inf] | |
193 | assert_equal {b} [r zrevrangebyscore zset (2 (-inf] | |
194 | assert_equal {c b} [r zrevrangebyscore zset (3 (0] | |
195 | assert_equal {f e} [r zrevrangebyscore zset (6 (3] | |
196 | assert_equal {f} [r zrevrangebyscore zset (+inf (4] | |
197 | assert_equal 2 [r zcount zset (0 (3] | |
198 | ||
199 | # test empty ranges | |
200 | r zrem zset a | |
201 | r zrem zset g | |
202 | ||
203 | # inclusive | |
204 | assert_equal {} [r zrangebyscore zset 4 2] | |
205 | assert_equal {} [r zrangebyscore zset 6 +inf] | |
206 | assert_equal {} [r zrangebyscore zset -inf -6] | |
207 | assert_equal {} [r zrevrangebyscore zset +inf 6] | |
208 | assert_equal {} [r zrevrangebyscore zset -6 -inf] | |
209 | ||
210 | # exclusive | |
211 | assert_equal {} [r zrangebyscore zset (4 (2] | |
212 | assert_equal {} [r zrangebyscore zset 2 (2] | |
213 | assert_equal {} [r zrangebyscore zset (2 2] | |
214 | assert_equal {} [r zrangebyscore zset (6 (+inf] | |
215 | assert_equal {} [r zrangebyscore zset (-inf (-6] | |
216 | assert_equal {} [r zrevrangebyscore zset (+inf (6] | |
217 | assert_equal {} [r zrevrangebyscore zset (-6 (-inf] | |
218 | ||
219 | # empty inner range | |
220 | assert_equal {} [r zrangebyscore zset 2.4 2.6] | |
221 | assert_equal {} [r zrangebyscore zset (2.4 2.6] | |
222 | assert_equal {} [r zrangebyscore zset 2.4 (2.6] | |
223 | assert_equal {} [r zrangebyscore zset (2.4 (2.6] | |
224 | } | |
225 | ||
226 | test "ZRANGEBYSCORE with WITHSCORES" { | |
227 | create_default_zset | |
228 | assert_equal {b 1 c 2 d 3} [r zrangebyscore zset 0 3 withscores] | |
229 | assert_equal {d 3 c 2 b 1} [r zrevrangebyscore zset 3 0 withscores] | |
230 | } | |
231 | ||
232 | test "ZRANGEBYSCORE with LIMIT" { | |
233 | create_default_zset | |
234 | assert_equal {b c} [r zrangebyscore zset 0 10 LIMIT 0 2] | |
235 | assert_equal {d e f} [r zrangebyscore zset 0 10 LIMIT 2 3] | |
236 | assert_equal {d e f} [r zrangebyscore zset 0 10 LIMIT 2 10] | |
237 | assert_equal {} [r zrangebyscore zset 0 10 LIMIT 20 10] | |
238 | assert_equal {f e} [r zrevrangebyscore zset 10 0 LIMIT 0 2] | |
239 | assert_equal {d c b} [r zrevrangebyscore zset 10 0 LIMIT 2 3] | |
240 | assert_equal {d c b} [r zrevrangebyscore zset 10 0 LIMIT 2 10] | |
241 | assert_equal {} [r zrevrangebyscore zset 10 0 LIMIT 20 10] | |
242 | } | |
243 | ||
244 | test "ZRANGEBYSCORE with LIMIT and WITHSCORES" { | |
245 | create_default_zset | |
246 | assert_equal {e 4 f 5} [r zrangebyscore zset 2 5 LIMIT 2 3 WITHSCORES] | |
247 | assert_equal {d 3 c 2} [r zrevrangebyscore zset 5 2 LIMIT 2 3 WITHSCORES] | |
248 | } | |
249 | ||
250 | test "ZRANGEBYSCORE with non-value min or max" { | |
251 | assert_error "*not a double*" {r zrangebyscore fooz str 1} | |
252 | assert_error "*not a double*" {r zrangebyscore fooz 1 str} | |
253 | assert_error "*not a double*" {r zrangebyscore fooz 1 NaN} | |
254 | } | |
255 | ||
256 | test "ZREMRANGEBYSCORE basics" { | |
257 | proc remrangebyscore {min max} { | |
258 | create_zset zset {1 a 2 b 3 c 4 d 5 e} | |
04a10b1a | 259 | assert_equal 1 [r exists zset] |
9ec4ea20 | 260 | r zremrangebyscore zset $min $max |
98578b57 | 261 | } |
9ec4ea20 PN |
262 | |
263 | # inner range | |
264 | assert_equal 3 [remrangebyscore 2 4] | |
265 | assert_equal {a e} [r zrange zset 0 -1] | |
266 | ||
267 | # start underflow | |
268 | assert_equal 1 [remrangebyscore -10 1] | |
269 | assert_equal {b c d e} [r zrange zset 0 -1] | |
270 | ||
271 | # end overflow | |
272 | assert_equal 1 [remrangebyscore 5 10] | |
273 | assert_equal {a b c d} [r zrange zset 0 -1] | |
274 | ||
275 | # switch min and max | |
276 | assert_equal 0 [remrangebyscore 4 2] | |
277 | assert_equal {a b c d e} [r zrange zset 0 -1] | |
278 | ||
279 | # -inf to mid | |
280 | assert_equal 3 [remrangebyscore -inf 3] | |
281 | assert_equal {d e} [r zrange zset 0 -1] | |
282 | ||
283 | # mid to +inf | |
284 | assert_equal 3 [remrangebyscore 3 +inf] | |
285 | assert_equal {a b} [r zrange zset 0 -1] | |
286 | ||
287 | # -inf to +inf | |
288 | assert_equal 5 [remrangebyscore -inf +inf] | |
289 | assert_equal {} [r zrange zset 0 -1] | |
290 | ||
291 | # exclusive min | |
292 | assert_equal 4 [remrangebyscore (1 5] | |
293 | assert_equal {a} [r zrange zset 0 -1] | |
294 | assert_equal 3 [remrangebyscore (2 5] | |
295 | assert_equal {a b} [r zrange zset 0 -1] | |
296 | ||
297 | # exclusive max | |
298 | assert_equal 4 [remrangebyscore 1 (5] | |
299 | assert_equal {e} [r zrange zset 0 -1] | |
300 | assert_equal 3 [remrangebyscore 1 (4] | |
301 | assert_equal {d e} [r zrange zset 0 -1] | |
302 | ||
303 | # exclusive min and max | |
304 | assert_equal 3 [remrangebyscore (1 (5] | |
305 | assert_equal {a e} [r zrange zset 0 -1] | |
04a10b1a PN |
306 | |
307 | # destroy when empty | |
308 | assert_equal 5 [remrangebyscore 1 5] | |
309 | assert_equal 0 [r exists zset] | |
9ec4ea20 PN |
310 | } |
311 | ||
312 | test "ZREMRANGEBYSCORE with non-value min or max" { | |
313 | assert_error "*not a double*" {r zremrangebyscore fooz str 1} | |
314 | assert_error "*not a double*" {r zremrangebyscore fooz 1 str} | |
315 | assert_error "*not a double*" {r zremrangebyscore fooz 1 NaN} | |
316 | } | |
317 | ||
318 | test "ZREMRANGEBYRANK basics" { | |
319 | proc remrangebyrank {min max} { | |
320 | create_zset zset {1 a 2 b 3 c 4 d 5 e} | |
04a10b1a | 321 | assert_equal 1 [r exists zset] |
9ec4ea20 | 322 | r zremrangebyrank zset $min $max |
98578b57 | 323 | } |
9ec4ea20 PN |
324 | |
325 | # inner range | |
326 | assert_equal 3 [remrangebyrank 1 3] | |
327 | assert_equal {a e} [r zrange zset 0 -1] | |
328 | ||
329 | # start underflow | |
330 | assert_equal 1 [remrangebyrank -10 0] | |
331 | assert_equal {b c d e} [r zrange zset 0 -1] | |
332 | ||
333 | # start overflow | |
334 | assert_equal 0 [remrangebyrank 10 -1] | |
335 | assert_equal {a b c d e} [r zrange zset 0 -1] | |
336 | ||
337 | # end underflow | |
338 | assert_equal 0 [remrangebyrank 0 -10] | |
339 | assert_equal {a b c d e} [r zrange zset 0 -1] | |
340 | ||
341 | # end overflow | |
342 | assert_equal 5 [remrangebyrank 0 10] | |
343 | assert_equal {} [r zrange zset 0 -1] | |
04a10b1a PN |
344 | |
345 | # destroy when empty | |
346 | assert_equal 5 [remrangebyrank 0 4] | |
347 | assert_equal 0 [r exists zset] | |
9ec4ea20 PN |
348 | } |
349 | ||
350 | test "ZUNIONSTORE against non-existing key doesn't set destination - $encoding" { | |
351 | r del zseta | |
352 | assert_equal 0 [r zunionstore dst_key 1 zseta] | |
353 | assert_equal 0 [r exists dst_key] | |
354 | } | |
355 | ||
356 | test "ZUNIONSTORE basics - $encoding" { | |
357 | r del zseta zsetb zsetc | |
358 | r zadd zseta 1 a | |
359 | r zadd zseta 2 b | |
360 | r zadd zseta 3 c | |
361 | r zadd zsetb 1 b | |
362 | r zadd zsetb 2 c | |
363 | r zadd zsetb 3 d | |
364 | ||
365 | assert_equal 4 [r zunionstore zsetc 2 zseta zsetb] | |
366 | assert_equal {a 1 b 3 d 3 c 5} [r zrange zsetc 0 -1 withscores] | |
367 | } | |
368 | ||
369 | test "ZUNIONSTORE with weights - $encoding" { | |
370 | assert_equal 4 [r zunionstore zsetc 2 zseta zsetb weights 2 3] | |
371 | assert_equal {a 2 b 7 d 9 c 12} [r zrange zsetc 0 -1 withscores] | |
372 | } | |
373 | ||
374 | test "ZUNIONSTORE with a regular set and weights - $encoding" { | |
375 | r del seta | |
376 | r sadd seta a | |
377 | r sadd seta b | |
378 | r sadd seta c | |
379 | ||
380 | assert_equal 4 [r zunionstore zsetc 2 seta zsetb weights 2 3] | |
381 | assert_equal {a 2 b 5 c 8 d 9} [r zrange zsetc 0 -1 withscores] | |
382 | } | |
383 | ||
384 | test "ZUNIONSTORE with AGGREGATE MIN - $encoding" { | |
385 | assert_equal 4 [r zunionstore zsetc 2 zseta zsetb aggregate min] | |
386 | assert_equal {a 1 b 1 c 2 d 3} [r zrange zsetc 0 -1 withscores] | |
387 | } | |
388 | ||
389 | test "ZUNIONSTORE with AGGREGATE MAX - $encoding" { | |
390 | assert_equal 4 [r zunionstore zsetc 2 zseta zsetb aggregate max] | |
391 | assert_equal {a 1 b 2 c 3 d 3} [r zrange zsetc 0 -1 withscores] | |
392 | } | |
393 | ||
394 | test "ZINTERSTORE basics - $encoding" { | |
395 | assert_equal 2 [r zinterstore zsetc 2 zseta zsetb] | |
396 | assert_equal {b 3 c 5} [r zrange zsetc 0 -1 withscores] | |
397 | } | |
398 | ||
399 | test "ZINTERSTORE with weights - $encoding" { | |
400 | assert_equal 2 [r zinterstore zsetc 2 zseta zsetb weights 2 3] | |
401 | assert_equal {b 7 c 12} [r zrange zsetc 0 -1 withscores] | |
402 | } | |
403 | ||
404 | test "ZINTERSTORE with a regular set and weights - $encoding" { | |
405 | r del seta | |
406 | r sadd seta a | |
407 | r sadd seta b | |
408 | r sadd seta c | |
409 | assert_equal 2 [r zinterstore zsetc 2 seta zsetb weights 2 3] | |
410 | assert_equal {b 5 c 8} [r zrange zsetc 0 -1 withscores] | |
411 | } | |
412 | ||
413 | test "ZINTERSTORE with AGGREGATE MIN - $encoding" { | |
414 | assert_equal 2 [r zinterstore zsetc 2 zseta zsetb aggregate min] | |
415 | assert_equal {b 1 c 2} [r zrange zsetc 0 -1 withscores] | |
416 | } | |
417 | ||
418 | test "ZINTERSTORE with AGGREGATE MAX - $encoding" { | |
419 | assert_equal 2 [r zinterstore zsetc 2 zseta zsetb aggregate max] | |
420 | assert_equal {b 2 c 3} [r zrange zsetc 0 -1 withscores] | |
421 | } | |
422 | ||
423 | foreach cmd {ZUNIONSTORE ZINTERSTORE} { | |
424 | test "$cmd with +inf/-inf scores - $encoding" { | |
425 | r del zsetinf1 zsetinf2 | |
426 | ||
427 | r zadd zsetinf1 +inf key | |
428 | r zadd zsetinf2 +inf key | |
429 | r $cmd zsetinf3 2 zsetinf1 zsetinf2 | |
430 | assert_equal inf [r zscore zsetinf3 key] | |
431 | ||
432 | r zadd zsetinf1 -inf key | |
433 | r zadd zsetinf2 +inf key | |
434 | r $cmd zsetinf3 2 zsetinf1 zsetinf2 | |
435 | assert_equal 0 [r zscore zsetinf3 key] | |
436 | ||
437 | r zadd zsetinf1 +inf key | |
438 | r zadd zsetinf2 -inf key | |
439 | r $cmd zsetinf3 2 zsetinf1 zsetinf2 | |
440 | assert_equal 0 [r zscore zsetinf3 key] | |
441 | ||
442 | r zadd zsetinf1 -inf key | |
443 | r zadd zsetinf2 -inf key | |
444 | r $cmd zsetinf3 2 zsetinf1 zsetinf2 | |
445 | assert_equal -inf [r zscore zsetinf3 key] | |
98578b57 | 446 | } |
9ec4ea20 PN |
447 | |
448 | test "$cmd with NaN weights $encoding" { | |
449 | r del zsetinf1 zsetinf2 | |
450 | ||
451 | r zadd zsetinf1 1.0 key | |
452 | r zadd zsetinf2 1.0 key | |
453 | assert_error "*weight value is not a double*" { | |
454 | r $cmd zsetinf3 2 zsetinf1 zsetinf2 weights nan nan | |
98578b57 PN |
455 | } |
456 | } | |
457 | } | |
25bb8a44 | 458 | } |
98578b57 | 459 | |
9ec4ea20 | 460 | basics ziplist |
100ed062 | 461 | basics skiplist |
9ec4ea20 | 462 | |
af9aed25 | 463 | test {ZINTERSTORE regression with two sets, intset+hashtable} { |
464 | r del seta setb setc | |
465 | r sadd set1 a | |
466 | r sadd set2 10 | |
467 | r zinterstore set3 2 set1 set2 | |
468 | } {0} | |
469 | ||
9ec4ea20 PN |
470 | proc stressers {encoding} { |
471 | if {$encoding == "ziplist"} { | |
472 | # Little extra to allow proper fuzzing in the sorting stresser | |
473 | r config set zset-max-ziplist-entries 256 | |
474 | r config set zset-max-ziplist-value 64 | |
475 | set elements 128 | |
100ed062 | 476 | } elseif {$encoding == "skiplist"} { |
9ec4ea20 PN |
477 | r config set zset-max-ziplist-entries 0 |
478 | r config set zset-max-ziplist-value 0 | |
479 | set elements 1000 | |
480 | } else { | |
481 | puts "Unknown sorted set encoding" | |
482 | exit | |
483 | } | |
25bb8a44 | 484 | |
9ec4ea20 PN |
485 | test "ZSCORE - $encoding" { |
486 | r del zscoretest | |
487 | set aux {} | |
488 | for {set i 0} {$i < $elements} {incr i} { | |
489 | set score [expr rand()] | |
490 | lappend aux $score | |
491 | r zadd zscoretest $score $i | |
492 | } | |
25bb8a44 | 493 | |
9ec4ea20 PN |
494 | assert_encoding $encoding zscoretest |
495 | for {set i 0} {$i < $elements} {incr i} { | |
496 | assert_equal [lindex $aux $i] [r zscore zscoretest $i] | |
497 | } | |
498 | } | |
25bb8a44 | 499 | |
9ec4ea20 PN |
500 | test "ZSCORE after a DEBUG RELOAD - $encoding" { |
501 | r del zscoretest | |
502 | set aux {} | |
503 | for {set i 0} {$i < $elements} {incr i} { | |
504 | set score [expr rand()] | |
505 | lappend aux $score | |
506 | r zadd zscoretest $score $i | |
507 | } | |
98578b57 | 508 | |
9ec4ea20 PN |
509 | r debug reload |
510 | assert_encoding $encoding zscoretest | |
511 | for {set i 0} {$i < $elements} {incr i} { | |
512 | assert_equal [lindex $aux $i] [r zscore zscoretest $i] | |
513 | } | |
514 | } | |
7236fdb2 | 515 | |
9ec4ea20 PN |
516 | test "ZSET sorting stresser - $encoding" { |
517 | set delta 0 | |
518 | for {set test 0} {$test < 2} {incr test} { | |
519 | unset -nocomplain auxarray | |
520 | array set auxarray {} | |
521 | set auxlist {} | |
522 | r del myzset | |
523 | for {set i 0} {$i < $elements} {incr i} { | |
524 | if {$test == 0} { | |
525 | set score [expr rand()] | |
526 | } else { | |
527 | set score [expr int(rand()*10)] | |
528 | } | |
529 | set auxarray($i) $score | |
530 | r zadd myzset $score $i | |
531 | # Random update | |
532 | if {[expr rand()] < .2} { | |
533 | set j [expr int(rand()*1000)] | |
534 | if {$test == 0} { | |
535 | set score [expr rand()] | |
536 | } else { | |
537 | set score [expr int(rand()*10)] | |
538 | } | |
539 | set auxarray($j) $score | |
540 | r zadd myzset $score $j | |
541 | } | |
542 | } | |
543 | foreach {item score} [array get auxarray] { | |
544 | lappend auxlist [list $score $item] | |
545 | } | |
546 | set sorted [lsort -command zlistAlikeSort $auxlist] | |
547 | set auxlist {} | |
548 | foreach x $sorted { | |
549 | lappend auxlist [lindex $x 1] | |
550 | } | |
551 | ||
552 | assert_encoding $encoding myzset | |
553 | set fromredis [r zrange myzset 0 -1] | |
554 | set delta 0 | |
555 | for {set i 0} {$i < [llength $fromredis]} {incr i} { | |
556 | if {[lindex $fromredis $i] != [lindex $auxlist $i]} { | |
557 | incr delta | |
558 | } | |
559 | } | |
560 | } | |
561 | assert_equal 0 $delta | |
562 | } | |
563 | ||
564 | test "ZRANGEBYSCORE fuzzy test, 100 ranges in $elements element sorted set - $encoding" { | |
7f7499ee PN |
565 | set err {} |
566 | r del zset | |
9ec4ea20 | 567 | for {set i 0} {$i < $elements} {incr i} { |
7f7499ee | 568 | r zadd zset [expr rand()] $i |
98578b57 | 569 | } |
9ec4ea20 PN |
570 | |
571 | assert_encoding $encoding zset | |
7f7499ee PN |
572 | for {set i 0} {$i < 100} {incr i} { |
573 | set min [expr rand()] | |
574 | set max [expr rand()] | |
575 | if {$min > $max} { | |
576 | set aux $min | |
577 | set min $max | |
578 | set max $aux | |
579 | } | |
580 | set low [r zrangebyscore zset -inf $min] | |
581 | set ok [r zrangebyscore zset $min $max] | |
582 | set high [r zrangebyscore zset $max +inf] | |
583 | set lowx [r zrangebyscore zset -inf ($min] | |
584 | set okx [r zrangebyscore zset ($min ($max] | |
585 | set highx [r zrangebyscore zset ($max +inf] | |
586 | ||
587 | if {[r zcount zset -inf $min] != [llength $low]} { | |
588 | append err "Error, len does not match zcount\n" | |
589 | } | |
590 | if {[r zcount zset $min $max] != [llength $ok]} { | |
591 | append err "Error, len does not match zcount\n" | |
592 | } | |
593 | if {[r zcount zset $max +inf] != [llength $high]} { | |
594 | append err "Error, len does not match zcount\n" | |
595 | } | |
596 | if {[r zcount zset -inf ($min] != [llength $lowx]} { | |
597 | append err "Error, len does not match zcount\n" | |
598 | } | |
599 | if {[r zcount zset ($min ($max] != [llength $okx]} { | |
600 | append err "Error, len does not match zcount\n" | |
601 | } | |
602 | if {[r zcount zset ($max +inf] != [llength $highx]} { | |
603 | append err "Error, len does not match zcount\n" | |
604 | } | |
98578b57 | 605 | |
7f7499ee PN |
606 | foreach x $low { |
607 | set score [r zscore zset $x] | |
608 | if {$score > $min} { | |
609 | append err "Error, score for $x is $score > $min\n" | |
610 | } | |
98578b57 | 611 | } |
7f7499ee PN |
612 | foreach x $lowx { |
613 | set score [r zscore zset $x] | |
614 | if {$score >= $min} { | |
615 | append err "Error, score for $x is $score >= $min\n" | |
616 | } | |
98578b57 | 617 | } |
7f7499ee PN |
618 | foreach x $ok { |
619 | set score [r zscore zset $x] | |
620 | if {$score < $min || $score > $max} { | |
621 | append err "Error, score for $x is $score outside $min-$max range\n" | |
622 | } | |
98578b57 | 623 | } |
7f7499ee PN |
624 | foreach x $okx { |
625 | set score [r zscore zset $x] | |
626 | if {$score <= $min || $score >= $max} { | |
627 | append err "Error, score for $x is $score outside $min-$max open range\n" | |
628 | } | |
98578b57 | 629 | } |
7f7499ee PN |
630 | foreach x $high { |
631 | set score [r zscore zset $x] | |
632 | if {$score < $max} { | |
633 | append err "Error, score for $x is $score < $max\n" | |
634 | } | |
98578b57 | 635 | } |
7f7499ee PN |
636 | foreach x $highx { |
637 | set score [r zscore zset $x] | |
638 | if {$score <= $max} { | |
639 | append err "Error, score for $x is $score <= $max\n" | |
640 | } | |
98578b57 PN |
641 | } |
642 | } | |
9ec4ea20 | 643 | assert_equal {} $err |
4774a53b PN |
644 | } |
645 | ||
9ec4ea20 | 646 | test "ZSETs skiplist implementation backlink consistency test - $encoding" { |
7f7499ee | 647 | set diff 0 |
7f7499ee PN |
648 | for {set j 0} {$j < $elements} {incr j} { |
649 | r zadd myzset [expr rand()] "Element-$j" | |
650 | r zrem myzset "Element-[expr int(rand()*$elements)]" | |
98578b57 | 651 | } |
9ec4ea20 PN |
652 | |
653 | assert_encoding $encoding myzset | |
7f7499ee PN |
654 | set l1 [r zrange myzset 0 -1] |
655 | set l2 [r zrevrange myzset 0 -1] | |
656 | for {set j 0} {$j < [llength $l1]} {incr j} { | |
657 | if {[lindex $l1 $j] ne [lindex $l2 end-$j]} { | |
658 | incr diff | |
659 | } | |
98578b57 | 660 | } |
9ec4ea20 PN |
661 | assert_equal 0 $diff |
662 | } | |
7f7499ee | 663 | |
9ec4ea20 | 664 | test "ZSETs ZRANK augmented skip list stress testing - $encoding" { |
7f7499ee PN |
665 | set err {} |
666 | r del myzset | |
9ec4ea20 PN |
667 | for {set k 0} {$k < 2000} {incr k} { |
668 | set i [expr {$k % $elements}] | |
7f7499ee PN |
669 | if {[expr rand()] < .2} { |
670 | r zrem myzset $i | |
671 | } else { | |
672 | set score [expr rand()] | |
673 | r zadd myzset $score $i | |
9ec4ea20 | 674 | assert_encoding $encoding myzset |
7f7499ee | 675 | } |
9ec4ea20 | 676 | |
7f7499ee PN |
677 | set card [r zcard myzset] |
678 | if {$card > 0} { | |
679 | set index [randomInt $card] | |
680 | set ele [lindex [r zrange myzset $index $index] 0] | |
681 | set rank [r zrank myzset $ele] | |
682 | if {$rank != $index} { | |
683 | set err "$ele RANK is wrong! ($rank != $index)" | |
684 | break | |
685 | } | |
98578b57 PN |
686 | } |
687 | } | |
9ec4ea20 PN |
688 | assert_equal {} $err |
689 | } | |
673e1fb7 | 690 | } |
5fc9229c | 691 | |
9ec4ea20 PN |
692 | tags {"slow"} { |
693 | stressers ziplist | |
100ed062 | 694 | stressers skiplist |
673e1fb7 | 695 | } |
98578b57 | 696 | } |