ZLEN renamed ZCARD for consistency with SCARD

[redis.git] / test-redis.tcl

diff --git a/test-redis.tcl b/test-redis.tcl
index 1672390346088270a423f5e33a80e78f558d9470..1eee7db3caa56d14c2dee1177429f7e314f09d75 100644 (file)
--- a/test-redis.tcl
+++ b/test-redis.tcl
@@ -38,6 +38,13 @@ proc randstring {min max {type binary}} {
     return $output
 }
 
     return $output
 }
 

+# Useful for some test
+proc zlistAlikeSort {a b} {
+    if {[lindex $a 0] > [lindex $b 0]} {return 1}
+    if {[lindex $a 0] < [lindex $b 0]} {return -1}
+    string compare [lindex $a 1] [lindex $b 1]
+}
+

 proc main {server port} {
     set r [redis $server $port]
     set err ""
 proc main {server port} {
     set r [redis $server $port]
     set err ""


@@ -481,11 +488,6 @@ proc main {server port} {
         lsort [$r smembers setres]
     } [lsort -uniq "[$r smembers set1] [$r smembers set2]"]
 
         lsort [$r smembers setres]
     } [lsort -uniq "[$r smembers set1] [$r smembers set2]"]
 

-    test {SUNIONSTORE with same src and dest} {
-        $r sunionstore set1 set1 set1
-        $r scard set1
-    } {1000}
-

     test {SINTER against three sets} {
         $r sadd set3 999
         $r sadd set3 995
     test {SINTER against three sets} {
         $r sadd set3 999
         $r sadd set3 995


@@ -499,11 +501,6 @@ proc main {server port} {
         lsort [$r smembers setres]
     } {995 999}
 
         lsort [$r smembers setres]
     } {995 999}
 

-    test {SINTERSTORE with same src and dest} {
-       $r sinterstore set1 set1 set1
-       $r scard set1
-    } {1000}
-

     test {SUNION with non existing keys} {
         lsort [$r sunion nokey1 set1 set2 nokey2]
     } [lsort -uniq "[$r smembers set1] [$r smembers set2]"]
     test {SUNION with non existing keys} {
         lsort [$r sunion nokey1 set1 set2 nokey2]
     } [lsort -uniq "[$r smembers set1] [$r smembers set2]"]


@@ -525,11 +522,22 @@ proc main {server port} {
         lsort [$r smembers sres]
     } {1 2 3 4}
 
         lsort [$r smembers sres]
     } {1 2 3 4}
 

+    test {SPOP basics} {
+        $r del myset
+        $r sadd myset 1
+        $r sadd myset 2
+        $r sadd myset 3
+        list [lsort [list [$r spop myset] [$r spop myset] [$r spop myset]]] [$r scard myset]
+    } {{1 2 3} 0}
+

     test {SAVE - make sure there are all the types as values} {
         $r lpush mysavelist hello
         $r lpush mysavelist world
         $r set myemptykey {}
         $r set mynormalkey {blablablba}
     test {SAVE - make sure there are all the types as values} {
         $r lpush mysavelist hello
         $r lpush mysavelist world
         $r set myemptykey {}
         $r set mynormalkey {blablablba}

+        $r zadd mytestzset a 10
+        $r zadd mytestzset b 20
+        $r zadd mytestzset c 30

         $r save
     } {OK}
     
         $r save
     } {OK}
     


@@ -653,6 +661,14 @@ proc main {server port} {
         list [$r lrange mylist 0 -1] $res
     } {{foo bar foobar foobared zap test} 2}
 
         list [$r lrange mylist 0 -1] $res
     } {{foo bar foobar foobared zap test} 2}
 

