tests/unit/sort.tcl

   1 start_server {
   2     tags {"sort"}
   3     overrides {
   4         "list-max-ziplist-value" 16
   5         "list-max-ziplist-entries" 32
   6         "set-max-intset-entries" 32
   7     }
   8 } {
   9     proc create_random_dataset {num cmd} {
  10         set tosort {}
  11         set result {}
  12         array set seenrand {}
  13         r del tosort
  14         for {set i 0} {$i < $num} {incr i} {
  15             # Make sure all the weights are different because
  16             # Redis does not use a stable sort but Tcl does.
  17             while 1 {
  18                 randpath {
  19                     set rint [expr int(rand()*1000000)]
  20                 } {
  21                     set rint [expr rand()]
  22                 }
  23                 if {![info exists seenrand($rint)]} break
  24             }
  25             set seenrand($rint) x
  26             r $cmd tosort $i
  27             r set weight_$i $rint
  28             r hset wobj_$i weight $rint
  29             lappend tosort [list $i $rint]
  30         }
  31         set sorted [lsort -index 1 -real $tosort]
  32         for {set i 0} {$i < $num} {incr i} {
  33             lappend result [lindex $sorted $i 0]
  34         }
  35         set _ $result
  36     }
  37
  38     foreach {num cmd enc title} {
  39         16 lpush ziplist "Ziplist"
  40         1000 lpush linkedlist "Linked list"
  41         10000 lpush linkedlist "Big Linked list"
  42         16 sadd intset "Intset"
  43         1000 sadd hashtable "Hash table"
  44         10000 sadd hashtable "Big Hash table"
  45     } {
  46         set result [create_random_dataset $num $cmd]
  47         assert_encoding $enc tosort
  48
  49         test "$title: SORT BY key" {
  50             assert_equal $result [r sort tosort BY weight_*]
  51         }
  52
  53         test "$title: SORT BY key with limit" {
  54             assert_equal [lrange $result 5 9] [r sort tosort BY weight_* LIMIT 5 5]
  55         }
  56
  57         test "$title: SORT BY hash field" {
  58             assert_equal $result [r sort tosort BY wobj_*->weight]
  59         }
  60     }
  61
  62     set result [create_random_dataset 16 lpush]
  63     test "SORT GET #" {
  64         assert_equal [lsort -integer $result] [r sort tosort GET #]
  65     }
  66
  67     test "SORT GET <const>" {
  68         r del foo
  69         set res [r sort tosort GET foo]
  70         assert_equal 16 [llength $res]
  71         foreach item $res { assert_equal {} $item }
  72     }
  73
  74     test "SORT GET (key and hash) with sanity check" {
  75         set l1 [r sort tosort GET # GET weight_*]
  76         set l2 [r sort tosort GET # GET wobj_*->weight]
  77         foreach {id1 w1} $l1 {id2 w2} $l2 {
  78             assert_equal $id1 $id2
  79             assert_equal $w1 [r get weight_$id1]
  80             assert_equal $w2 [r get weight_$id1]
  81         }
  82     }
  83
  84     test "SORT BY key STORE" {
  85         r sort tosort BY weight_* store sort-res
  86         assert_equal $result [r lrange sort-res 0 -1]
  87         assert_equal 16 [r llen sort-res]
  88         assert_encoding ziplist sort-res
  89     }
  90
  91     test "SORT BY hash field STORE" {
  92         r sort tosort BY wobj_*->weight store sort-res
  93         assert_equal $result [r lrange sort-res 0 -1]
  94         assert_equal 16 [r llen sort-res]
  95         assert_encoding ziplist sort-res
  96     }
  97
  98     test "SORT DESC" {
  99         assert_equal [lsort -decreasing -integer $result] [r sort tosort DESC]
 100     }
 101
 102     test "SORT ALPHA against integer encoded strings" {
 103         r del mylist
 104         r lpush mylist 2
 105         r lpush mylist 1
 106         r lpush mylist 3
 107         r lpush mylist 10
 108         r sort mylist alpha
 109     } {1 10 2 3}
 110
 111     test "SORT sorted set" {
 112         r del zset
 113         r zadd zset 1 a
 114         r zadd zset 5 b
 115         r zadd zset 2 c
 116         r zadd zset 10 d
 117         r zadd zset 3 e
 118         r sort zset alpha desc
 119     } {e d c b a}
 120
 121     test "SORT sorted set: +inf and -inf handling" {
 122         r del zset
 123         r zadd zset -100 a
 124         r zadd zset 200 b
 125         r zadd zset -300 c
 126         r zadd zset 1000000 d
 127         r zadd zset +inf max
 128         r zadd zset -inf min
 129         r zrange zset 0 -1
 130     } {min c a b d max}
 131
 132     test "SORT regression for issue #19, sorting floats" {
 133         r flushdb
 134         set floats {1.1 5.10 3.10 7.44 2.1 5.75 6.12 0.25 1.15}
 135         foreach x $floats {
 136             r lpush mylist $x
 137         }
 138         assert_equal [lsort -real $floats] [r sort mylist]
 139     }
 140
 141     test "SORT with STORE returns zero if result is empty (github isse 224)" {
 142         r flushdb
 143         r sort foo store bar
 144     } {0}
 145
 146     test "SORT with STORE does not create empty lists (github issue 224)" {
 147         r flushdb
 148         r lpush foo bar
 149         r sort foo limit 10 10 store zap
 150         r exists zap
 151     } {0}
 152
 153     tags {"slow"} {
 154         set num 100
 155         set res [create_random_dataset $num lpush]
 156
 157         test "SORT speed, $num element list BY key, 100 times" {
 158             set start [clock clicks -milliseconds]
 159             for {set i 0} {$i < 100} {incr i} {
 160                 set sorted [r sort tosort BY weight_* LIMIT 0 10]
 161             }
 162             set elapsed [expr [clock clicks -milliseconds]-$start]
 163             if {$::verbose} {
 164                 puts -nonewline "\n  Average time to sort: [expr double($elapsed)/100] milliseconds "
 165                 flush stdout
 166             }
 167         }
 168
 169         test "SORT speed, $num element list BY hash field, 100 times" {
 170             set start [clock clicks -milliseconds]
 171             for {set i 0} {$i < 100} {incr i} {
 172                 set sorted [r sort tosort BY wobj_*->weight LIMIT 0 10]
 173             }
 174             set elapsed [expr [clock clicks -milliseconds]-$start]
 175             if {$::verbose} {
 176                 puts -nonewline "\n  Average time to sort: [expr double($elapsed)/100] milliseconds "
 177                 flush stdout
 178             }
 179         }
 180
 181         test "SORT speed, $num element list directly, 100 times" {
 182             set start [clock clicks -milliseconds]
 183             for {set i 0} {$i < 100} {incr i} {
 184                 set sorted [r sort tosort LIMIT 0 10]
 185             }
 186             set elapsed [expr [clock clicks -milliseconds]-$start]
 187             if {$::verbose} {
 188                 puts -nonewline "\n  Average time to sort: [expr double($elapsed)/100] milliseconds "
 189                 flush stdout
 190             }
 191         }
 192
 193         test "SORT speed, $num element list BY <const>, 100 times" {
 194             set start [clock clicks -milliseconds]
 195             for {set i 0} {$i < 100} {incr i} {
 196                 set sorted [r sort tosort BY nokey LIMIT 0 10]
 197             }
 198             set elapsed [expr [clock clicks -milliseconds]-$start]
 199             if {$::verbose} {
 200                 puts -nonewline "\n  Average time to sort: [expr double($elapsed)/100] milliseconds "
 201                 flush stdout
 202             }
 203         }
 204     }
 205 }