SERVER-48107: Retry replSetStepDown in rollback_test.js

(cherry picked from commit 5edffc61e61b004811d7cde7728d9908aff4acc5)

1 files changed, 4 insertions, 1 deletions

diff --git a/jstests/replsets/libs/rollback_test.js b/jstests/replsets/libs/rollback_test.js
index f0d47043797..723bc1c6cef 100644
--- a/jstests/replsets/libs/rollback_test.js
+++ b/jstests/replsets/libs/rollback_test.js
@@ -367,7 +367,10 @@ function RollbackTest(name = "RollbackTest", replSet) {
         try {
             // The stepdown freeze period is short because the node is disconnected from
             // the rest of the replica set, so it physically can't become the primary.
-            curPrimary.adminCommand({replSetStepDown: 1, force: true});
+            assert.soon(() => {
+                const res = curPrimary.adminCommand({replSetStepDown: 1, force: true});
+                return (res.ok || res.code === ErrorCodes.NotMaster);
+            });
         } catch (e) {
             // Stepdown may fail if the node has already started stepping down.
             print('Caught exception from replSetStepDown: ' + e);