97 lines
3.3 KiB
JavaScript
97 lines
3.3 KiB
JavaScript
|
//
|
||
|
|
// 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();
|
||
|
|
|
||
|
|
})()
|