1 start_server
{tags
{"zset"}} {
2 proc create_zset
{key items
} {
4 foreach {score
entry} $items {
5 r zadd
$key $score $entry
9 test
{ZSET basic ZADD and score
update} {
13 set aux1
[r zrange ztmp
0 -1]
15 set aux2
[r zrange ztmp
0 -1]
23 test
{ZCARD non existing key
} {
27 test
"ZRANGE basics" {
34 assert_equal
{a b c d
} [r zrange ztmp
0 -1]
35 assert_equal
{a b c
} [r zrange ztmp
0 -2]
36 assert_equal
{b c d
} [r zrange ztmp
1 -1]
37 assert_equal
{b c
} [r zrange ztmp
1 -2]
38 assert_equal
{c d
} [r zrange ztmp
-2 -1]
39 assert_equal
{c
} [r zrange ztmp
-2 -2]
41 # out of range start index
42 assert_equal
{a b c
} [r zrange ztmp
-5 2]
43 assert_equal
{a b
} [r zrange ztmp
-5 1]
44 assert_equal
{} [r zrange ztmp
5 -1]
45 assert_equal
{} [r zrange ztmp
5 -2]
47 # out of range end index
48 assert_equal
{a b c d
} [r zrange ztmp
0 5]
49 assert_equal
{b c d
} [r zrange ztmp
1 5]
50 assert_equal
{} [r zrange ztmp
0 -5]
51 assert_equal
{} [r zrange ztmp
1 -5]
54 assert_equal
{a
1 b
2 c
3 d
4} [r zrange ztmp
0 -1 withscores
]
57 test
"ZREVRANGE basics" {
64 assert_equal
{d c b a
} [r zrevrange ztmp
0 -1]
65 assert_equal
{d c b
} [r zrevrange ztmp
0 -2]
66 assert_equal
{c b a
} [r zrevrange ztmp
1 -1]
67 assert_equal
{c b
} [r zrevrange ztmp
1 -2]
68 assert_equal
{b a
} [r zrevrange ztmp
-2 -1]
69 assert_equal
{b
} [r zrevrange ztmp
-2 -2]
71 # out of range start index
72 assert_equal
{d c b
} [r zrevrange ztmp
-5 2]
73 assert_equal
{d c
} [r zrevrange ztmp
-5 1]
74 assert_equal
{} [r zrevrange ztmp
5 -1]
75 assert_equal
{} [r zrevrange ztmp
5 -2]
77 # out of range end index
78 assert_equal
{d c b a
} [r zrevrange ztmp
0 5]
79 assert_equal
{c b a
} [r zrevrange ztmp
1 5]
80 assert_equal
{} [r zrevrange ztmp
0 -5]
81 assert_equal
{} [r zrevrange ztmp
1 -5]
84 assert_equal
{d
4 c
3 b
2 a
1} [r zrevrange ztmp
0 -1 withscores
]
91 list [r zrank zranktmp x
] [r zrank zranktmp y
] [r zrank zranktmp z
]
94 test
{ZREVRANK basics
} {
95 list [r zrevrank zranktmp x
] [r zrevrank zranktmp y
] [r zrevrank zranktmp z
]
98 test
{ZRANK
- after deletion
} {
100 list [r zrank zranktmp x
] [r zrank zranktmp z
]
106 for {set i
0} {$i < 1000} {incr i
} {
107 set score
[expr rand
()]
109 r zadd zscoretest
$score $i
111 for {set i
0} {$i < 1000} {incr i
} {
112 if {[r zscore zscoretest
$i] != [lindex $aux $i]} {
113 set err
"Expected score was [lindex $aux $i] but got [r zscore zscoretest $i] for element $i"
120 test
{ZSCORE
after a DEBUG RELOAD
} {
124 for {set i
0} {$i < 1000} {incr i
} {
125 set score
[expr rand
()]
127 r zadd zscoretest
$score $i
130 for {set i
0} {$i < 1000} {incr i
} {
131 if {[r zscore zscoretest
$i] != [lindex $aux $i]} {
132 set err
"Expected score was [lindex $aux $i] but got [r zscore zscoretest $i] for element $i"
139 test
{ZSETs stress tester
- sorting is working well?
} {
141 for {set test
0} {$test < 2} {incr test
} {
142 unset -nocomplain auxarray
143 array set auxarray
{}
146 for {set i
0} {$i < 1000} {incr i
} {
148 set score
[expr rand
()]
150 set score
[expr int
(rand
()*10)]
152 set auxarray
($i) $score
153 r zadd myzset
$score $i
155 if {[expr rand
()] < .2} {
156 set j
[expr int
(rand
()*1000)]
158 set score
[expr rand
()]
160 set score
[expr int
(rand
()*10)]
162 set auxarray
($j) $score
163 r zadd myzset
$score $j
166 foreach {item score
} [array get auxarray
] {
167 lappend auxlist
[list $score $item]
169 set sorted
[lsort -command zlistAlikeSort
$auxlist]
172 lappend auxlist
[lindex $x 1]
174 set fromredis
[r zrange myzset
0 -1]
176 for {set i
0} {$i < [llength $fromredis]} {incr i
} {
177 if {[lindex $fromredis $i] != [lindex $auxlist $i]} {
185 test
{ZINCRBY
- can create a new sorted
set} {
188 list [r zrange zset
0 -1] [r zscore zset foo
]
191 test
{ZINCRBY
- increment and decrement
} {
194 set v1
[r zrange zset
0 -1]
195 r zincrby zset
10 bar
196 r zincrby zset
-5 foo
197 r zincrby zset
-5 bar
198 set v2
[r zrange zset
0 -1]
199 list $v1 $v2 [r zscore zset foo
] [r zscore zset bar
]
200 } {{bar foo
} {foo bar
} -2 6}
202 proc create_default_zset
{} {
203 create_zset zset
{-inf a
1 b
2 c
3 d
4 e
5 f
+inf g
}
206 test
"ZRANGEBYSCORE/ZREVRANGEBYSCORE/ZCOUNT basics" {
210 assert_equal
{a b c
} [r zrangebyscore zset
-inf 2]
211 assert_equal
{b c d
} [r zrangebyscore zset
0 3]
212 assert_equal
{d e f
} [r zrangebyscore zset
3 6]
213 assert_equal
{e f g
} [r zrangebyscore zset
4 +inf
]
214 assert_equal
{c b a
} [r zrevrangebyscore zset
2 -inf]
215 assert_equal
{d c b
} [r zrevrangebyscore zset
3 0]
216 assert_equal
{f e d
} [r zrevrangebyscore zset
6 3]
217 assert_equal
{g f e
} [r zrevrangebyscore zset
+inf
4]
218 assert_equal
3 [r zcount zset
0 3]
221 assert_equal
{b
} [r zrangebyscore zset
(-inf (2]
222 assert_equal
{b c
} [r zrangebyscore zset
(0 (3]
223 assert_equal
{e f
} [r zrangebyscore zset
(3 (6]
224 assert_equal
{f
} [r zrangebyscore zset
(4 (+inf
]
225 assert_equal
{b
} [r zrevrangebyscore zset
(2 (-inf]
226 assert_equal
{c b
} [r zrevrangebyscore zset
(3 (0]
227 assert_equal
{f e
} [r zrevrangebyscore zset
(6 (3]
228 assert_equal
{f
} [r zrevrangebyscore zset
(+inf
(4]
229 assert_equal
2 [r zcount zset
(0 (3]
232 test
"ZRANGEBYSCORE with WITHSCORES" {
234 assert_equal
{b
1 c
2 d
3} [r zrangebyscore zset
0 3 withscores
]
235 assert_equal
{d
3 c
2 b
1} [r zrevrangebyscore zset
3 0 withscores
]
238 test
"ZRANGEBYSCORE with LIMIT" {
240 assert_equal
{b c
} [r zrangebyscore zset
0 10 LIMIT
0 2]
241 assert_equal
{d e f
} [r zrangebyscore zset
0 10 LIMIT
2 3]
242 assert_equal
{d e f
} [r zrangebyscore zset
0 10 LIMIT
2 10]
243 assert_equal
{} [r zrangebyscore zset
0 10 LIMIT
20 10]
244 assert_equal
{f e
} [r zrevrangebyscore zset
10 0 LIMIT
0 2]
245 assert_equal
{d c b
} [r zrevrangebyscore zset
10 0 LIMIT
2 3]
246 assert_equal
{d c b
} [r zrevrangebyscore zset
10 0 LIMIT
2 10]
247 assert_equal
{} [r zrevrangebyscore zset
10 0 LIMIT
20 10]
250 test
"ZRANGEBYSCORE with LIMIT and WITHSCORES" {
252 assert_equal
{e
4 f
5} [r zrangebyscore zset
2 5 LIMIT
2 3 WITHSCORES
]
253 assert_equal
{d
3 c
2} [r zrevrangebyscore zset
5 2 LIMIT
2 3 WITHSCORES
]
256 test
"ZRANGEBYSCORE with non-value min or max" {
257 assert_error
"*not a double*" {r zrangebyscore fooz str
1}
258 assert_error
"*not a double*" {r zrangebyscore fooz
1 str
}
259 assert_error
"*not a double*" {r zrangebyscore fooz
1 NaN
}
263 test
{ZRANGEBYSCORE fuzzy test
, 100 ranges in
1000 elements sorted
set} {
266 for {set i
0} {$i < 1000} {incr i
} {
267 r zadd zset
[expr rand
()] $i
269 for {set i
0} {$i < 100} {incr i
} {
270 set min
[expr rand
()]
271 set max
[expr rand
()]
277 set low
[r zrangebyscore zset
-inf $min]
278 set ok
[r zrangebyscore zset
$min $max]
279 set high
[r zrangebyscore zset
$max +inf
]
280 set lowx
[r zrangebyscore zset
-inf ($min]
281 set okx
[r zrangebyscore zset
($min ($max]
282 set highx
[r zrangebyscore zset
($max +inf
]
284 if {[r zcount zset
-inf $min] != [llength $low]} {
285 append err
"Error, len does not match zcount\n"
287 if {[r zcount zset
$min $max] != [llength $ok]} {
288 append err
"Error, len does not match zcount\n"
290 if {[r zcount zset
$max +inf
] != [llength $high]} {
291 append err
"Error, len does not match zcount\n"
293 if {[r zcount zset
-inf ($min] != [llength $lowx]} {
294 append err
"Error, len does not match zcount\n"
296 if {[r zcount zset
($min ($max] != [llength $okx]} {
297 append err
"Error, len does not match zcount\n"
299 if {[r zcount zset
($max +inf
] != [llength $highx]} {
300 append err
"Error, len does not match zcount\n"
304 set score
[r zscore zset
$x]
306 append err
"Error, score for $x is $score > $min\n"
310 set score
[r zscore zset
$x]
311 if {$score >= $min} {
312 append err
"Error, score for $x is $score >= $min\n"
316 set score
[r zscore zset
$x]
317 if {$score < $min ||
$score > $max} {
318 append err
"Error, score for $x is $score outside $min-$max range\n"
322 set score
[r zscore zset
$x]
323 if {$score <= $min ||
$score >= $max} {
324 append err
"Error, score for $x is $score outside $min-$max open range\n"
328 set score
[r zscore zset
$x]
330 append err
"Error, score for $x is $score < $max\n"
334 set score
[r zscore zset
$x]
335 if {$score <= $max} {
336 append err
"Error, score for $x is $score <= $max\n"
344 test
"ZREMRANGEBYSCORE basics" {
345 proc remrangebyscore
{min max
} {
346 create_zset zset
{1 a
2 b
3 c
4 d
5 e
}
347 r zremrangebyscore zset
$min $max
351 assert_equal
3 [remrangebyscore
2 4]
352 assert_equal
{a e
} [r zrange zset
0 -1]
355 assert_equal
1 [remrangebyscore
-10 1]
356 assert_equal
{b c d e
} [r zrange zset
0 -1]
359 assert_equal
1 [remrangebyscore
5 10]
360 assert_equal
{a b c d
} [r zrange zset
0 -1]
363 assert_equal
0 [remrangebyscore
4 2]
364 assert_equal
{a b c d e
} [r zrange zset
0 -1]
367 assert_equal
3 [remrangebyscore
-inf 3]
368 assert_equal
{d e
} [r zrange zset
0 -1]
371 assert_equal
3 [remrangebyscore
3 +inf
]
372 assert_equal
{a b
} [r zrange zset
0 -1]
375 assert_equal
5 [remrangebyscore
-inf +inf
]
376 assert_equal
{} [r zrange zset
0 -1]
379 assert_equal
4 [remrangebyscore
(1 5]
380 assert_equal
{a
} [r zrange zset
0 -1]
381 assert_equal
3 [remrangebyscore
(2 5]
382 assert_equal
{a b
} [r zrange zset
0 -1]
385 assert_equal
4 [remrangebyscore
1 (5]
386 assert_equal
{e
} [r zrange zset
0 -1]
387 assert_equal
3 [remrangebyscore
1 (4]
388 assert_equal
{d e
} [r zrange zset
0 -1]
390 # exclusive min and max
391 assert_equal
3 [remrangebyscore
(1 (5]
392 assert_equal
{a e
} [r zrange zset
0 -1]
395 test
"ZREMRANGEBYSCORE with non-value min or max" {
396 assert_error
"*not a double*" {r zremrangebyscore fooz str
1}
397 assert_error
"*not a double*" {r zremrangebyscore fooz
1 str
}
398 assert_error
"*not a double*" {r zremrangebyscore fooz
1 NaN
}
401 test
"ZREMRANGEBYRANK basics" {
402 proc remrangebyrank
{min max
} {
403 create_zset zset
{1 a
2 b
3 c
4 d
5 e
}
404 r zremrangebyrank zset
$min $max
408 assert_equal
3 [remrangebyrank
1 3]
409 assert_equal
{a e
} [r zrange zset
0 -1]
412 assert_equal
1 [remrangebyrank
-10 0]
413 assert_equal
{b c d e
} [r zrange zset
0 -1]
416 assert_equal
0 [remrangebyrank
10 -1]
417 assert_equal
{a b c d e
} [r zrange zset
0 -1]
420 assert_equal
0 [remrangebyrank
0 -10]
421 assert_equal
{a b c d e
} [r zrange zset
0 -1]
424 assert_equal
5 [remrangebyrank
0 10]
425 assert_equal
{} [r zrange zset
0 -1]
428 test
{ZUNIONSTORE against non-existing key doesn't
set destination
} {
430 list [r zunionstore dst_key
1 zseta
] [r exists dst_key
]
433 test
{ZUNIONSTORE basics
} {
434 r del zseta zsetb zsetc
441 list [r zunionstore zsetc
2 zseta zsetb
] [r zrange zsetc
0 -1 withscores
]
442 } {4 {a
1 b
3 d
3 c
5}}
444 test
{ZUNIONSTORE with weights
} {
445 list [r zunionstore zsetc
2 zseta zsetb weights
2 3] [r zrange zsetc
0 -1 withscores
]
446 } {4 {a
2 b
7 d
9 c
12}}
448 test
{ZUNIONSTORE with a regular
set and weights
} {
453 list [r zunionstore zsetc
2 seta zsetb weights
2 3] [r zrange zsetc
0 -1 withscores
]
454 } {4 {a
2 b
5 c
8 d
9}}
456 test
{ZUNIONSTORE with AGGREGATE MIN
} {
457 list [r zunionstore zsetc
2 zseta zsetb aggregate min
] [r zrange zsetc
0 -1 withscores
]
458 } {4 {a
1 b
1 c
2 d
3}}
460 test
{ZUNIONSTORE with AGGREGATE MAX
} {
461 list [r zunionstore zsetc
2 zseta zsetb aggregate max
] [r zrange zsetc
0 -1 withscores
]
462 } {4 {a
1 b
2 c
3 d
3}}
464 test
{ZINTERSTORE basics
} {
465 list [r zinterstore zsetc
2 zseta zsetb
] [r zrange zsetc
0 -1 withscores
]
468 test
{ZINTERSTORE with weights
} {
469 list [r zinterstore zsetc
2 zseta zsetb weights
2 3] [r zrange zsetc
0 -1 withscores
]
472 test
{ZINTERSTORE with a regular
set and weights
} {
477 list [r zinterstore zsetc
2 seta zsetb weights
2 3] [r zrange zsetc
0 -1 withscores
]
480 test
{ZINTERSTORE with AGGREGATE MIN
} {
481 list [r zinterstore zsetc
2 zseta zsetb aggregate min
] [r zrange zsetc
0 -1 withscores
]
484 test
{ZINTERSTORE with AGGREGATE MAX
} {
485 list [r zinterstore zsetc
2 zseta zsetb aggregate max
] [r zrange zsetc
0 -1 withscores
]
488 foreach cmd
{ZUNIONSTORE ZINTERSTORE
} {
489 test
"$cmd with +inf/-inf scores" {
490 r del zsetinf1 zsetinf2
492 r zadd zsetinf1
+inf key
493 r zadd zsetinf2
+inf key
494 r
$cmd zsetinf3
2 zsetinf1 zsetinf2
495 assert_equal inf
[r zscore zsetinf3 key
]
497 r zadd zsetinf1
-inf key
498 r zadd zsetinf2
+inf key
499 r
$cmd zsetinf3
2 zsetinf1 zsetinf2
500 assert_equal
0 [r zscore zsetinf3 key
]
502 r zadd zsetinf1
+inf key
503 r zadd zsetinf2
-inf key
504 r
$cmd zsetinf3
2 zsetinf1 zsetinf2
505 assert_equal
0 [r zscore zsetinf3 key
]
507 r zadd zsetinf1
-inf key
508 r zadd zsetinf2
-inf key
509 r
$cmd zsetinf3
2 zsetinf1 zsetinf2
510 assert_equal
-inf [r zscore zsetinf3 key
]
513 test
"$cmd with NaN weights" {
514 r del zsetinf1 zsetinf2
516 r zadd zsetinf1
1.0 key
517 r zadd zsetinf2
1.0 key
518 assert_error
"*weight value is not a double*" {
519 r
$cmd zsetinf3
2 zsetinf1 zsetinf2 weights nan nan
525 test
{ZSETs skiplist implementation backlink consistency test
} {
528 for {set j
0} {$j < $elements} {incr j
} {
529 r zadd myzset
[expr rand
()] "Element-$j"
530 r zrem myzset
"Element-[expr int(rand()*$elements)]"
532 set l1
[r zrange myzset
0 -1]
533 set l2
[r zrevrange myzset
0 -1]
534 for {set j
0} {$j < [llength $l1]} {incr j
} {
535 if {[lindex $l1 $j] ne
[lindex $l2 end-
$j]} {
542 test
{ZSETs ZRANK augmented skip
list stress testing
} {
545 for {set k
0} {$k < 10000} {incr k
} {
546 set i
[expr {$k%1000}]
547 if {[expr rand
()] < .2} {
550 set score
[expr rand
()]
551 r zadd myzset
$score $i
553 set card
[r zcard myzset
]
555 set index
[randomInt
$card]
556 set ele
[lindex [r zrange myzset
$index $index] 0]
557 set rank
[r zrank myzset
$ele]
558 if {$rank != $index} {
559 set err
"$ele RANK is wrong! ($rank != $index)"
568 test
{ZSET element can't be
set to NaN with ZADD
} {
569 assert_error
"*not a double*" {r zadd myzset nan abc
}
572 test
{ZSET element can't be
set to NaN with ZINCRBY
} {
573 assert_error
"*not a double*" {r zadd myzset nan abc
}
576 test
{ZINCRBY calls leading to NaN result in
error} {
577 r zincrby myzset
+inf abc
578 assert_error
"*NaN*" {r zincrby myzset
-inf abc
}