// test arbitration

var baseName = "jstests_pair3test";

ismaster = function( n ) {
    var im = n.getDB( "admin" ).runCommand( { "ismaster" : 1 } );
    print( "ismaster: " + tojson( im ) );
    assert( im, "command ismaster failed" );
    return im.ismaster;
}

connect = function() {
    startMongoProgram( "mongobridge", "--port", alPort, "--dest", "localhost:" + aPort );
    startMongoProgram( "mongobridge", "--port", arPort, "--dest", "localhost:" + aPort );
    startMongoProgram( "mongobridge", "--port", lpPort, "--dest", "localhost:" + lPort );
    startMongoProgram( "mongobridge", "--port", rpPort, "--dest", "localhost:" + rPort );
}

doTest1 = function() {
    ports = allocatePorts( 7 );
    aPort = ports[ 0 ];
    alPort = ports[ 1 ];
    arPort = ports[ 2 ];
    lPort = ports[ 3 ];
    lpPort = ports[ 4 ];
    rPort = ports[ 5 ];
    rpPort = ports[ 6 ];
    
    connect();
    
    a = startMongod( "--port", aPort, "--dbpath", "/data/db/" + baseName + "-arbiter", "--nohttpinterface" );
    l = startMongod( "--port", lPort, "--dbpath", "/data/db/" + baseName + "-left", "--pairwith", "127.0.0.1:" + rpPort, "127.0.0.1:" + alPort, "--oplogSize", "1", "--nohttpinterface" );
    r = startMongod( "--port", rPort, "--dbpath", "/data/db/" + baseName + "-right", "--pairwith", "127.0.0.1:" + lpPort, "127.0.0.1:" + arPort, "--oplogSize", "1", "--nohttpinterface" );
    
    // normal startup
    
    assert.soon( function() {
                lm = ismaster( l );
                rm = ismaster( r );
                
                assert( lm == -1 || lm == 0, "lm value invalid" );
                assert( rm == -1 || rm == 0 || rm == 1, "rm value invalid" );
                
                return ( lm == 0 && rm == 1 );
                } );

    // disconnect l (slave)
    
    stopMongoProgram( alPort );
    stopMongoProgram( lpPort );
    stopMongoProgram( rpPort );
    
    assert.soon( function() {
                lm = ismaster( l );
                rm = ismaster( r );
                
                assert( lm == 0 || lm == -3, "lm value invalid" );
                assert( rm == 1, "rm value invalid" );
                
                return ( lm == -3 );
                } );
    
    // disconnect r ( master )
    
    stopMongoProgram( arPort );
    
    assert.soon( function() {
                rm = ismaster( r );
                assert( rm == 1 || rm == -3, "rm value invalid" );
                return ( rm == -3 );
                } );
    
    // reconnect
    
    startMongoProgram( "mongobridge", "--port", lpPort, "--dest", "localhost:" + lPort );
    startMongoProgram( "mongobridge", "--port", rpPort, "--dest", "localhost:" + rPort );
    // no arbiter yet, to ensure r becomes master.
    assert.soon( function() {
                lm = ismaster( l );
                rm = ismaster( r );
                
                assert( lm == -3 || lm == 0, "lm value invalid" );
                assert( rm == -3 || rm == 1, "rm value invalid" );
                
                return ( lm == 0 && rm == 1 );
                } );
    startMongoProgram( "mongobridge", "--port", alPort, "--dest", "localhost:" + aPort );
    startMongoProgram( "mongobridge", "--port", arPort, "--dest", "localhost:" + aPort );

    // disconnect r ( master )
    stopMongoProgram( arPort );
    stopMongoProgram( lpPort );
    stopMongoProgram( rpPort );
    
    assert.soon( function() {
                lm = ismaster( l );
                rm = ismaster( r );
                
                assert( lm == 0 || lm == 1, "lm value invalid" );
                assert( rm == 1 || rm == -3, "rm value invalid" );
                
                return ( rm == -3 && lm == 1 );
                } );
    
    // disconnect l ( new master )
    stopMongoProgram( alPort );
    
    assert.soon( function() {
                lm = ismaster( l );
                assert( lm == 1 || lm == -3, "lm value invalid" );
                return ( lm == -3 );
                } );

    // reconnect
    
    startMongoProgram( "mongobridge", "--port", lpPort, "--dest", "localhost:" + lPort );
    startMongoProgram( "mongobridge", "--port", rpPort, "--dest", "localhost:" + rPort );
    // no arbiter yet, to ensure r becomes master.
    assert.soon( function() {
                lm = ismaster( l );
                rm = ismaster( r );
                
                assert( lm == -3 || lm == 0, "lm value invalid" );
                assert( rm == -3 || rm == 1, "rm value invalid" );
                
                return ( lm == 0 && rm == 1 );
                } );
    startMongoProgram( "mongobridge", "--port", alPort, "--dest", "localhost:" + aPort );
    startMongoProgram( "mongobridge", "--port", arPort, "--dest", "localhost:" + aPort );

    // disconnect l ( slave )

    stopMongoProgram( alPort );
    stopMongoProgram( lpPort );
    stopMongoProgram( rpPort );
    
    assert.soon( function() {
                lm = ismaster( l );
                rm = ismaster( r );
                
                assert( lm == 0 || lm == -3, "lm value invalid" );
                assert( rm == 1, "rm value invalid" );
                
                return ( lm == -3 );
                } );

    // reconnect l
    
    startMongoProgram( "mongobridge", "--port", alPort, "--dest", "localhost:" + aPort );
    startMongoProgram( "mongobridge", "--port", lpPort, "--dest", "localhost:" + lPort );
    startMongoProgram( "mongobridge", "--port", rpPort, "--dest", "localhost:" + rPort );
    
    assert.soon( function() {
                lm = ismaster( l );
                rm = ismaster( r );
                
                assert( lm == 0 || lm == -3, "lm value invalid" );
                assert( rm == 1, "rm value invalid" );
                
                return ( lm == 0 );
                } );
    
    // disconnect r ( master )
    stopMongoProgram( arPort );
    stopMongoProgram( lpPort );
    stopMongoProgram( rpPort );
    
    assert.soon( function() {
                lm = ismaster( l );
                rm = ismaster( r );
                
                assert( lm == 0 || lm == 1, "lm value invalid" );
                assert( rm == 1 || rm == -3, "rm value invalid" );
                
                return ( rm == -3 && lm == 1 );
                } );
    
    // reconnect r, without arbiter to prevent r becoming master.
    startMongoProgram( "mongobridge", "--port", lpPort, "--dest", "localhost:" + lPort );
    startMongoProgram( "mongobridge", "--port", rpPort, "--dest", "localhost:" + rPort );
    
    assert.soon( function() {
                lm = ismaster( l );
                rm = ismaster( r );
                
                assert( rm == 0 || rm == -3, "rm value invalid" );
                assert( lm == 1, "lm value invalid" );
                
                return ( rm == 0 );
                } );

    ports.forEach( function( x ) { stopMongoProgram( x ); } );
}

