1 files changed, 121 insertions, 0 deletions
diff --git a/jstests/serverless/shard_split_drop_state_doc_collection_decision_fullfilled.js b/jstests/serverless/shard_split_drop_state_doc_collection_decision_fullfilled.js
new file mode 100644
index 00000000000..1c5120d3e95
--- /dev/null
+++ b/jstests/serverless/shard_split_drop_state_doc_collection_decision_fullfilled.js
@@ -0,0 +1,121 @@
+/**
+ * Tests dropping the donor state doc collections after the shard split decision promise is
+ * fulfilled.
+ *
+ * @tags: [
+ *   incompatible_with_eft,
+ *   incompatible_with_macos,
+ *   incompatible_with_windows_tls,
+ *   requires_majority_read_concern,
+ *   requires_persistence,
+ *   serverless,
+ *   requires_fcv_52,
+ *   featureFlagShardSplit
+ * ]
+ */
+
+(function() {
+"use strict";
+
+load("jstests/libs/fail_point_util.js");
+load("jstests/libs/uuid_util.js");
+load("jstests/serverless/libs/basic_serverless_test.js");
+
+const recipientTagName = "recipientNode";
+const recipientSetName = "recipient";
+
+TestData.skipCheckDBHashes = true;
+
+function testDroppingStateDocCollections(
+    test,
+    fpName,
+    {dropDonorsCollection = false, retryWithDifferentMigrationId = false, expectedAbortReason}) {
+    jsTest.log(`Testing with failpoint: ${fpName} dropDonorsCollection: ${
+        dropDonorsCollection}, retryWithDifferentMigrationId: ${retryWithDifferentMigrationId}`);
+
+    test.addRecipientNodes();
+    let donorPrimary = test.donor.getPrimary();
+
+    const tenantIds = ["tenant1", "tenant2"];
+
+    const operation = test.createSplitOperation(tenantIds);
+    let migrationId = operation.migrationId;
+
+    let commitShardSplitThread = undefined;
+
+    let fp = configureFailPoint(donorPrimary.getDB("admin"), fpName);
+    commitShardSplitThread = operation.commitAsync();
+    fp.wait();
+
+    if (dropDonorsCollection) {
+        assert(donorPrimary.getCollection(BasicServerlessTest.kConfigSplitDonorsNS).drop());
+        let donorDoc = findSplitOperation(donorPrimary, migrationId);
+        assert.eq(donorDoc, null);
+
+        const currOpDonor = assert.commandWorked(
+            donorPrimary.adminCommand({currentOp: true, desc: "shard split operation"}));
+        assert.eq(currOpDonor.inprog.length, 0);
+
+        // Trigger stepup to allow the donor service to rebuild.
+        assert.commandWorked(donorPrimary.adminCommand({replSetStepDown: 30, force: true}));
+    }
+
+    fp.off();
+    commitShardSplitThread.join();
+    if (expectedAbortReason) {
+        const data = commitShardSplitThread.returnData();
+        assert.commandFailedWithCode(data, expectedAbortReason);
+        assert.eq(data.code, expectedAbortReason);
+    }
+    test.removeRecipientNodesFromDonor();
+    if (!dropDonorsCollection) {
+        operation.forget();
+        test.waitForGarbageCollection(migrationId, tenantIds);
+    }
+    test.removeAndStopRecipientNodes();
+    test.reconfigDonorSetAfterSplit();
+
+    test.addRecipientNodes();
+
+    const operation2 =
+        retryWithDifferentMigrationId ? test.createSplitOperation(tenantIds) : operation;
+    migrationId = operation2.migrationId;
+    const runMigrationRes = operation2.commit();
+
+    assert.commandWorked(runMigrationRes);
+    assert.eq(runMigrationRes.state, "committed");
+
+    operation2.forget();
+
+    test.cleanupSuccesfulCommitted(migrationId, tenantIds);
+}
+
+jsTest.log("Test dropping donor and recipient state doc collections during a shard split.");
+const test = new BasicServerlessTest({
+    recipientTagName,
+    recipientSetName,
+    quickGarbageCollection: true,
+    initiateWithShortElectionTimeout: true
+});
+
+const fpName = "pauseShardSplitAfterDecision";
+testDroppingStateDocCollections(test, fpName, {dropDonorsCollection: true});
+
+testDroppingStateDocCollections(
+    test, fpName, {dropDonorsCollection: true, retryWithDifferentMigrationId: true});
+
+if (fpName) {
+    // if we do not have a failpoint and we do not drop the donor state doc there is no need
+    // to run a test.
+    testDroppingStateDocCollections(test, fpName, {dropDonorsCollection: false});
+
+    testDroppingStateDocCollections(test, fpName, {
+        dropDonorsCollection: false,
+        expectedAbortReason: (fpName == "abortShardSplitBeforeLeavingBlockingState")
+            ? ErrorCodes.TenantMigrationAborted
+            : null
+    });
+}
+
+test.stop();
+})();