diff options
| -rw-r--r-- | tests/cluster/cluster.tcl | 19 | ||||
| -rw-r--r-- | tests/cluster/tests/00-base.tcl | 10 |
2 files changed, 29 insertions, 0 deletions
diff --git a/tests/cluster/cluster.tcl b/tests/cluster/cluster.tcl index f85f96fd6..824f3a80f 100644 --- a/tests/cluster/cluster.tcl +++ b/tests/cluster/cluster.tcl @@ -42,3 +42,22 @@ proc get_myself id { return {} } +# Return the value of the specified CLUSTER INFO field. +proc CI {n field} { + get_info_field [R $n cluster info] $field +} + +# Assuming nodes are reest, this function performs slots allocation. +# Only the first 'n' nodes are used. +proc cluster_allocate_slots {n} { + set slot 16383 + while {$slot >= 0} { + # Allocate successive slots to random nodes. + set node [randomInt $n] + lappend slots_$node $slot + incr slot -1 + } + for {set j 0} {$j < $n} {incr j} { + R $j cluster addslots {*}[set slots_${j}] + } +} diff --git a/tests/cluster/tests/00-base.tcl b/tests/cluster/tests/00-base.tcl index 6eb81f897..4777414e7 100644 --- a/tests/cluster/tests/00-base.tcl +++ b/tests/cluster/tests/00-base.tcl @@ -51,6 +51,16 @@ test "Check if nodes auto-discovery works" { } } +test "Before slots allocation, all nodes report cluster failure" { + foreach_redis_id id { + assert {[CI $id cluster_state] eq {fail}} + } +} + +test "It is possible to perform slot allocation" { + cluster_allocate_slots 5 +} + test "After the join, every node gets a different config epoch" { set trynum 60 while {[incr trynum -1] != 0} { |