summaryrefslogtreecommitdiff
path: root/jstests/replsets/slaveok_read_pref.js
blob: 78b0139adfdf963b79777faac41acfe6f822d785 (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
// Test that slaveOk is implicitly allowed for queries on a secondary with a read preference other
// than 'primary', and that queries which do have 'primary' read preference fail.
(function() {
"use strict";

const readPrefs =
    [undefined, "primary", "secondary", "primaryPreferred", "secondaryPreferred", "nearest"];

const rst = new ReplSetTest({nodes: 3});
rst.startSet();

const nodes = rst.nodeList();
rst.initiate({
    _id: jsTestName(),
    members: [
        {_id: 0, host: nodes[0]},
        {_id: 1, host: nodes[1], priority: 0},
        {_id: 2, host: nodes[2], arbiterOnly: true}
    ]
});

const priDB = rst.getPrimary().getDB(jsTestName());
assert(priDB.dropDatabase());

assert.commandWorked(priDB.test.insert({a: 1}, {writeConcern: {w: "majority"}}));

const secDB = rst.getSecondary().getDB(jsTestName());

for (let readMode of ["commands", "legacy"]) {
    for (let readPref of readPrefs) {
        for (let slaveOk of [true, false]) {
            const testType = {readMode: readMode, readPref: readPref, slaveOk: slaveOk};

            secDB.getMongo().forceReadMode(readMode);
            secDB.getMongo().setSlaveOk(slaveOk);

            const cursor = (readPref ? secDB.test.find().readPref(readPref) : secDB.test.find());

            if (readPref === "primary" || (!readPref && !slaveOk)) {
                // Attempting to run the query throws an error of type NotMasterNoSlaveOk.
                const slaveOkErr = assert.throws(() => cursor.itcount(), [], tojson(testType));
                assert.commandFailedWithCode(slaveOkErr, ErrorCodes.NotMasterNoSlaveOk);
            } else {
                // Succeeds for all non-primary readPrefs, and for no readPref iff slaveOk.
                const docCount = assert.doesNotThrow(() => cursor.itcount(), [], tojson(testType));
                assert.eq(docCount, 1);
            }
        }
    }
}

rst.stopSet();
})();