SERVER-24298 add ensurePrimaryShard to csrs_upgrade_mongod_using_movechunk.js

1 files changed, 5 insertions, 0 deletions

diff --git a/jstests/sharding/csrs_upgrade/csrs_upgrade_mongod_using_movechunk.js b/jstests/sharding/csrs_upgrade/csrs_upgrade_mongod_using_movechunk.js
index 1894a4b3f69..9965d9c1b8d 100644
--- a/jstests/sharding/csrs_upgrade/csrs_upgrade_mongod_using_movechunk.js
+++ b/jstests/sharding/csrs_upgrade/csrs_upgrade_mongod_using_movechunk.js
@@ -21,6 +21,11 @@ var st;
     jsTest.log("Enabling sharding on " + testDBName);
     assert.commandWorked(st.s0.adminCommand({enablesharding: testDBName}));
 
+    // Ensure shard0 as the primary shard so that after it is restarted, it can still perform a
+    // moveChunk (if it had no chunks, the moveChunk would fail).
+    jsTest.log("Ensuring " + st.shard0.name + " as primary shard");
+    st.ensurePrimaryShard(testDBName, st.shard0.name);
+
     jsTest.log("Creating a sharded collection " + dataCollectionName);
     assert.commandWorked(st.s0.adminCommand({shardcollection: dataCollectionName, key: {_id: 1}}));