SERVER-22459 multiversion migration session id test

1 files changed, 92 insertions, 0 deletions

diff --git a/jstests/multiVersion/migration_session_id.js b/jstests/multiVersion/migration_session_id.js
new file mode 100644
index 00000000000..418356d39b8
--- /dev/null
+++ b/jstests/multiVersion/migration_session_id.js
@@ -0,0 +1,92 @@
+//
+// Testing migration session ID functionality in migrations between two mongods
+// where either the donor or recipient shard does not have migration session ID
+// implemented. Migrations should still be successful.
+//
+
+load("./jstests/multiVersion/libs/verify_versions.js");
+
+(function() {
+"use strict";
+
+var options = {shards: [{binVersion : "3.0"},
+                        {binVersion : "3.0"},
+                        {binVersion : "3.2"},
+                        {binVersion : "3.2"}],
+               mongos: 1,
+               other: {mongosOptions: {binVersion : "3.0"},
+                       sync: true}
+};
+
+var st = new ShardingTest(options);
+
+assert.binVersion(st.shard0, "3.0");
+assert.binVersion(st.shard1, "3.0");
+assert.binVersion(st.shard2, "3.2");
+assert.binVersion(st.shard3, "3.2");
+assert.binVersion(st.s0, "3.0");
+
+var mongos = st.s0,
+    admin = mongos.getDB('admin'),
+    shards = mongos.getCollection('config.shards').find().toArray(),
+
+    fooDB = "fooTest",
+    fooNS = fooDB + ".foo",
+    fooColl = mongos.getCollection(fooNS),
+    fooDonor = st.shard0,
+    fooRecipient = st.shard2,
+    fooDonorColl = fooDonor.getCollection(fooNS),
+    fooRecipientColl = fooRecipient.getCollection(fooNS),
+
+    barDB = "barTest",
+    barNS = barDB + ".foo",
+    barColl = mongos.getCollection(barNS),
+    barDonor = st.shard3,
+    barRecipient = st.shard1,
+    barDonorColl = barDonor.getCollection(barNS),
+    barRecipientColl = barRecipient.getCollection(barNS);
+
+assert.commandWorked(admin.runCommand({enableSharding: fooDB}));
+assert.commandWorked(admin.runCommand({enableSharding: barDB}));
+st.ensurePrimaryShard(fooDB, shards[0]._id);
+st.ensurePrimaryShard(barDB, shards[3]._id);
+
+assert.commandWorked(admin.runCommand({shardCollection: fooNS, key: {a: 1}}));
+assert.commandWorked(admin.runCommand({split: fooNS, middle: {a: 10}}));
+assert.commandWorked(admin.runCommand({shardCollection: barNS, key: {a: 1}}));
+assert.commandWorked(admin.runCommand({split: barNS, middle: {a: 10}}));
+
+fooColl.insert({a: 0});
+assert.eq(null, fooColl.getDB().getLastError());
+fooColl.insert({a: 10});
+assert.eq(null, fooColl.getDB().getLastError());
+assert.eq(0, fooRecipientColl.count());
+assert.eq(2, fooDonorColl.count());
+assert.eq(2, fooColl.count());
+
+barColl.insert({a: 0});
+assert.eq(null, barColl.getDB().getLastError());
+barColl.insert({a: 10});
+assert.eq(null, barColl.getDB().getLastError());
+assert.eq(0, barRecipientColl.count());
+assert.eq(2, barDonorColl.count());
+assert.eq(2, barColl.count());
+
+/**
+ * Perform two migrations:
+ *      shard0 (v3.0) -> shard2 (v3.2)
+ *      shard3 (v3.2) -> shard1 (v3.0)
+ */
+
+assert.commandWorked(admin.runCommand({moveChunk: fooNS, find: {a: 10}, to: shards[2]._id}));
+assert.commandWorked(admin.runCommand({moveChunk: barNS, find: {a: 10}, to: shards[1]._id}));
+assert.eq(1, fooRecipientColl.count(), "Foo collection migration failed.");
+assert.eq(1, fooDonorColl.count(), "Foo donor lost its document");
+assert.eq(2, fooColl.count(), "Incorrect number of documents in foo collection");
+assert.eq(1, barRecipientColl.count(), "Bar collection migration failed.");
+assert.eq(1, barDonorColl.count(), "Bar donor lost its document");
+assert.eq(2, barColl.count(), "Incorrect number of documents in bar collection");
+
+st.stop();
+
+})();