SERVER-21406 extend timeout in rollback2.js when running in pv0

1 files changed, 14 insertions, 2 deletions

diff --git a/jstests/replsets/rollback2.js b/jstests/replsets/rollback2.js
index 0dc47e93eb6..ef71bfba252 100644
--- a/jstests/replsets/rollback2.js
+++ b/jstests/replsets/rollback2.js
@@ -69,11 +69,23 @@ load("jstests/replsets/rslib.js");
     a.createCollection("kap2", { capped: true, size: 5501 });
     replTest.awaitReplication();
 
+    var timeout;
+    if (replTest.getConfigFromPrimary().protocolVersion == 1) {
+        timeout = 30 * 1000;
+    } else {
+        timeout = 60 * 1000;
+    }
     // isolate A and wait for B to become master
     conns[0].disconnect(conns[1]);
     conns[0].disconnect(conns[2]);
-    assert.soon(function () { try { return B.isMaster().ismaster; } catch(e) { return false; } });
-    
+    assert.soon(function () {
+        try {
+            return B.isMaster().ismaster;
+        } catch(e) {
+            return false;
+        }
+    }, timeout);
+
     // do operations on B and B alone, these will be rolled back
     assert.writeOK(b.bar.insert({ q: 4 }));
     assert.writeOK(b.bar.update({ q: 3 }, { q: 3, rb: true }));