SERVER-1463 test failover with arbiter

2 files changed, 53 insertions, 0 deletions

diff --git a/jstests/replsets/replsetarb2.js b/jstests/replsets/replsetarb2.js
new file mode 100644
index 00000000000..0dd8a3ded31
--- /dev/null
+++ b/jstests/replsets/replsetarb2.js
@@ -0,0 +1,45 @@
+// Election when master fails and remaining nodes are an arbiter and a slave.
+// Note that in this scenario, the arbiter needs two votes.
+
+doTest = function( signal ) {
+
+    var replTest = new ReplSetTest( {name: 'unicomplex', nodes: 3} );
+    var nodes = replTest.nodeList();
+
+    print(tojson(nodes));
+
+    var conns = replTest.startSet();
+    var r = replTest.initiate({"_id" : "unicomplex", 
+                "members" : [
+                             {"_id" : 0, "host" : nodes[0] },
+                             {"_id" : 1, "host" : nodes[1], "arbiterOnly" : true, "votes": 2},
+                             {"_id" : 2, "host" : nodes[2] }]});
+
+    // Make sure we have a master
+    var master = replTest.getMaster();
+
+    // Make sure we have an arbiter
+    assert.soon(function() {
+        res = conns[1].getDB("admin").runCommand({replSetGetStatus: 1});
+        printjson(res);
+        return res.myState == 7;
+    }, "Aribiter failed to initialize.");
+
+    // Wait for initial replication
+    master.getDB("foo").foo.insert({a: "foo"});
+    replTest.awaitReplication();
+
+    // Now kill the original master
+    mId = replTest.getNodeId( master );
+    replTest.stop( mId );
+
+    // And make sure that the slave is promoted
+    new_master = replTest.getMaster();
+
+    newMasterId = replTest.getNodeId( new_master );
+    assert( newMasterId == 2, "Slave wasn't promoted to new master");
+
+    replTest.stopSet( signal );
+}
+
+doTest( 15 );
diff --git a/shell/servers.js b/shell/servers.js
index d1ace93a447..16b8de8c0f3 100644
--- a/shell/servers.js
+++ b/shell/servers.js
@@ -1146,6 +1146,14 @@ ReplSetTest.prototype.awaitReplication = function() {
            var synced = true;
            for(var i=0; i<this.liveNodes.slaves.length; i++) {
              var slave = this.liveNodes.slaves[i];
+
+             // Continue if we're connected to an arbiter
+             if(res = slave.getDB("admin").runCommand({replSetGetStatus: 1})) {
+                 if(res.myState == 7) {
+                     continue;
+                 }
+             }
+
              slave.getDB("admin").getMongo().setSlaveOk();
              var log = slave.getDB("local")['replset.minvalid'];
              if(log.find().hasNext()) {