// Validates the remove/drain shard functionality when there is data on the shard being removed
(function() {
    'use strict';

    var st = new ShardingTest({name: "remove_shard3", shards: 2, mongos: 2});

    assert.commandWorked(st.s0.adminCommand({enableSharding: 'TestDB'}));
    st.ensurePrimaryShard('TestDB', 'shard0000');
    assert.commandWorked(st.s0.adminCommand({shardCollection: 'TestDB.Coll', key: {_id: 1}}));
    assert.commandWorked(st.s0.adminCommand({split: 'TestDB.Coll', middle: {_id: 0}}));

    // Insert some documents and make sure there are docs on both shards
    st.s0.getDB('TestDB').Coll.insert({_id: -1, value: 'Negative value'});
    st.s0.getDB('TestDB').Coll.insert({_id: 1, value: 'Positive value'});

    assert.commandWorked(st.s0.adminCommand(
        {moveChunk: 'TestDB.Coll', find: {_id: 1}, to: 'shard0001', _waitForDelete: true}));

    // Make sure both mongos instances know of the latest metadata
    assert.eq(2, st.s0.getDB('TestDB').Coll.find({}).toArray().length);
    assert.eq(2, st.s1.getDB('TestDB').Coll.find({}).toArray().length);

    // Remove shard0001
    var removeRes;
    removeRes = assert.commandWorked(st.s0.adminCommand({removeShard: 'shard0001'}));
    assert.eq('started', removeRes.state);
    removeRes = assert.commandWorked(st.s0.adminCommand({removeShard: 'shard0001'}));
    assert.eq('ongoing', removeRes.state);

    // Move the one chunk off shard0001
    assert.commandWorked(st.s0.adminCommand(
        {moveChunk: 'TestDB.Coll', find: {_id: 1}, to: 'shard0000', _waitForDelete: true}));

    // Remove shard must succeed now
    removeRes = assert.commandWorked(st.s0.adminCommand({removeShard: 'shard0001'}));
    assert.eq('completed', removeRes.state);

    // Make sure both mongos instance refresh their metadata and do not reference the missing shard
    assert.eq(2, st.s0.getDB('TestDB').Coll.find({}).toArray().length);
    assert.eq(2, st.s1.getDB('TestDB').Coll.find({}).toArray().length);

    st.stop();

})();