summaryrefslogtreecommitdiff
path: root/jstests/sharding/balance_repl.js
blob: fdc0d15509c0c080c24881eea1a0386687cc1514 (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
68
69
/**
 * Testing migrations are successful and immediately visible on the secondaries, when
 * secondaryThrottle is used.
 */
(function() {
'use strict';

// The mongod secondaries are set to priority 0 and votes 0 to prevent the primaries
// from stepping down during migrations on slow evergreen builders.
var s = new ShardingTest({
    shards: 2,
    other: {
        chunkSize: 1,
        rs0: {
            nodes: [
                {rsConfig: {votes: 1}},
                {rsConfig: {priority: 0, votes: 0}},
            ],
        },
        rs1: {
            nodes: [
                {rsConfig: {votes: 1}},
                {rsConfig: {priority: 0, votes: 0}},
            ],
        }
    }
});

var bulk = s.s0.getDB('TestDB').TestColl.initializeUnorderedBulkOp();
for (var i = 0; i < 2100; i++) {
    bulk.insert({_id: i, x: i});
}
assert.writeOK(bulk.execute());

assert.commandWorked(s.s0.adminCommand({enablesharding: 'TestDB'}));
s.ensurePrimaryShard('TestDB', s.shard0.shardName);
assert.commandWorked(s.s0.adminCommand({shardcollection: 'TestDB.TestColl', key: {_id: 1}}));

for (i = 0; i < 20; i++) {
    assert.commandWorked(s.s0.adminCommand({split: 'TestDB.TestColl', middle: {_id: i * 100}}));
}

var collPrimary = (new Mongo(s.s0.host)).getDB('TestDB').TestColl;
assert.eq(2100, collPrimary.find().itcount());

var collSlaveOk = (new Mongo(s.s0.host)).getDB('TestDB').TestColl;
collSlaveOk.setSlaveOk();
assert.eq(2100, collSlaveOk.find().itcount());

for (i = 0; i < 20; i++) {
    // Needs to waitForDelete because we'll be performing a slaveOk query, and secondaries don't
    // have a chunk manager so it doesn't know how to filter out docs it doesn't own.
    assert.commandWorked(s.s0.adminCommand({
        moveChunk: 'TestDB.TestColl',
        find: {_id: i * 100},
        to: s.shard1.shardName,
        _secondaryThrottle: true,
        writeConcern: {w: 2},
        _waitForDelete: true
    }));

    assert.eq(2100,
              collSlaveOk.find().itcount(),
              'Incorrect count when reading from secondary. Count from primary is ' +
                  collPrimary.find().itcount());
}

s.stop();
}());