/* Tests count and distinct using slaveOk. Also tests a scenario querying a set
 * where only one secondary is up.
 */

var st = new ShardingTest( testName = "countSlaveOk",
                           numShards = 1,
                           verboseLevel = 0,
                           numMongos = 1,
                           { rs : true, 
                             rs0 : { nodes : 2 }
                           })

var rst = st._rs[0].test

// Insert data into replica set
var conn = new Mongo( st.s.host )
conn.setLogLevel( 3 )

var coll = conn.getCollection( "test.countSlaveOk" )
coll.drop()

for( var i = 0; i < 300; i++ ){
    coll.insert( { i : i % 10 } )
}

var connA = conn
var connB = new Mongo( st.s.host )
var connC = new Mongo( st.s.host )

// Make sure the writes get through, otherwise we can continue to error these one-at-a-time
coll.getDB().getLastError()

st.printShardingStatus()

// Wait for client to update itself and replication to finish
rst.awaitReplication()

var primary = rst.getPrimary()
var sec = rst.getSecondary()

// Data now inserted... stop the master, since only two in set, other will still be secondary
rst.stop( rst.getMaster(), undefined, true )
printjson( rst.status() )

// Wait for the mongos to recognize the slave
ReplSetTest.awaitRSClientHosts( conn, sec, { ok : true, secondary : true } )

// Make sure that mongos realizes that primary is already down
ReplSetTest.awaitRSClientHosts( conn, primary, { ok : false });

// Need to check slaveOk=true first, since slaveOk=false will destroy conn in pool when
// master is down
conn.setSlaveOk()

// count using the command path
assert.eq( 30, coll.find({ i : 0 }).count() )
// count using the query path
assert.eq( 30, coll.find({ i : 0 }).itcount() );
assert.eq( 10, coll.distinct("i").length )

try {
    conn.setSlaveOk( false )
    // Should throw exception, since not slaveOk'd
    coll.find({ i : 0 }).count()
    
    print( "Should not reach here!" )
    printjson( coll.getDB().getLastError() )                 
    assert( false )
    
}
catch( e ){
    print( "Non-slaveOk'd connection failed." )
}

// Finish
st.stop()