SERVER-47645 Must invalidate all sessions on stepdown

(cherry picked from commit 5f1a69aaf69bc12124f68e7b489a1437f9cdd575)

1 files changed, 67 insertions, 0 deletions

diff --git a/jstests/replsets/invalidate_sessions_on_stepdown.js b/jstests/replsets/invalidate_sessions_on_stepdown.js
new file mode 100644
index 00000000000..50916865780
--- /dev/null
+++ b/jstests/replsets/invalidate_sessions_on_stepdown.js
@@ -0,0 +1,67 @@
+/**
+ * Tests that a txnNumber for a transaction that was aborted in-memory can be reused by a new
+ * primary. Mongos does this as an optimization.
+ * @tags: [uses_transactions, uses_prepare_transaction]
+ */
+(function() {
+"use strict";
+
+const rst = new ReplSetTest({name: "invalidate_sessions_on_stepdown", nodes: 2});
+rst.startSet();
+rst.initiateWithHighElectionTimeout();
+
+const dbName = "test";
+const collName = "coll";
+const node0 = rst.getPrimary();
+const node1 = rst.getSecondary();
+const node0Session = node0.startSession();
+const sessionId = node0Session.getSessionId();
+const node0SessionDB = node0Session.getDatabase(dbName);
+
+assert.commandWorked(node0SessionDB.coll.insert({a: 1}, {writeConcern: {"w": "majority"}}));
+
+jsTestLog("Run a transaction with txnNumber 0 on the primary.");
+assert.commandWorked(node0SessionDB.runCommand({
+    insert: collName,
+    documents: [{b: 1}],
+    txnNumber: NumberLong(0),
+    startTransaction: true,
+    autocommit: false
+}));
+
+jsTestLog("Step up the secondary. The primary will abort the transaction when it steps down.");
+assert.commandWorked(node1.adminCommand({replSetStepUp: 1}));
+assert.eq(node1, rst.getPrimary());
+
+const node1DB = node1.getDB(dbName);
+
+jsTestLog("Run a transaction with txnNumber 0 and the same session ID on the new primary.");
+assert.commandWorked(node1DB.runCommand({
+    insert: collName,
+    documents: [{c: 1}],
+    lsid: sessionId,
+    txnNumber: NumberLong(0),
+    startTransaction: true,
+    autocommit: false
+}));
+let res = assert.commandWorked(node1DB.adminCommand({
+    prepareTransaction: 1,
+    lsid: sessionId,
+    txnNumber: NumberLong(0),
+    autocommit: false,
+    writeConcern: {w: "majority"}
+}));
+assert.commandWorked(node1DB.adminCommand({
+    commitTransaction: 1,
+    commitTimestamp: res.prepareTimestamp,
+    lsid: sessionId,
+    txnNumber: NumberLong(0),
+    autocommit: false,
+    writeConcern: {w: "majority"}
+}));
+assert.eq(2, node0SessionDB.coll.find().itcount());
+assert.eq(0, node0SessionDB.coll.find({b: 1}).itcount());
+assert.eq(1, node0SessionDB.coll.find({c: 1}).itcount());
+
+rst.stopSet();
+})();