// this time don't start connected
doTest2 = function() {
    ports = allocatePorts( 7 );
    aPort = ports[ 0 ];
    alPort = ports[ 1 ];
    arPort = ports[ 2 ];
    lPort = ports[ 3 ];
    lpPort = ports[ 4 ];
    rPort = ports[ 5 ];
    rpPort = ports[ 6 ];
    
    a = startMongod( "--port", aPort, "--dbpath", "/data/db/" + baseName + "-arbiter", "--nohttpinterface" );
    l = startMongod( "--port", lPort, "--dbpath", "/data/db/" + baseName + "-left", "--pairwith", "127.0.0.1:" + rpPort, "127.0.0.1:" + alPort, "--oplogSize", "1", "--nohttpinterface" );
    r = startMongod( "--port", rPort, "--dbpath", "/data/db/" + baseName + "-right", "--pairwith", "127.0.0.1:" + lpPort, "127.0.0.1:" + arPort, "--oplogSize", "1", "--nohttpinterface" );

    assert.soon( function() {
                lm = ismaster( l );
                rm = ismaster( r );
                
                assert( lm == -1 || lm == -3, "lm value invalid" );
                assert( rm == -1 || rm == -3, "rm value invalid" );
                
                return ( lm == -3 && rm == -3 );
                } );
    
    startMongoProgram( "mongobridge", "--port", arPort, "--dest", "localhost:" + aPort );

    // there hasn't been an initial sync, no no node will become master
    
    for( i = 0; i < 10; ++i ) {
        assert( ismaster( l ) == -3 && ismaster( r ) == -3 );
        sleep( 500 );
    }

    stopMongoProgram( arPort );

    startMongoProgram( "mongobridge", "--port", alPort, "--dest", "localhost:" + aPort );

    for( i = 0; i < 10; ++i ) {
        assert( ismaster( l ) == -3 && ismaster( r ) == -3 );
        sleep( 500 );
    }    

    stopMongoProgram( alPort );
    
    // connect l and r without a
    
    startMongoProgram( "mongobridge", "--port", lpPort, "--dest", "localhost:" + lPort );
    startMongoProgram( "mongobridge", "--port", rpPort, "--dest", "localhost:" + rPort );
    
    assert.soon( function() {
                lm = ismaster( l );
                rm = ismaster( r );
                
                assert( lm == 0 || lm == -3, "lm value invalid" );
                assert( rm == 1 || rm == -3 || rm == 0, "rm value invalid" );
                
                return ( lm == 0 && rm == 1 );
                } );
    
    ports.forEach( function( x ) { stopMongoProgram( x ); } );    
}

doTest1();
doTest2();