New Cluster Command: CLUSTER DELSLOTSRANGE and CLUSTER ADDSLOTSRANGE (#9445)

1 files changed, 68 insertions, 0 deletions

diff --git a/tests/cluster/tests/23-multiple-slot-operations.tcl b/tests/cluster/tests/23-multiple-slot-operations.tcl
new file mode 100644
index 000000000..906033c72
--- /dev/null
+++ b/tests/cluster/tests/23-multiple-slot-operations.tcl
@@ -0,0 +1,68 @@
+# Check the multiple slot add and remove commands
+
+source "../tests/includes/init-tests.tcl"
+
+proc cluster_allocate_with_continuous_slots {n} {
+    R 0 cluster ADDSLOTSRANGE 0 3276
+    R 1 cluster ADDSLOTSRANGE 3277 6552
+    R 2 cluster ADDSLOTSRANGE 6553 9828
+    R 3 cluster ADDSLOTSRANGE 9829 13104
+    R 4 cluster ADDSLOTSRANGE 13105 16383
+}
+
+proc cluster_create_with_continuous_slots {masters slaves} {
+    cluster_allocate_with_continuous_slots $masters
+    if {$slaves} {
+        cluster_allocate_slaves $masters $slaves
+    }
+    assert_cluster_state ok
+}
+
+
+test "Create a 5 nodes cluster" {
+    cluster_create_with_continuous_slots 5 5
+}
+
+test "Cluster should start ok" {
+    assert_cluster_state ok
+}
+
+set master1 [Rn 0]
+set master2 [Rn 1]
+set master3 [Rn 2]
+set master4 [Rn 3]
+set master5 [Rn 4]
+
+
+test "Continuous slots distribution" {
+    assert_match "* 0-3276*" [$master1 CLUSTER NODES]
+    assert_match "* 3277-6552*" [$master2 CLUSTER NODES]
+    assert_match "* 6553-9828*" [$master3 CLUSTER NODES]
+    assert_match "* 9829-13104*" [$master4 CLUSTER NODES]
+    assert_match "* 13105-16383*" [$master5 CLUSTER NODES]
+    assert_match "*0 3276*" [$master1 CLUSTER SLOTS]
+    assert_match "*3277 6552*" [$master2 CLUSTER SLOTS]
+    assert_match "*6553 9828*" [$master3 CLUSTER SLOTS]
+    assert_match "*9829 13104*" [$master4 CLUSTER SLOTS]
+    assert_match "*13105 16383*" [$master5 CLUSTER SLOTS]
+
+    $master1 CLUSTER DELSLOTSRANGE 3001 3050
+    assert_match "* 0-3000 3051-3276*" [$master1 CLUSTER NODES]
+    assert_match "*0 3000*3051 3276*" [$master1 CLUSTER SLOTS]
+
+    $master2 CLUSTER DELSLOTSRANGE 5001 5500
+    assert_match "* 3277-5000 5501-6552*" [$master2 CLUSTER NODES]
+    assert_match "*3277 5000*5501 6552*" [$master2 CLUSTER SLOTS]
+
+    $master3 CLUSTER DELSLOTSRANGE 7001 7100 8001 8500
+    assert_match "* 6553-7000 7101-8000 8501-9828*" [$master3 CLUSTER NODES]
+    assert_match "*6553 7000*7101 8000*8501 9828*" [$master3 CLUSTER SLOTS]
+
+    $master4 CLUSTER DELSLOTSRANGE 11001 12000 12101 12200
+    assert_match "* 9829-11000 12001-12100 12201-13104*" [$master4 CLUSTER NODES]
+    assert_match "*9829 11000*12001 12100*12201 13104*" [$master4 CLUSTER SLOTS]
+
+    $master5 CLUSTER DELSLOTSRANGE 13501 14000 15001 16000
+    assert_match "* 13105-13500 14001-15000 16001-16383*" [$master5 CLUSTER NODES]
+    assert_match "*13105 13500*14001 15000*16001 16383*" [$master5 CLUSTER SLOTS]
+}