SERVER-28053 allow network errors when turning off rollbackHangBeforeStart failpoint

1 files changed, 16 insertions, 0 deletions

diff --git a/src/mongo/shell/assert.js b/src/mongo/shell/assert.js
index b26380fdb18..f3d984628b5 100644
--- a/src/mongo/shell/assert.js
+++ b/src/mongo/shell/assert.js
@@ -267,6 +267,22 @@ assert.retryNoExcept = function(func, msg, num_attempts, intervalMS) {
     assert.retry(safeFunc, msg, num_attempts, intervalMS);
 };
 
+/**
+ * Runs the given command on the 'admin' database of the provided node. Asserts that the command
+ * worked but allows network errors to occur.
+ */
+assert.adminCommandWorkedAllowingNetworkError = function(node, commandObj) {
+    try {
+        assert.commandWorked(node.adminCommand(commandObj));
+    } catch (e) {
+        // Ignore errors due to connection failures.
+        if (!isNetworkError(e)) {
+            throw e;
+        }
+        print("Caught network error: " + tojson(e));
+    }
+};
+
 assert.time = function(f, msg, timeout /*ms*/) {
     if (assert._debug && msg)
         print("in assert for: " + msg);