summaryrefslogtreecommitdiff
path: root/jstests/sharding/drop_sharded_db.js
blob: a10c2cba5aa999517bb98c001295ab4064b346f2 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
// Tests the dropping of a sharded database SERVER-3471 SERVER-1726
(function() {
var st = new ShardingTest({shards: 2});

var mongos = st.s0;
var config = mongos.getDB("config");

var dbA = mongos.getDB("DropSharded_A");
var dbB = mongos.getDB("DropSharded_B");
var dbC = mongos.getDB("DropSharded_C");

// Dropping a database that doesn't exist will result in an info field in the response.
var res = assert.commandWorked(dbA.dropDatabase());
assert.eq('database does not exist', res.info);

var numDocs = 3000;
var numColls = 10;
for (var i = 0; i < numDocs; i++) {
    dbA.getCollection("data" + (i % numColls)).insert({_id: i});
    dbB.getCollection("data" + (i % numColls)).insert({_id: i});
    dbC.getCollection("data" + (i % numColls)).insert({_id: i});
}

var key = {_id: 1};
for (var i = 0; i < numColls; i++) {
    st.shardColl(dbA.getCollection("data" + i), key);
    st.shardColl(dbB.getCollection("data" + i), key);
    st.shardColl(dbC.getCollection("data" + i), key);
}

// Insert a document to an unsharded collection and make sure that the document is there.
assert.commandWorked(dbA.unsharded.insert({dummy: 1}));
var shardName = config.databases.findOne({_id: dbA.getName()}).primary;
var shardHostConn = new Mongo(config.shards.findOne({_id: shardName}).host);
var dbAOnShard = shardHostConn.getDB(dbA.getName());
assert.neq(null, dbAOnShard.unsharded.findOne({dummy: 1}));

// Drop the non-suffixed db and ensure that it is the only one that was dropped.
assert.commandWorked(dbA.dropDatabase());
var dbs = mongos.getDBNames();
for (var i = 0; i < dbs.length; i++) {
    assert.neq(dbs[i], "" + dbA);
}

assert.eq(0, config.databases.count({_id: dbA.getName()}));
assert.eq(1, config.databases.count({_id: dbB.getName()}));
assert.eq(1, config.databases.count({_id: dbC.getName()}));

// 10 dropped collections
assert.eq(numColls, config.collections.count({_id: RegExp("^" + dbA + "\\..*"), dropped: true}));

// 20 active (dropped is missing)
assert.eq(numColls, config.collections.count({_id: RegExp("^" + dbB + "\\..*")}));
assert.eq(numColls, config.collections.count({_id: RegExp("^" + dbC + "\\..*")}));

for (var i = 0; i < numColls; i++) {
    assert.eq(numDocs / numColls, dbB.getCollection("data" + (i % numColls)).find().itcount());
    assert.eq(numDocs / numColls, dbC.getCollection("data" + (i % numColls)).find().itcount());
}

// Check that the unsharded collection should have been dropped.
assert.eq(null, dbAOnShard.unsharded.findOne());

st.stop();
})();