From 29f8ca9ec2fef177f98928ee4c1ec6168db5cf87 Mon Sep 17 00:00:00 2001
From: Tess Avitabile <tess.avitabile@mongodb.com>
Date: Thu, 21 May 2020 14:26:24 -0400
Subject: SERVER-48371 transactions_during_step_down.js must abort transaction
 in the shell

(cherry picked from commit ec3fe1a2b7cbf6586b20373fdca1e60dd9e406d4)
---
 jstests/replsets/transactions_during_step_down.js | 40 +++++++++++++----------
 1 file changed, 22 insertions(+), 18 deletions(-)

diff --git a/jstests/replsets/transactions_during_step_down.js b/jstests/replsets/transactions_during_step_down.js
index c74062cb5e5..6e73475ad73 100644
--- a/jstests/replsets/transactions_during_step_down.js
+++ b/jstests/replsets/transactions_during_step_down.js
@@ -31,20 +31,17 @@ TestData.dbName = dbName;
 TestData.collName = collName;
 TestData.skipRetryOnNetworkError = true;
 
-function startTxn({parallel: parallel = true}) {
-    var txnFunc = () => {
-        jsTestLog("Starting a new transaction.");
-        const session = db.getMongo().startSession();
-        const sessionDb = session.getDatabase(TestData.dbName);
-        const sessionColl = sessionDb[TestData.collName];
-        session.startTransaction({writeConcern: {w: "majority"}});
-        print(TestData.cmd);
-        eval(TestData.cmd);
-
-        // Validate that the connection is not closed on step down.
-        assert.commandWorked(db.adminCommand({ping: 1}));
-    };
-    return parallel ? startParallelShell(txnFunc, primary.port) : txnFunc();
+function txnFunc() {
+    jsTestLog("Starting a new transaction.");
+    const session = db.getMongo().startSession();
+    const sessionDb = session.getDatabase(TestData.dbName);
+    const sessionColl = sessionDb[TestData.collName];
+    session.startTransaction({writeConcern: {w: "majority"}});
+    print(TestData.cmd);
+    eval(TestData.cmd);
+
+    // Validate that the connection is not closed on step down.
+    assert.commandWorked(db.adminCommand({ping: 1}));
 }
 
 function runStepDown() {
@@ -78,7 +75,7 @@ function testTxnFailsWithCode({
     // Start transaction.
     TestData.cmd =
         preOp + `assert.commandFailedWithCode(${op}, ErrorCodes.InterruptedDueToReplStateChange);`;
-    const waitForTxnShell = startTxn({});
+    const waitForTxnShell = startParallelShell(txnFunc, primary.port);
 
     jsTestLog("Waiting for primary to reach failPoint '" + failPoint + "'.");
     waitForCurOpByFailPoint(primaryAdmin, nss, failPoint);
@@ -120,15 +117,22 @@ testAbortOrCommitTxnFailsWithCode(
     {failPoint: "hangBeforeAbortingTxn", op: "session.abortTransaction_forTesting()"});
 
 jsTestLog("Testing stepdown during running transaction in inactive state.");
-TestData.cmd = "assert.commandWorked(sessionColl.insert({_id: 'inactiveTxnOp'}))";
 // Do not start the transaction in parallel shell because when the parallel
 // shell work is done, implicit call to "endSessions" and "abortTransaction"
 // cmds are made. So, during step down we might not have any running
-// transaction to interrupt.
-startTxn({parallel: false});
+jsTestLog("Starting a new transaction.");
+const session = db.getMongo().startSession();
+const sessionDb = session.getDatabase(TestData.dbName);
+const sessionColl = sessionDb[TestData.collName];
+session.startTransaction({writeConcern: {w: "majority"}});
+assert.commandWorked(sessionColl.insert({_id: 'inactiveTxnOp'}));
 
 // Call step down & validate data.
 runStepDown();
 
+// Even though the transaction was aborted by the stepdown, we must still update the shell's
+// transaction state to aborted.
+assert.commandFailedWithCode(session.abortTransaction_forTesting(), ErrorCodes.NoSuchTransaction);
+
 rst.stopSet();
 })();
-- 
cgit v1.2.1