From 145f3a4368ac2efe92d58ddb2462229e5c27365c Mon Sep 17 00:00:00 2001
From: jannaerin <golden.janna@gmail.com>
Date: Mon, 6 Aug 2018 11:31:54 -0400
Subject: SERVER-36306 Add FSM workloads that run transactions workloads with
 all threads using same session

---
 src/mongo/shell/assert.js | 9 +++++++++
 1 file changed, 9 insertions(+)

(limited to 'src/mongo/shell/assert.js')

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});
     };
-- 
cgit v1.2.1