summaryrefslogtreecommitdiff
path: root/jstests/sharding/drop_sharded_db.js
blob: 9a0a56ca24533f164b45b698ab43dba633d3a5e1 (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
66
67
// 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.writeOK(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.
    dbA.dropDatabase();
    var dbs = mongos.getDBNames();
    for (var i = 0; i < dbs.length; i++) {
        assert.neq(dbs, "" + 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();

})();