jstests/sharding/hash_shard_num_chunks.js


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

// Hash sharding with initial chunk count set.

(function() {

    var s = new ShardingTest({shards: 3, mongos: 1});
    s.stopBalancer();

    var dbname = "test";
    var coll = "foo";
    var db = s.getDB(dbname);

    assert.commandWorked(db.adminCommand({enablesharding: dbname}));
    s.ensurePrimaryShard(dbname, 'shard0001');

    assert.commandWorked(db.adminCommand(
        {shardcollection: dbname + "." + coll, key: {a: "hashed"}, numInitialChunks: 500}));

    s.printShardingStatus();

    var numChunks = s.config.chunks.count();
    assert.eq(numChunks, 500, "should be exactly 500 chunks");

    var shards = s.config.shards.find();
    shards.forEach(
        // check that each shard has one third the numInitialChunks
        function(shard) {
            var numChunksOnShard = s.config.chunks.find({"shard": shard._id}).count();
            assert.gte(numChunksOnShard, Math.floor(500 / 3));
        });

    // Check that the collection gets dropped correctly (which doesn't happen if pre-splitting fails
    // to
    // create the collection on all shards).
    res = db.runCommand({"drop": coll});
    assert.eq(res.ok, 1, "couldn't drop empty, pre-split collection");

    s.stop();

})();