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