SERVER-22661 Add initialization from shard identity doc

1 files changed, 51 insertions, 0 deletions

diff --git a/jstests/sharding/shard_aware_primary_failover.js b/jstests/sharding/shard_aware_primary_failover.js
new file mode 100644
index 00000000000..6fac918494e
--- /dev/null
+++ b/jstests/sharding/shard_aware_primary_failover.js
@@ -0,0 +1,51 @@
+/**
+ * Test that a new primary that gets elected will properly perform shard initialization.
+ */
+
+(function() {
+    "use strict";
+
+    var st = new ShardingTest({shards: 1});
+
+    var replTest = new ReplSetTest({nodes: 3});
+    replTest.startSet();
+    var nodes = replTest.nodeList();
+    replTest.initiate({
+        _id: replTest.name,
+        members: [
+            {_id: 0, host: nodes[0]},
+            {_id: 1, host: nodes[1]},
+            {_id: 2, host: nodes[2], arbiterOnly: true}
+        ],
+        protocolVersion: 1
+    });
+
+    var primaryConn = replTest.getPrimary();
+
+    var shardIdentityDoc = {
+        _id: 'shardIdentity',
+        configsvrConnectionString: st.configRS.getURL(),
+        shardName: 'newShard',
+        clusterId: ObjectId()
+    };
+
+    assert.writeOK(primaryConn.getDB('admin')
+                       .system.version.insert(shardIdentityDoc, {writeConcern: {w: 'majority'}}));
+
+    replTest.stopMaster();
+    replTest.waitForMaster();
+
+    primaryConn = replTest.getPrimary();
+
+    var res = primaryConn.getDB('admin').runCommand({shardingState: 1});
+
+    assert(res.enabled);
+    assert.eq(shardIdentityDoc.configsvrConnectionString, res.configServer);
+    assert.eq(shardIdentityDoc.shardName, res.shardName);
+    assert.eq(shardIdentityDoc.clusterId, res.clusterId);
+
+    replTest.stopSet();
+
+    st.stop();
+
+})();