+    test {LREM, deleting objects that may be encoded as integers} {
+        $r lpush myotherlist 1
+        $r lpush myotherlist 2
+        $r lpush myotherlist 3
+        $r lrem myotherlist 1 2
+        $r llen myotherlist
+    } {2}
+

     test {MGET} {
         $r flushall
         $r set foo BAR
     test {MGET} {
         $r flushall
         $r set foo BAR


@@ -744,19 +760,118 @@ proc main {server port} {
         format $err
     } {ERR*}
 
         format $err
     } {ERR*}
 

+    test {MSET base case} {
+        $r mset x 10 y "foo bar" z "x x x x x x x\n\n\r\n"
+        $r mget x y z
+    } [list 10 {foo bar} "x x x x x x x\n\n\r\n"]
+
+    test {MSET wrong number of args} {
+        catch {$r mset x 10 y "foo bar" z} err
+        format $err
+    } {*wrong number*}
+
+    test {MSETNX with already existent key} {
+        list [$r msetnx x1 xxx y2 yyy x 20] [$r exists x1] [$r exists y2]
+    } {0 0 0}
+
+    test {MSETNX with not existing keys} {
+        list [$r msetnx x1 xxx y2 yyy] [$r get x1] [$r get y2]
+    } {1 xxx yyy}
+
+    test {ZSET basic ZADD and score update} {
+        $r zadd ztmp 10 x
+        $r zadd ztmp 20 y
+        $r zadd ztmp 30 z
+        set aux1 [$r zrange ztmp 0 -1]
+        $r zadd ztmp 1 y
+        set aux2 [$r zrange ztmp 0 -1]
+        list $aux1 $aux2
+    } {{x y z} {y x z}}
+
+    test {ZSCORE} {
+        list [$r zscore ztmp x] [$r zscore ztmp y] [$r zscore ztmp z]
+    } {10 1 30}
+
+    test {ZRANGE and ZREVRANGE} {
+        list [$r zrange ztmp 0 -1] [$r zrevrange ztmp 0 -1]
+    } {{y x z} {z x y}}
+
+    test {ZSETs stress tester - sorting is working well?} {
+        set delta 0
+        for {set test 0} {$test < 2} {incr test} {
+            unset -nocomplain auxarray
+            array set auxarray {}
+            set auxlist {}
+            $r del myzset
+            for {set i 0} {$i < 1000} {incr i} {
+                if {$test == 0} {
+                    set score [expr rand()]
+                } else {
+                    set score [expr int(rand()*10)]
+                }
+                set auxarray($i) $score
+                $r zadd myzset $score $i
+                # Random update
+                if {[expr rand()] < .2} {
+                    set j [expr int(rand()*1000)]
+                    if {$test == 0} {
+                        set score [expr rand()]
+                    } else {
+                        set score [expr int(rand()*10)]
+                    }
+                    set auxarray($j) $score
+                    $r zadd myzset $score $j
+                }
+            }
+            foreach {item score} [array get auxarray] {
+                lappend auxlist [list $score $item]
+            }
+            set sorted [lsort -command zlistAlikeSort $auxlist]
+            set auxlist {}
+            foreach x $sorted {
+                lappend auxlist [lindex $x 1]
+            }
+            set fromredis [$r zrange myzset 0 -1]
+            set delta 0
+            for {set i 0} {$i < [llength $fromredis]} {incr i} {
+                if {[lindex $fromredis $i] != [lindex $auxlist $i]} {
+                    incr delta
+                }
+            }
+        }
+        format $delta
+    } {0}
+
+    test {ZSETs skiplist implementation backlink consistency test} {
+        set diff 0
+        set elements 10000
+        for {set j 0} {$j < $elements} {incr j} {
+            $r zadd myzset [expr rand()] "Element-$j"
+            $r zrem myzset "Element-[expr int(rand()*$elements)]"
+        }
+        set l1 [$r zrange myzset 0 -1]
+        set l2 [$r zrevrange myzset 0 -1]
+        for {set j 0} {$j < [llength $l1]} {incr j} {
+            if {[lindex $l1 $j] ne [lindex $l2 end-$j]} {
+                incr diff
+            }
+        }
+        format $diff
+    } {0}
+

     foreach fuzztype {binary alpha compr} {
         test "FUZZ stresser with data model $fuzztype" {
             set err 0
     foreach fuzztype {binary alpha compr} {
         test "FUZZ stresser with data model $fuzztype" {
             set err 0

-            for {set i 0} {$i < 1000} {incr i} {
+            for {set i 0} {$i < 10000} {incr i} {

                 set fuzz [randstring 0 512 $fuzztype]
                 $r set foo $fuzz
                 set got [$r get foo]
                 if {$got ne $fuzz} {
                 set fuzz [randstring 0 512 $fuzztype]
                 $r set foo $fuzz
                 set got [$r get foo]
                 if {$got ne $fuzz} {

-                    incr err
+                    set err [list $fuzz $got]

                     break
                 }
             }
                     break
                 }
             }

-            format $err
+            set _ $err

         } {0}
     }
 
         } {0}
     }