//
// Basic tests for moveChunk.
//

(function() {
'use strict';

var st = new ShardingTest({mongos:1, shards:2});

var mongos = st.s0;

var kDbName = 'db';

var shards = mongos.getCollection('config.shards').find().toArray();

var shard0 = shards[0]._id;
var shard1 = shards[1]._id;

function testHashed() {
    var ns =  kDbName + '.fooHashed'; 

    // Errors if either bounds is not a valid shard key
    assert.commandWorked(mongos.adminCommand({shardCollection: ns, key: {_id: 'hashed'}}));

    var aChunk = mongos.getDB('config').chunks.findOne({_id: RegExp(ns), shard: shard0});
    assert(aChunk); 
    assert.commandFailed(mongos.adminCommand({moveChunk: ns, bounds: [aChunk.min, aChunk.max-1], 
                                             to: shard1}));

    // Fail if find and bounds are both set.
    assert.commandFailed(mongos.adminCommand({moveChunk: ns, find: {_id: 1}, 
                                             bounds: [aChunk.min, aChunk.max], to: shard1}));

    // Using find on collections with hash shard keys should not crash
    assert.commandFailed(mongos.adminCommand({moveChunk: ns, find: {_id: 1}, to: shard1}));

    // Fail if chunk is already at shard
    assert.commandFailed(mongos.adminCommand({moveChunk: ns, bounds: [aChunk.min, aChunk.max], 
                                             to: shard0}));

    assert.commandWorked(mongos.adminCommand({moveChunk: ns, bounds: [aChunk.min, aChunk.max], 
                                             to: shard1}));
    assert.eq(0, mongos.getDB('config').chunks.count({_id: aChunk._id, shard: shard0}));
    assert.eq(1, mongos.getDB('config').chunks.count({_id: aChunk._id, shard: shard1}));

    mongos.getDB(kDbName).fooHashed.drop();
}

function testNotHashed(keyDoc) {
    var ns = kDbName + '.foo';

    // Fail if find is not a valid shard key.
    assert.commandWorked(mongos.adminCommand({shardCollection: ns, key: keyDoc}));

    var chunkId = mongos.getDB('config').chunks.findOne({_id: RegExp(ns), shard: shard0})._id;

    assert.commandFailed(mongos.adminCommand({moveChunk: ns, find: {xxx: 1}, to: shard1}));
    assert.eq(shard0, mongos.getDB('config').chunks.findOne({_id: chunkId}).shard);

    assert.commandWorked(mongos.adminCommand({moveChunk: ns, find: keyDoc, to: shard1}));
    assert.eq(shard1, mongos.getDB('config').chunks.findOne({_id: chunkId}).shard);

    // Fail if to shard does not exists
    assert.commandFailed(mongos.adminCommand({moveChunk: ns, find: keyDoc, to: 'WrongShard'}));

    // Fail if chunk is already at shard
    assert.commandFailed(mongos.adminCommand({moveChunk: ns, find: keyDoc, to: shard1}));
    assert.eq(shard1, mongos.getDB('config').chunks.findOne({_id: chunkId}).shard);

    mongos.getDB(kDbName).foo.drop();
}

assert.commandWorked(mongos.adminCommand({enableSharding : kDbName}));

st.ensurePrimaryShard(kDbName, shard0);

// Fail if invalid namespace.
var res = assert.commandFailed(mongos.adminCommand({moveChunk: '', find: {_id: 1}, to: shard1}));
assert.eq( res.info); 

// Fail if database does not exist.
assert.commandFailed(mongos.adminCommand({moveChunk: 'a.b', find: {_id: 1}, to: shard1}));

// Fail if collection is unsharded. 
assert.commandFailed(mongos.adminCommand({moveChunk: kDbName + '.xxx', 
                                         find: {_id: 1}, to: shard1}));

testHashed();

testNotHashed({a:1});

testNotHashed({a:1, b:1});

st.stop();

})();