SERVER-30080 LogicalSessions in the shell

Basic logical session implementation for the shell

1 files changed, 22 insertions, 0 deletions

diff --git a/jstests/libs/override_methods/enable_sessions.js b/jstests/libs/override_methods/enable_sessions.js
new file mode 100644
index 00000000000..2b1be14268b
--- /dev/null
+++ b/jstests/libs/override_methods/enable_sessions.js
@@ -0,0 +1,22 @@
+/**
+ * Enables sessions on the db object
+ */
+(function() {
+    "use strict";
+
+    db = db.getMongo().startSession().getDatabase(db.getName());
+
+    var originalStartParallelShell = startParallelShell;
+    startParallelShell = function(jsCode, port, noConnect) {
+        var newCode;
+        var overridesFile = "jstests/libs/override_methods/enable_sessions.js";
+        if (typeof(jsCode) === "function") {
+            // Load the override file and immediately invoke the supplied function.
+            newCode = `load("${overridesFile}"); (${jsCode})();`;
+        } else {
+            newCode = `load("${overridesFile}"); ${jsCode};`;
+        }
+
+        return originalStartParallelShell(newCode, port, noConnect);
+    };
+})();