]> git.saurik.com Git - redis.git/blob - test-redis.tcl
8568a60d27a10b3bf631d6116a6e194796ca08cd
[redis.git] / test-redis.tcl
1 # test-redis.tcl
2 # Redis test suite. Copyright (C) 2009 Salvatore Sanfilippo antirez@gmail.com
3 # This softare is released under the BSD License. See the COPYING file for
4 # more information.
5
6 set tcl_precision 17
7 source redis.tcl
8
9 set ::passed 0
10 set ::failed 0
11 set ::testnum 0
12
13 proc test {name code okpattern} {
14 incr ::testnum
15 if {$::testnum < $::first || $::testnum > $::last} return
16 puts -nonewline [format "%-70s " "#$::testnum $name"]
17 flush stdout
18 set retval [uplevel 1 $code]
19 if {$okpattern eq $retval || [string match $okpattern $retval]} {
20 puts "PASSED"
21 incr ::passed
22 } else {
23 puts "!! ERROR expected\n'$okpattern'\nbut got\n'$retval'"
24 incr ::failed
25 }
26 }
27
28 proc randstring {min max {type binary}} {
29 set len [expr {$min+int(rand()*($max-$min+1))}]
30 set output {}
31 if {$type eq {binary}} {
32 set minval 0
33 set maxval 255
34 } elseif {$type eq {alpha}} {
35 set minval 48
36 set maxval 122
37 } elseif {$type eq {compr}} {
38 set minval 48
39 set maxval 52
40 }
41 while {$len} {
42 append output [format "%c" [expr {$minval+int(rand()*($maxval-$minval+1))}]]
43 incr len -1
44 }
45 return $output
46 }
47
48 # Useful for some test
49 proc zlistAlikeSort {a b} {
50 if {[lindex $a 0] > [lindex $b 0]} {return 1}
51 if {[lindex $a 0] < [lindex $b 0]} {return -1}
52 string compare [lindex $a 1] [lindex $b 1]
53 }
54
55 proc waitForBgsave r {
56 while 1 {
57 set i [$r info]
58 if {[string match {*bgsave_in_progress:1*} $i]} {
59 puts -nonewline "\nWaiting for background save to finish... "
60 flush stdout
61 after 1000
62 } else {
63 break
64 }
65 }
66 }
67
68 proc main {server port} {
69 set r [redis $server $port]
70 $r select 9
71 set err ""
72 set res ""
73
74 # The following AUTH test should be enabled only when requirepass
75 # <PASSWORD> is set in redis.conf and redis-server was started with
76 # redis.conf as the first argument.
77
78 #test {AUTH with requirepass in redis.conf} {
79 # $r auth foobared
80 #} {OK}
81
82 test {DEL all keys to start with a clean DB} {
83 foreach key [$r keys *] {$r del $key}
84 $r dbsize
85 } {0}
86
87 test {SET and GET an item} {
88 $r set x foobar
89 $r get x
90 } {foobar}
91
92 test {SET and GET an empty item} {
93 $r set x {}
94 $r get x
95 } {}
96
97 test {DEL against a single item} {
98 $r del x
99 $r get x
100 } {}
101
102 test {Vararg DEL} {
103 $r set foo1 a
104 $r set foo2 b
105 $r set foo3 c
106 list [$r del foo1 foo2 foo3 foo4] [$r mget foo1 foo2 foo3]
107 } {3 {{} {} {}}}
108
109 test {KEYS with pattern} {
110 foreach key {key_x key_y key_z foo_a foo_b foo_c} {
111 $r set $key hello
112 }
113 lsort [$r keys foo*]
114 } {foo_a foo_b foo_c}
115
116 test {KEYS to get all keys} {
117 lsort [$r keys *]
118 } {foo_a foo_b foo_c key_x key_y key_z}
119
120 test {DBSIZE} {
121 $r dbsize
122 } {6}
123
124 test {DEL all keys} {
125 foreach key [$r keys *] {$r del $key}
126 $r dbsize
127 } {0}
128
129 test {Very big payload in GET/SET} {
130 set buf [string repeat "abcd" 1000000]
131 $r set foo $buf
132 $r get foo
133 } [string repeat "abcd" 1000000]
134
135 test {SET 10000 numeric keys and access all them in reverse order} {
136 for {set x 0} {$x < 10000} {incr x} {
137 $r set $x $x
138 }
139 set sum 0
140 for {set x 9999} {$x >= 0} {incr x -1} {
141 incr sum [$r get $x]
142 }
143 format $sum
144 } {49995000}
145
146 test {DBSIZE should be 10001 now} {
147 $r dbsize
148 } {10001}
149
150 test {INCR against non existing key} {
151 set res {}
152 append res [$r incr novar]
153 append res [$r get novar]
154 } {11}
155
156 test {INCR against key created by incr itself} {
157 $r incr novar
158 } {2}
159
160 test {INCR against key originally set with SET} {
161 $r set novar 100
162 $r incr novar
163 } {101}
164
165 test {INCR over 32bit value} {
166 $r set novar 17179869184
167 $r incr novar
168 } {17179869185}
169
170 test {INCRBY over 32bit value with over 32bit increment} {
171 $r set novar 17179869184
172 $r incrby novar 17179869184
173 } {34359738368}
174
175 test {DECRBY over 32bit value with over 32bit increment, negative res} {
176 $r set novar 17179869184
177 $r decrby novar 17179869185
178 } {-1}
179
180 test {SETNX target key missing} {
181 $r setnx novar2 foobared
182 $r get novar2
183 } {foobared}
184
185 test {SETNX target key exists} {
186 $r setnx novar2 blabla
187 $r get novar2
188 } {foobared}
189
190 test {SETNX will overwrite EXPIREing key} {
191 $r set x 10
192 $r expire x 10000
193 $r setnx x 20
194 $r get x
195 } {20}
196
197 test {EXISTS} {
198 set res {}
199 $r set newkey test
200 append res [$r exists newkey]
201 $r del newkey
202 append res [$r exists newkey]
203 } {10}
204
205 test {Zero length value in key. SET/GET/EXISTS} {
206 $r set emptykey {}
207 set res [$r get emptykey]
208 append res [$r exists emptykey]
209 $r del emptykey
210 append res [$r exists emptykey]
211 } {10}
212
213 test {Commands pipelining} {
214 set fd [$r channel]
215 puts -nonewline $fd "SET k1 4\r\nxyzk\r\nGET k1\r\nPING\r\n"
216 flush $fd
217 set res {}
218 append res [string match OK* [::redis::redis_read_reply $fd]]
219 append res [::redis::redis_read_reply $fd]
220 append res [string match PONG* [::redis::redis_read_reply $fd]]
221 format $res
222 } {1xyzk1}
223
224 test {Non existing command} {
225 catch {$r foobaredcommand} err
226 string match ERR* $err
227 } {1}
228
229 test {Basic LPUSH, RPUSH, LLENGTH, LINDEX} {
230 $r lpush mylist a
231 $r lpush mylist b
232 $r rpush mylist c
233 set res [$r llen mylist]
234 append res [$r lindex mylist 0]
235 append res [$r lindex mylist 1]
236 append res [$r lindex mylist 2]
237 } {3bac}
238
239 test {DEL a list} {
240 $r del mylist
241 $r exists mylist
242 } {0}
243
244 test {Create a long list and check every single element with LINDEX} {
245 set ok 0
246 for {set i 0} {$i < 1000} {incr i} {
247 $r rpush mylist $i
248 }
249 for {set i 0} {$i < 1000} {incr i} {
250 if {[$r lindex mylist $i] eq $i} {incr ok}
251 if {[$r lindex mylist [expr (-$i)-1]] eq [expr 999-$i]} {
252 incr ok
253 }
254 }
255 format $ok
256 } {2000}
257
258 test {Test elements with LINDEX in random access} {
259 set ok 0
260 for {set i 0} {$i < 1000} {incr i} {
261 set rint [expr int(rand()*1000)]
262 if {[$r lindex mylist $rint] eq $rint} {incr ok}
263 if {[$r lindex mylist [expr (-$rint)-1]] eq [expr 999-$rint]} {
264 incr ok
265 }
266 }
267 format $ok
268 } {2000}
269
270 test {Check if the list is still ok after a DEBUG RELOAD} {
271 $r debug reload
272 set ok 0
273 for {set i 0} {$i < 1000} {incr i} {
274 set rint [expr int(rand()*1000)]
275 if {[$r lindex mylist $rint] eq $rint} {incr ok}
276 if {[$r lindex mylist [expr (-$rint)-1]] eq [expr 999-$rint]} {
277 incr ok
278 }
279 }
280 format $ok
281 } {2000}
282
283 test {LLEN against non-list value error} {
284 $r del mylist
285 $r set mylist foobar
286 catch {$r llen mylist} err
287 format $err
288 } {ERR*}
289
290 test {LINDEX against non-list value error} {
291 catch {$r lindex mylist 0} err
292 format $err
293 } {ERR*}
294
295 test {LPUSH against non-list value error} {
296 catch {$r lpush mylist 0} err
297 format $err
298 } {ERR*}
299
300 test {RPUSH against non-list value error} {
301 catch {$r rpush mylist 0} err
302 format $err
303 } {ERR*}
304
305 test {RPOPLPUSH base case} {
306 $r del mylist
307 $r rpush mylist a
308 $r rpush mylist b
309 $r rpush mylist c
310 $r rpush mylist d
311 set v1 [$r rpoplpush mylist newlist]
312 set v2 [$r rpoplpush mylist newlist]
313 set l1 [$r lrange mylist 0 -1]
314 set l2 [$r lrange newlist 0 -1]
315 list $v1 $v2 $l1 $l2
316 } {d c {a b} {c d}}
317
318 test {RPOPLPUSH with the same list as src and dst} {
319 $r del mylist
320 $r rpush mylist a
321 $r rpush mylist b
322 $r rpush mylist c
323 set l1 [$r lrange mylist 0 -1]
324 set v [$r rpoplpush mylist mylist]
325 set l2 [$r lrange mylist 0 -1]
326 list $l1 $v $l2
327 } {{a b c} c {c a b}}
328
329 test {RPOPLPUSH target list already exists} {
330 $r del mylist
331 $r del newlist
332 $r rpush mylist a
333 $r rpush mylist b
334 $r rpush mylist c
335 $r rpush mylist d
336 $r rpush newlist x
337 set v1 [$r rpoplpush mylist newlist]
338 set v2 [$r rpoplpush mylist newlist]
339 set l1 [$r lrange mylist 0 -1]
340 set l2 [$r lrange newlist 0 -1]
341 list $v1 $v2 $l1 $l2
342 } {d c {a b} {c d x}}
343
344 test {RPOPLPUSH against non existing key} {
345 $r del mylist
346 $r del newlist
347 set v1 [$r rpoplpush mylist newlist]
348 list $v1 [$r exists mylist] [$r exists newlist]
349 } {{} 0 0}
350
351 test {RPOPLPUSH against non list src key} {
352 $r del mylist
353 $r del newlist
354 $r set mylist x
355 catch {$r rpoplpush mylist newlist} err
356 list [$r type mylist] [$r exists newlist] [string range $err 0 2]
357 } {string 0 ERR}
358
359 test {RPOPLPUSH against non list dst key} {
360 $r del mylist
361 $r del newlist
362 $r rpush mylist a
363 $r rpush mylist b
364 $r rpush mylist c
365 $r rpush mylist d
366 $r set newlist x
367 catch {$r rpoplpush mylist newlist} err
368 list [$r lrange mylist 0 -1] [$r type newlist] [string range $err 0 2]
369 } {{a b c d} string ERR}
370
371 test {RENAME basic usage} {
372 $r set mykey hello
373 $r rename mykey mykey1
374 $r rename mykey1 mykey2
375 $r get mykey2
376 } {hello}
377
378 test {RENAME source key should no longer exist} {
379 $r exists mykey
380 } {0}
381
382 test {RENAME against already existing key} {
383 $r set mykey a
384 $r set mykey2 b
385 $r rename mykey2 mykey
386 set res [$r get mykey]
387 append res [$r exists mykey2]
388 } {b0}
389
390 test {RENAMENX basic usage} {
391 $r del mykey
392 $r del mykey2
393 $r set mykey foobar
394 $r renamenx mykey mykey2
395 set res [$r get mykey2]
396 append res [$r exists mykey]
397 } {foobar0}
398
399 test {RENAMENX against already existing key} {
400 $r set mykey foo
401 $r set mykey2 bar
402 $r renamenx mykey mykey2
403 } {0}
404
405 test {RENAMENX against already existing key (2)} {
406 set res [$r get mykey]
407 append res [$r get mykey2]
408 } {foobar}
409
410 test {RENAME against non existing source key} {
411 catch {$r rename nokey foobar} err
412 format $err
413 } {ERR*}
414
415 test {RENAME where source and dest key is the same} {
416 catch {$r rename mykey mykey} err
417 format $err
418 } {ERR*}
419
420 test {DEL all keys again (DB 0)} {
421 foreach key [$r keys *] {
422 $r del $key
423 }
424 $r dbsize
425 } {0}
426
427 test {DEL all keys again (DB 1)} {
428 $r select 10
429 foreach key [$r keys *] {
430 $r del $key
431 }
432 set res [$r dbsize]
433 $r select 9
434 format $res
435 } {0}
436
437 test {MOVE basic usage} {
438 $r set mykey foobar
439 $r move mykey 10
440 set res {}
441 lappend res [$r exists mykey]
442 lappend res [$r dbsize]
443 $r select 10
444 lappend res [$r get mykey]
445 lappend res [$r dbsize]
446 $r select 9
447 format $res
448 } [list 0 0 foobar 1]
449
450 test {MOVE against key existing in the target DB} {
451 $r set mykey hello
452 $r move mykey 10
453 } {0}
454
455 test {SET/GET keys in different DBs} {
456 $r set a hello
457 $r set b world
458 $r select 10
459 $r set a foo
460 $r set b bared
461 $r select 9
462 set res {}
463 lappend res [$r get a]
464 lappend res [$r get b]
465 $r select 10
466 lappend res [$r get a]
467 lappend res [$r get b]
468 $r select 9
469 format $res
470 } {hello world foo bared}
471
472 test {Basic LPOP/RPOP} {
473 $r del mylist
474 $r rpush mylist 1
475 $r rpush mylist 2
476 $r lpush mylist 0
477 list [$r lpop mylist] [$r rpop mylist] [$r lpop mylist] [$r llen mylist]
478 } [list 0 2 1 0]
479
480 test {LPOP/RPOP against empty list} {
481 $r lpop mylist
482 } {}
483
484 test {LPOP against non list value} {
485 $r set notalist foo
486 catch {$r lpop notalist} err
487 format $err
488 } {ERR*kind*}
489
490 test {Mass LPUSH/LPOP} {
491 set sum 0
492 for {set i 0} {$i < 1000} {incr i} {
493 $r lpush mylist $i
494 incr sum $i
495 }
496 set sum2 0
497 for {set i 0} {$i < 500} {incr i} {
498 incr sum2 [$r lpop mylist]
499 incr sum2 [$r rpop mylist]
500 }
501 expr $sum == $sum2
502 } {1}
503
504 test {LRANGE basics} {
505 for {set i 0} {$i < 10} {incr i} {
506 $r rpush mylist $i
507 }
508 list [$r lrange mylist 1 -2] \
509 [$r lrange mylist -3 -1] \
510 [$r lrange mylist 4 4]
511 } {{1 2 3 4 5 6 7 8} {7 8 9} 4}
512
513 test {LRANGE inverted indexes} {
514 $r lrange mylist 6 2
515 } {}
516
517 test {LRANGE out of range indexes including the full list} {
518 $r lrange mylist -1000 1000
519 } {0 1 2 3 4 5 6 7 8 9}
520
521 test {LRANGE against non existing key} {
522 $r lrange nosuchkey 0 1
523 } {}
524
525 test {LTRIM basics} {
526 $r del mylist
527 for {set i 0} {$i < 100} {incr i} {
528 $r lpush mylist $i
529 $r ltrim mylist 0 4
530 }
531 $r lrange mylist 0 -1
532 } {99 98 97 96 95}
533
534 test {LSET} {
535 $r lset mylist 1 foo
536 $r lset mylist -1 bar
537 $r lrange mylist 0 -1
538 } {99 foo 97 96 bar}
539
540 test {LSET out of range index} {
541 catch {$r lset mylist 10 foo} err
542 format $err
543 } {ERR*range*}
544
545 test {LSET against non existing key} {
546 catch {$r lset nosuchkey 10 foo} err
547 format $err
548 } {ERR*key*}
549
550 test {LSET against non list value} {
551 $r set nolist foobar
552 catch {$r lset nolist 0 foo} err
553 format $err
554 } {ERR*value*}
555
556 test {SADD, SCARD, SISMEMBER, SMEMBERS basics} {
557 $r sadd myset foo
558 $r sadd myset bar
559 list [$r scard myset] [$r sismember myset foo] \
560 [$r sismember myset bar] [$r sismember myset bla] \
561 [lsort [$r smembers myset]]
562 } {2 1 1 0 {bar foo}}
563
564 test {SADD adding the same element multiple times} {
565 $r sadd myset foo
566 $r sadd myset foo
567 $r sadd myset foo
568 $r scard myset
569 } {2}
570
571 test {SADD against non set} {
572 catch {$r sadd mylist foo} err
573 format $err
574 } {ERR*kind*}
575
576 test {SREM basics} {
577 $r sadd myset ciao
578 $r srem myset foo
579 lsort [$r smembers myset]
580 } {bar ciao}
581
582 test {Mass SADD and SINTER with two sets} {
583 for {set i 0} {$i < 1000} {incr i} {
584 $r sadd set1 $i
585 $r sadd set2 [expr $i+995]
586 }
587 lsort [$r sinter set1 set2]
588 } {995 996 997 998 999}
589
590 test {SUNION with two sets} {
591 lsort [$r sunion set1 set2]
592 } [lsort -uniq "[$r smembers set1] [$r smembers set2]"]
593
594 test {SINTERSTORE with two sets} {
595 $r sinterstore setres set1 set2
596 lsort [$r smembers setres]
597 } {995 996 997 998 999}
598
599 test {SINTERSTORE with two sets, after a DEBUG RELOAD} {
600 $r debug reload
601 $r sinterstore setres set1 set2
602 lsort [$r smembers setres]
603 } {995 996 997 998 999}
604
605 test {SUNIONSTORE with two sets} {
606 $r sunionstore setres set1 set2
607 lsort [$r smembers setres]
608 } [lsort -uniq "[$r smembers set1] [$r smembers set2]"]
609
610 test {SUNIONSTORE against non existing keys} {
611 $r set setres xxx
612 list [$r sunionstore setres foo111 bar222] [$r exists xxx]
613 } {0 0}
614
615 test {SINTER against three sets} {
616 $r sadd set3 999
617 $r sadd set3 995
618 $r sadd set3 1000
619 $r sadd set3 2000
620 lsort [$r sinter set1 set2 set3]
621 } {995 999}
622
623 test {SINTERSTORE with three sets} {
624 $r sinterstore setres set1 set2 set3
625 lsort [$r smembers setres]
626 } {995 999}
627
628 test {SUNION with non existing keys} {
629 lsort [$r sunion nokey1 set1 set2 nokey2]
630 } [lsort -uniq "[$r smembers set1] [$r smembers set2]"]
631
632 test {SDIFF with two sets} {
633 for {set i 5} {$i < 1000} {incr i} {
634 $r sadd set4 $i
635 }
636 lsort [$r sdiff set1 set4]
637 } {0 1 2 3 4}
638
639 test {SDIFF with three sets} {
640 $r sadd set5 0
641 lsort [$r sdiff set1 set4 set5]
642 } {1 2 3 4}
643
644 test {SDIFFSTORE with three sets} {
645 $r sdiffstore sres set1 set4 set5
646 lsort [$r smembers sres]
647 } {1 2 3 4}
648
649 test {SPOP basics} {
650 $r del myset
651 $r sadd myset 1
652 $r sadd myset 2
653 $r sadd myset 3
654 list [lsort [list [$r spop myset] [$r spop myset] [$r spop myset]]] [$r scard myset]
655 } {{1 2 3} 0}
656
657 test {SAVE - make sure there are all the types as values} {
658 # Wait for a background saving in progress to terminate
659 waitForBgsave $r
660 $r lpush mysavelist hello
661 $r lpush mysavelist world
662 $r set myemptykey {}
663 $r set mynormalkey {blablablba}
664 $r zadd mytestzset a 10
665 $r zadd mytestzset b 20
666 $r zadd mytestzset c 30
667 $r save
668 } {OK}
669
670 test {SRANDMEMBER} {
671 $r del myset
672 $r sadd myset a
673 $r sadd myset b
674 $r sadd myset c
675 unset -nocomplain myset
676 array set myset {}
677 for {set i 0} {$i < 100} {incr i} {
678 set myset([$r srandmember myset]) 1
679 }
680 lsort [array names myset]
681 } {a b c}
682
683 test {Create a random list} {
684 set tosort {}
685 array set seenrand {}
686 for {set i 0} {$i < 10000} {incr i} {
687 while 1 {
688 # Make sure all the weights are different because
689 # Redis does not use a stable sort but Tcl does.
690 set rint [expr int(rand()*1000000)]
691 if {![info exists seenrand($rint)]} break
692 }
693 set seenrand($rint) x
694 $r lpush tosort $i
695 $r set weight_$i $rint
696 lappend tosort [list $i $rint]
697 }
698 set sorted [lsort -index 1 -real $tosort]
699 set res {}
700 for {set i 0} {$i < 10000} {incr i} {
701 lappend res [lindex $sorted $i 0]
702 }
703 format {}
704 } {}
705
706 test {SORT with BY against the newly created list} {
707 $r sort tosort {BY weight_*}
708 } $res
709
710 test {SORT direct, numeric, against the newly created list} {
711 $r sort tosort
712 } [lsort -integer $res]
713
714 test {SORT decreasing sort} {
715 $r sort tosort {DESC}
716 } [lsort -decreasing -integer $res]
717
718 test {SORT speed, sorting 10000 elements list using BY, 100 times} {
719 set start [clock clicks -milliseconds]
720 for {set i 0} {$i < 100} {incr i} {
721 set sorted [$r sort tosort {BY weight_* LIMIT 0 10}]
722 }
723 set elapsed [expr [clock clicks -milliseconds]-$start]
724 puts -nonewline "\n Average time to sort: [expr double($elapsed)/100] milliseconds "
725 flush stdout
726 format {}
727 } {}
728
729 test {SORT speed, sorting 10000 elements list directly, 100 times} {
730 set start [clock clicks -milliseconds]
731 for {set i 0} {$i < 100} {incr i} {
732 set sorted [$r sort tosort {LIMIT 0 10}]
733 }
734 set elapsed [expr [clock clicks -milliseconds]-$start]
735 puts -nonewline "\n Average time to sort: [expr double($elapsed)/100] milliseconds "
736 flush stdout
737 format {}
738 } {}
739
740 test {SORT speed, pseudo-sorting 10000 elements list, BY <const>, 100 times} {
741 set start [clock clicks -milliseconds]
742 for {set i 0} {$i < 100} {incr i} {
743 set sorted [$r sort tosort {BY nokey LIMIT 0 10}]
744 }
745 set elapsed [expr [clock clicks -milliseconds]-$start]
746 puts -nonewline "\n Average time to sort: [expr double($elapsed)/100] milliseconds "
747 flush stdout
748 format {}
749 } {}
750
751 test {SORT regression for issue #19, sorting floats} {
752 $r flushdb
753 foreach x {1.1 5.10 3.10 7.44 2.1 5.75 6.12 0.25 1.15} {
754 $r lpush mylist $x
755 }
756 $r sort mylist
757 } [lsort -real {1.1 5.10 3.10 7.44 2.1 5.75 6.12 0.25 1.15}]
758
759 test {SORT with GET #} {
760 $r del mylist
761 $r lpush mylist 1
762 $r lpush mylist 2
763 $r lpush mylist 3
764 $r mset weight_1 10 weight_2 5 weight_3 30
765 $r sort mylist BY weight_* GET #
766 } {2 1 3}
767
768 test {LREM, remove all the occurrences} {
769 $r flushdb
770 $r rpush mylist foo
771 $r rpush mylist bar
772 $r rpush mylist foobar
773 $r rpush mylist foobared
774 $r rpush mylist zap
775 $r rpush mylist bar
776 $r rpush mylist test
777 $r rpush mylist foo
778 set res [$r lrem mylist 0 bar]
779 list [$r lrange mylist 0 -1] $res
780 } {{foo foobar foobared zap test foo} 2}
781
782 test {LREM, remove the first occurrence} {
783 set res [$r lrem mylist 1 foo]
784 list [$r lrange mylist 0 -1] $res
785 } {{foobar foobared zap test foo} 1}
786
787 test {LREM, remove non existing element} {
788 set res [$r lrem mylist 1 nosuchelement]
789 list [$r lrange mylist 0 -1] $res
790 } {{foobar foobared zap test foo} 0}
791
792 test {LREM, starting from tail with negative count} {
793 $r flushdb
794 $r rpush mylist foo
795 $r rpush mylist bar
796 $r rpush mylist foobar
797 $r rpush mylist foobared
798 $r rpush mylist zap
799 $r rpush mylist bar
800 $r rpush mylist test
801 $r rpush mylist foo
802 $r rpush mylist foo
803 set res [$r lrem mylist -1 bar]
804 list [$r lrange mylist 0 -1] $res
805 } {{foo bar foobar foobared zap test foo foo} 1}
806
807 test {LREM, starting from tail with negative count (2)} {
808 set res [$r lrem mylist -2 foo]
809 list [$r lrange mylist 0 -1] $res
810 } {{foo bar foobar foobared zap test} 2}
811
812 test {LREM, deleting objects that may be encoded as integers} {
813 $r lpush myotherlist 1
814 $r lpush myotherlist 2
815 $r lpush myotherlist 3
816 $r lrem myotherlist 1 2
817 $r llen myotherlist
818 } {2}
819
820 test {MGET} {
821 $r flushdb
822 $r set foo BAR
823 $r set bar FOO
824 $r mget foo bar
825 } {BAR FOO}
826
827 test {MGET against non existing key} {
828 $r mget foo baazz bar
829 } {BAR {} FOO}
830
831 test {MGET against non-string key} {
832 $r sadd myset ciao
833 $r sadd myset bau
834 $r mget foo baazz bar myset
835 } {BAR {} FOO {}}
836
837 test {RANDOMKEY} {
838 $r flushdb
839 $r set foo x
840 $r set bar y
841 set foo_seen 0
842 set bar_seen 0
843 for {set i 0} {$i < 100} {incr i} {
844 set rkey [$r randomkey]
845 if {$rkey eq {foo}} {
846 set foo_seen 1
847 }
848 if {$rkey eq {bar}} {
849 set bar_seen 1
850 }
851 }
852 list $foo_seen $bar_seen
853 } {1 1}
854
855 test {RANDOMKEY against empty DB} {
856 $r flushdb
857 $r randomkey
858 } {}
859
860 test {RANDOMKEY regression 1} {
861 $r flushdb
862 $r set x 10
863 $r del x
864 $r randomkey
865 } {}
866
867 test {GETSET (set new value)} {
868 list [$r getset foo xyz] [$r get foo]
869 } {{} xyz}
870
871 test {GETSET (replace old value)} {
872 $r set foo bar
873 list [$r getset foo xyz] [$r get foo]
874 } {bar xyz}
875
876 test {SMOVE basics} {
877 $r sadd myset1 a
878 $r sadd myset1 b
879 $r sadd myset1 c
880 $r sadd myset2 x
881 $r sadd myset2 y
882 $r sadd myset2 z
883 $r smove myset1 myset2 a
884 list [lsort [$r smembers myset2]] [lsort [$r smembers myset1]]
885 } {{a x y z} {b c}}
886
887 test {SMOVE non existing key} {
888 list [$r smove myset1 myset2 foo] [lsort [$r smembers myset2]] [lsort [$r smembers myset1]]
889 } {0 {a x y z} {b c}}
890
891 test {SMOVE non existing src set} {
892 list [$r smove noset myset2 foo] [lsort [$r smembers myset2]]
893 } {0 {a x y z}}
894
895 test {SMOVE non existing dst set} {
896 list [$r smove myset2 myset3 y] [lsort [$r smembers myset2]] [lsort [$r smembers myset3]]
897 } {1 {a x z} y}
898
899 test {SMOVE wrong src key type} {
900 $r set x 10
901 catch {$r smove x myset2 foo} err
902 format $err
903 } {ERR*}
904
905 test {SMOVE wrong dst key type} {
906 $r set x 10
907 catch {$r smove myset2 x foo} err
908 format $err
909 } {ERR*}
910
911 test {MSET base case} {
912 $r mset x 10 y "foo bar" z "x x x x x x x\n\n\r\n"
913 $r mget x y z
914 } [list 10 {foo bar} "x x x x x x x\n\n\r\n"]
915
916 test {MSET wrong number of args} {
917 catch {$r mset x 10 y "foo bar" z} err
918 format $err
919 } {*wrong number*}
920
921 test {MSETNX with already existent key} {
922 list [$r msetnx x1 xxx y2 yyy x 20] [$r exists x1] [$r exists y2]
923 } {0 0 0}
924
925 test {MSETNX with not existing keys} {
926 list [$r msetnx x1 xxx y2 yyy] [$r get x1] [$r get y2]
927 } {1 xxx yyy}
928
929 test {MSETNX should remove all the volatile keys even on failure} {
930 $r mset x 1 y 2 z 3
931 $r expire y 10000
932 $r expire z 10000
933 list [$r msetnx x A y B z C] [$r mget x y z]
934 } {0 {1 {} {}}}
935
936 test {ZSET basic ZADD and score update} {
937 $r zadd ztmp 10 x
938 $r zadd ztmp 20 y
939 $r zadd ztmp 30 z
940 set aux1 [$r zrange ztmp 0 -1]
941 $r zadd ztmp 1 y
942 set aux2 [$r zrange ztmp 0 -1]
943 list $aux1 $aux2
944 } {{x y z} {y x z}}
945
946 test {ZCARD basics} {
947 $r zcard ztmp
948 } {3}
949
950 test {ZCARD non existing key} {
951 $r zcard ztmp-blabla
952 } {0}
953
954 test {ZSCORE} {
955 set aux {}
956 set err {}
957 for {set i 0} {$i < 1000} {incr i} {
958 set score [expr rand()]
959 lappend aux $score
960 $r zadd zscoretest $score $i
961 }
962 for {set i 0} {$i < 1000} {incr i} {
963 if {[$r zscore zscoretest $i] != [lindex $aux $i]} {
964 set err "Expected score was [lindex $aux $i] but got [$r zscore zscoretest $i] for element $i"
965 break
966 }
967 }
968 set _ $err
969 } {}
970
971 test {ZSCORE after a DEBUG RELOAD} {
972 set aux {}
973 set err {}
974 $r del zscoretest
975 for {set i 0} {$i < 1000} {incr i} {
976 set score [expr rand()]
977 lappend aux $score
978 $r zadd zscoretest $score $i
979 }
980 $r debug reload
981 for {set i 0} {$i < 1000} {incr i} {
982 if {[$r zscore zscoretest $i] != [lindex $aux $i]} {
983 set err "Expected score was [lindex $aux $i] but got [$r zscore zscoretest $i] for element $i"
984 break
985 }
986 }
987 set _ $err
988 } {}
989
990 test {ZRANGE and ZREVRANGE} {
991 list [$r zrange ztmp 0 -1] [$r zrevrange ztmp 0 -1]
992 } {{y x z} {z x y}}
993
994 test {ZSETs stress tester - sorting is working well?} {
995 set delta 0
996 for {set test 0} {$test < 2} {incr test} {
997 unset -nocomplain auxarray
998 array set auxarray {}
999 set auxlist {}
1000 $r del myzset
1001 for {set i 0} {$i < 1000} {incr i} {
1002 if {$test == 0} {
1003 set score [expr rand()]
1004 } else {
1005 set score [expr int(rand()*10)]
1006 }
1007 set auxarray($i) $score
1008 $r zadd myzset $score $i
1009 # Random update
1010 if {[expr rand()] < .2} {
1011 set j [expr int(rand()*1000)]
1012 if {$test == 0} {
1013 set score [expr rand()]
1014 } else {
1015 set score [expr int(rand()*10)]
1016 }
1017 set auxarray($j) $score
1018 $r zadd myzset $score $j
1019 }
1020 }
1021 foreach {item score} [array get auxarray] {
1022 lappend auxlist [list $score $item]
1023 }
1024 set sorted [lsort -command zlistAlikeSort $auxlist]
1025 set auxlist {}
1026 foreach x $sorted {
1027 lappend auxlist [lindex $x 1]
1028 }
1029 set fromredis [$r zrange myzset 0 -1]
1030 set delta 0
1031 for {set i 0} {$i < [llength $fromredis]} {incr i} {
1032 if {[lindex $fromredis $i] != [lindex $auxlist $i]} {
1033 incr delta
1034 }
1035 }
1036 }
1037 format $delta
1038 } {0}
1039
1040 test {ZINCRBY - can create a new sorted set} {
1041 $r del zset
1042 $r zincrby zset 1 foo
1043 list [$r zrange zset 0 -1] [$r zscore zset foo]
1044 } {foo 1}
1045
1046 test {ZINCRBY - increment and decrement} {
1047 $r zincrby zset 2 foo
1048 $r zincrby zset 1 bar
1049 set v1 [$r zrange zset 0 -1]
1050 $r zincrby zset 10 bar
1051 $r zincrby zset -5 foo
1052 $r zincrby zset -5 bar
1053 set v2 [$r zrange zset 0 -1]
1054 list $v1 $v2 [$r zscore zset foo] [$r zscore zset bar]
1055 } {{bar foo} {foo bar} -2 6}
1056
1057 test {ZRANGEBYSCORE basics} {
1058 $r del zset
1059 $r zadd zset 1 a
1060 $r zadd zset 2 b
1061 $r zadd zset 3 c
1062 $r zadd zset 4 d
1063 $r zadd zset 5 e
1064 $r zrangebyscore zset 2 4
1065 } {b c d}
1066
1067 test {ZRANGEBYSCORE fuzzy test, 100 ranges in 1000 elements sorted set} {
1068 set err {}
1069 $r del zset
1070 for {set i 0} {$i < 1000} {incr i} {
1071 $r zadd zset [expr rand()] $i
1072 }
1073 for {set i 0} {$i < 100} {incr i} {
1074 set min [expr rand()]
1075 set max [expr rand()]
1076 if {$min > $max} {
1077 set aux $min
1078 set min $max
1079 set max $aux
1080 }
1081 set low [$r zrangebyscore zset -inf $min]
1082 set ok [$r zrangebyscore zset $min $max]
1083 set high [$r zrangebyscore zset $max +inf]
1084 foreach x $low {
1085 set score [$r zscore zset $x]
1086 if {$score > $min} {
1087 append err "Error, score for $x is $score > $min\n"
1088 }
1089 }
1090 foreach x $ok {
1091 set score [$r zscore zset $x]
1092 if {$score < $min || $score > $max} {
1093 append err "Error, score for $x is $score outside $min-$max range\n"
1094 }
1095 }
1096 foreach x $high {
1097 set score [$r zscore zset $x]
1098 if {$score < $max} {
1099 append err "Error, score for $x is $score < $max\n"
1100 }
1101 }
1102 }
1103 set _ $err
1104 } {}
1105
1106 test {ZRANGEBYSCORE with LIMIT} {
1107 $r del zset
1108 $r zadd zset 1 a
1109 $r zadd zset 2 b
1110 $r zadd zset 3 c
1111 $r zadd zset 4 d
1112 $r zadd zset 5 e
1113 list \
1114 [$r zrangebyscore zset 0 10 LIMIT 0 2] \
1115 [$r zrangebyscore zset 0 10 LIMIT 2 3] \
1116 [$r zrangebyscore zset 0 10 LIMIT 2 10] \
1117 [$r zrangebyscore zset 0 10 LIMIT 20 10]
1118 } {{a b} {c d e} {c d e} {}}
1119
1120 test {ZREMRANGE basics} {
1121 $r del zset
1122 $r zadd zset 1 a
1123 $r zadd zset 2 b
1124 $r zadd zset 3 c
1125 $r zadd zset 4 d
1126 $r zadd zset 5 e
1127 list [$r zremrangebyscore zset 2 4] [$r zrange zset 0 -1]
1128 } {3 {a e}}
1129
1130 test {ZREMRANGE from -inf to +inf} {
1131 $r del zset
1132 $r zadd zset 1 a
1133 $r zadd zset 2 b
1134 $r zadd zset 3 c
1135 $r zadd zset 4 d
1136 $r zadd zset 5 e
1137 list [$r zremrangebyscore zset -inf +inf] [$r zrange zset 0 -1]
1138 } {5 {}}
1139
1140 test {SORT against sorted sets} {
1141 $r del zset
1142 $r zadd zset 1 a
1143 $r zadd zset 5 b
1144 $r zadd zset 2 c
1145 $r zadd zset 10 d
1146 $r zadd zset 3 e
1147 $r sort zset alpha desc
1148 } {e d c b a}
1149
1150 test {Sorted sets +inf and -inf handling} {
1151 $r del zset
1152 $r zadd zset -100 a
1153 $r zadd zset 200 b
1154 $r zadd zset -300 c
1155 $r zadd zset 1000000 d
1156 $r zadd zset +inf max
1157 $r zadd zset -inf min
1158 $r zrange zset 0 -1
1159 } {min c a b d max}
1160
1161 test {EXPIRE - don't set timeouts multiple times} {
1162 $r set x foobar
1163 set v1 [$r expire x 5]
1164 set v2 [$r ttl x]
1165 set v3 [$r expire x 10]
1166 set v4 [$r ttl x]
1167 list $v1 $v2 $v3 $v4
1168 } {1 5 0 5}
1169
1170 test {EXPIRE - It should be still possible to read 'x'} {
1171 $r get x
1172 } {foobar}
1173
1174 test {EXPIRE - After 6 seconds the key should no longer be here} {
1175 after 6000
1176 list [$r get x] [$r exists x]
1177 } {{} 0}
1178
1179 test {EXPIRE - Delete on write policy} {
1180 $r del x
1181 $r lpush x foo
1182 $r expire x 1000
1183 $r lpush x bar
1184 $r lrange x 0 -1
1185 } {bar}
1186
1187 test {EXPIREAT - Check for EXPIRE alike behavior} {
1188 $r del x
1189 $r set x foo
1190 $r expireat x [expr [clock seconds]+15]
1191 $r ttl x
1192 } {1[345]}
1193
1194 test {ZSETs skiplist implementation backlink consistency test} {
1195 set diff 0
1196 set elements 10000
1197 for {set j 0} {$j < $elements} {incr j} {
1198 $r zadd myzset [expr rand()] "Element-$j"
1199 $r zrem myzset "Element-[expr int(rand()*$elements)]"
1200 }
1201 set l1 [$r zrange myzset 0 -1]
1202 set l2 [$r zrevrange myzset 0 -1]
1203 for {set j 0} {$j < [llength $l1]} {incr j} {
1204 if {[lindex $l1 $j] ne [lindex $l2 end-$j]} {
1205 incr diff
1206 }
1207 }
1208 format $diff
1209 } {0}
1210
1211 foreach fuzztype {binary alpha compr} {
1212 test "FUZZ stresser with data model $fuzztype" {
1213 set err 0
1214 for {set i 0} {$i < 10000} {incr i} {
1215 set fuzz [randstring 0 512 $fuzztype]
1216 $r set foo $fuzz
1217 set got [$r get foo]
1218 if {$got ne $fuzz} {
1219 set err [list $fuzz $got]
1220 break
1221 }
1222 }
1223 set _ $err
1224 } {0}
1225 }
1226
1227 test {BGSAVE} {
1228 $r flushdb
1229 $r save
1230 $r set x 10
1231 $r bgsave
1232 waitForBgsave $r
1233 $r debug reload
1234 $r get x
1235 } {10}
1236
1237 test {Handle an empty query well} {
1238 set fd [$r channel]
1239 puts -nonewline $fd "\r\n"
1240 flush $fd
1241 $r ping
1242 } {PONG}
1243
1244 test {Negative multi bulk command does not create problems} {
1245 set fd [$r channel]
1246 puts -nonewline $fd "*-10\r\n"
1247 flush $fd
1248 $r ping
1249 } {PONG}
1250
1251 test {Negative multi bulk payload} {
1252 set fd [$r channel]
1253 puts -nonewline $fd "SET x -10\r\n"
1254 flush $fd
1255 gets $fd
1256 } {*invalid bulk*}
1257
1258 test {Too big bulk payload} {
1259 set fd [$r channel]
1260 puts -nonewline $fd "SET x 2000000000\r\n"
1261 flush $fd
1262 gets $fd
1263 } {*invalid bulk*count*}
1264
1265 test {Multi bulk request not followed by bulk args} {
1266 set fd [$r channel]
1267 puts -nonewline $fd "*1\r\nfoo\r\n"
1268 flush $fd
1269 gets $fd
1270 } {*protocol error*}
1271
1272 test {Generic wrong number of args} {
1273 catch {$r ping x y z} err
1274 set _ $err
1275 } {*wrong*arguments*ping*}
1276
1277 test {SELECT an out of range DB} {
1278 catch {$r select 1000000} err
1279 set _ $err
1280 } {*invalid*}
1281
1282 # Leave the user with a clean DB before to exit
1283 test {FLUSHDB} {
1284 set aux {}
1285 $r select 9
1286 $r flushdb
1287 lappend aux [$r dbsize]
1288 $r select 10
1289 $r flushdb
1290 lappend aux [$r dbsize]
1291 } {0 0}
1292
1293 test {Perform a final SAVE to leave a clean DB on disk} {
1294 $r save
1295 } {OK}
1296
1297 catch {
1298 if {[string match {*Darwin*} [exec uname -a]]} {
1299 test {Check for memory leaks} {
1300 exec leaks redis-server
1301 } {*0 leaks*}
1302 }
1303 }
1304
1305 puts "\n[expr $::passed+$::failed] tests, $::passed passed, $::failed failed"
1306 if {$::failed > 0} {
1307 puts "\n*** WARNING!!! $::failed FAILED TESTS ***\n"
1308 }
1309 }
1310
1311 proc stress {} {
1312 set r [redis]
1313 $r select 9
1314 $r flushdb
1315 while 1 {
1316 set randkey [expr int(rand()*10000)]
1317 set randval [expr int(rand()*10000)]
1318 set randidx0 [expr int(rand()*10)]
1319 set randidx1 [expr int(rand()*10)]
1320 set cmd [expr int(rand()*20)]
1321 catch {
1322 if {$cmd == 0} {$r set $randkey $randval}
1323 if {$cmd == 1} {$r get $randkey}
1324 if {$cmd == 2} {$r incr $randkey}
1325 if {$cmd == 3} {$r lpush $randkey $randval}
1326 if {$cmd == 4} {$r rpop $randkey}
1327 if {$cmd == 5} {$r del $randkey}
1328 if {$cmd == 6} {$r llen $randkey}
1329 if {$cmd == 7} {$r lrange $randkey $randidx0 $randidx1}
1330 if {$cmd == 8} {$r ltrim $randkey $randidx0 $randidx1}
1331 if {$cmd == 9} {$r lindex $randkey $randidx0}
1332 if {$cmd == 10} {$r lset $randkey $randidx0 $randval}
1333 if {$cmd == 11} {$r sadd $randkey $randval}
1334 if {$cmd == 12} {$r srem $randkey $randval}
1335 if {$cmd == 13} {$r smove $randkey $randval}
1336 if {$cmd == 14} {$r scard $randkey}
1337 if {$cmd == 15} {$r expire $randkey [expr $randval%60]}
1338 }
1339 flush stdout
1340 }
1341 $r flushdb
1342 $r close
1343 }
1344
1345 # Set a few configuration defaults
1346 set ::host 127.0.0.1
1347 set ::port 6379
1348 set ::stress 0
1349 set ::flush 0
1350 set ::first 0
1351 set ::last 1000000
1352
1353 # Parse arguments
1354 for {set j 0} {$j < [llength $argv]} {incr j} {
1355 set opt [lindex $argv $j]
1356 set arg [lindex $argv [expr $j+1]]
1357 set lastarg [expr {$arg eq {}}]
1358 if {$opt eq {-h} && !$lastarg} {
1359 set ::host $arg
1360 incr j
1361 } elseif {$opt eq {-p} && !$lastarg} {
1362 set ::port $arg
1363 incr j
1364 } elseif {$opt eq {-stress}} {
1365 set ::stress 1
1366 } elseif {$opt eq {--flush}} {
1367 set ::flush 1
1368 } elseif {$opt eq {--first} && !$lastarg} {
1369 set ::first $arg
1370 incr j
1371 } elseif {$opt eq {--last} && !$lastarg} {
1372 set ::last $arg
1373 incr j
1374 } else {
1375 puts "Wrong argument: $opt"
1376 exit 1
1377 }
1378 }
1379
1380 # Before to run the test check if DB 9 and DB 10 are empty
1381 set r [redis]
1382
1383 if {$::flush} {
1384 $r flushall
1385 }
1386
1387 $r select 9
1388 set db9size [$r dbsize]
1389 $r select 10
1390 set db10size [$r dbsize]
1391 if {$db9size != 0 || $db10size != 0} {
1392 puts "Can't run the tests against DB 9 and 10: DBs are not empty."
1393 exit 1
1394 }
1395 $r close
1396 unset r
1397 unset db9size
1398 unset db10size
1399
1400 if {$::stress} {
1401 stress
1402 } else {
1403 main $::host $::port
1404 }