SERVER-63817 adding support of forgetShardSplit cmd to the shard_split_wait_for_block_timestamps

1 files changed, 4 insertions, 5 deletions

diff --git a/jstests/serverless/shard_split_wait_for_block_timestamp.js b/jstests/serverless/shard_split_wait_for_block_timestamp.js
index 0416a9f0261..2b95fedfa37 100644
--- a/jstests/serverless/shard_split_wait_for_block_timestamp.js
+++ b/jstests/serverless/shard_split_wait_for_block_timestamp.js
@@ -62,9 +62,8 @@ const awaitFirstSplitOperation = startParallelShell(
 awaitFirstSplitOperation();
 assertMigrationState(donorPrimary, firstOperationId, "aborted");
 
-jsTestLog(`Removing state document: ${tojson(firstOperationId)}`);
-// TODO(SERVER-62366): use `forgetShardSplit` instead of manually deleting the state document.
-assert.commandWorked(cleanupMigrationDocument(donorPrimary, firstOperationId));
+jsTestLog(`Running forgetShardSplit command: ${tojson(firstOperationId)}`);
+assert.commandWorked(adminDb.runCommand({forgetShardSplit: 1, migrationId: firstOperationId}));
 
 jsTestLog("Restarting replication on recipient nodes, and running new split operation");
 test.recipientNodes.forEach(node => restartServerReplication(node));
@@ -89,8 +88,8 @@ const awaitSecondSplitOperation = startParallelShell(
 awaitSecondSplitOperation();
 assertMigrationState(donorPrimary, secondOperationId, "committed");
 
-jsTestLog("Running forgetShardSplit command");
-assert.commandWorked(adminDb.runCommand({forgetShardSplit: 1, migrationId}));
+jsTestLog(`Running forgetShardSplit command: ${tojson(secondOperationId)}`);
+assert.commandWorked(adminDb.runCommand({forgetShardSplit: 1, migrationId: secondOperationId}));
 
 test.stop();
 })();