SERVER-37241 Add testing to verify proper expiration of sessions in the sessions collection

(cherry picked from commit 3caa3c4a4be7b84823f22f481365f58b124d6d00)

1 files changed, 79 insertions, 0 deletions

diff --git a/jstests/libs/pin_getmore_cursor.js b/jstests/libs/pin_getmore_cursor.js
new file mode 100644
index 00000000000..f0bf195f9b6
--- /dev/null
+++ b/jstests/libs/pin_getmore_cursor.js
@@ -0,0 +1,79 @@
+/**
+ * Pins a cursor in a seperate shell and then runs the given function.
+ * 'conn': a connection to an instance of a mongod or mongos.
+ * 'sessionId': The id present if the database is currently in a session.
+ * 'dbName': the database to use with the cursor.
+ * 'assertFunction': a function containing the test to be run after a cursor is pinned and hanging.
+ * 'runGetMoreFunc': A function to generate a string that will be executed in the parallel shell.
+ * 'failPointName': The string name of the failpoint where the cursor will hang. The function turns
+ * the failpoint on, the assert function should turn it off whenever it is appropriate for the test.
+ * 'assertEndCounts': The boolean indicating whether we want to assert the number of open or
+ * pinned cursors.
+ */
+
+function withPinnedCursor(
+    {conn, sessionId, db, assertFunction, runGetMoreFunc, failPointName, assertEndCounts}) {
+    // This test runs manual getMores using different connections, which will not inherit the
+    // implicit session of the cursor establishing command.
+    TestData.disableImplicitSessions = true;
+
+    const coll = db.jstest_with_pinned_cursor;
+    coll.drop();
+    db.active_cursor_sentinel.drop();
+    for (let i = 0; i < 100; ++i) {
+        assert.writeOK(coll.insert({value: i}));
+    }
+    let cleanup = null;
+    try {
+        // Enable the specified failpoint.
+        assert.commandWorked(
+            db.adminCommand({configureFailPoint: failPointName, mode: "alwaysOn"}));
+
+        // Issue an initial find in order to create a cursor and obtain its cursorID.
+        let cmdRes = db.runCommand({find: coll.getName(), batchSize: 2});
+        assert.commandWorked(cmdRes);
+        const cursorId = cmdRes.cursor.id;
+        assert.neq(cursorId, NumberLong(0));
+
+        // Let the cursor hang in a different shell with the information it needs to do a getMore.
+        let code = "let cursorId = " + cursorId.toString() + ";";
+        code += "let collName = '" + coll.getName() + "';";
+        if (sessionId) {
+            code += "let sessionId = " + tojson(sessionId) + ";";
+        }
+        code += "(" + runGetMoreFunc.toString() + ")();";
+        code += "db.active_cursor_sentinel.insert({});";
+        cleanup = startParallelShell(code, conn.port);
+
+        // Wait until we know the failpoint has been reached.
+        assert.soon(function() {
+            const arr = db.getSiblingDB("admin")
+                            .aggregate([
+                                {$currentOp: {localOps: true, allUsers: true}},
+                                {$match: {"msg": failPointName}}
+                            ])
+                            .toArray();
+            return arr.length > 0;
+        });
+        assertFunction(cursorId, coll);
+        // Eventually the cursor should be cleaned up.
+        assert.commandWorked(db.adminCommand({configureFailPoint: failPointName, mode: "off"}));
+
+        if (assertEndCounts) {
+            assert.soon(() => db.active_cursor_sentinel.find().itcount() > 0);
+            assert.eq(db.serverStatus().metrics.cursor.open.pinned, 0);
+        }
+
+        // Trying to kill the cursor again should result in the cursor not being found.
+        cmdRes = db.runCommand({killCursors: coll.getName(), cursors: [cursorId]});
+        assert.commandWorked(cmdRes);
+        assert.eq(cmdRes.cursorsKilled, []);
+        assert.eq(cmdRes.cursorsAlive, []);
+        assert.eq(cmdRes.cursorsNotFound, [cursorId]);
+        assert.eq(cmdRes.cursorsUnknown, []);
+    } finally {
+        if (cleanup) {
+            cleanup();
+        }
+    }
+}