SERVER-36306 Add FSM workloads that run transactions workloads with all threads using same session

1 files changed, 9 insertions, 0 deletions

diff --git a/src/mongo/shell/assert.js b/src/mongo/shell/assert.js
index 4e6b762475a..de7bb1219a6 100644
--- a/src/mongo/shell/assert.js
+++ b/src/mongo/shell/assert.js
@@ -617,6 +617,15 @@ assert = (function() {
         return res;
     }
 
+    assert.commandWorkedOrFailedWithCode = function commandWorkedOrFailedWithCode(
+        res, errorCodeSet, msg) {
+        if (!res.ok) {
+            return assert.commandFailedWithCode(res, errorCodeSet, msg);
+        } else {
+            return assert.commandWorked(res, msg);
+        }
+    };
+
     assert.commandWorked = function(res, msg) {
         return _assertCommandWorked(res, msg, {ignoreWriteErrors: false});
     };