SERVER-26283 add tests for balancer shell commands

1 files changed, 33 insertions, 0 deletions

diff --git a/jstests/sharding/balancer_shell_commands.js b/jstests/sharding/balancer_shell_commands.js
new file mode 100644
index 00000000000..13bb89335f6
--- /dev/null
+++ b/jstests/sharding/balancer_shell_commands.js
@@ -0,0 +1,33 @@
+/**
+ * Test shell balancer commands.
+ *  sh.setBalancerState
+ *  sh.getBalancerLockDetails
+ *  sh.getBalancerState
+ */
+
+var db;
+
+(function() {
+    "use strict";
+    var shardingTest = new ShardingTest(
+        {name: "shell_commands", shards: 1, mongos: 1, other: {enableBalancer: true}});
+    db = shardingTest.getDB("test");
+
+    assert(sh.getBalancerState(), "Balancer should have been enabled during cluster setup");
+
+    // Test that the balancer can be disabled and the balancer lock is still locked.
+    sh.setBalancerState(false);
+    var lock = sh.getBalancerLockDetails();
+    assert.eq("balancer", lock._id);
+    assert.eq(2, lock.state);
+    assert(!sh.getBalancerState(), "Failed to disable balancer");
+
+    // Test that the balancer can be enabled and the balancer lock is still locked.
+    sh.setBalancerState(true);
+    lock = sh.getBalancerLockDetails();
+    assert.eq("balancer", lock._id);
+    assert.eq(2, lock.state);
+    assert(sh.getBalancerState(), "Failed to enable balancer");
+
+    shardingTest.stop();
+})();