Cluster test: slots allocation.

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} {