summaryrefslogtreecommitdiff
path: root/jstests/noPassthrough/stepdown_query.js
blob: 2db51e7155836ac04f31938bc3a2db23740a53f7 (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
70
71
72
73
74
75
76
77
78
79
80
81
/**
 * Tests that a query with default read preference ("primary") will succeed even if the node being
 * queried steps down before the final result batch has been delivered.
 * @tags: [requires_replication, requires_sharding]
 */

// Checking UUID consistency involves talking to a shard node, which in this test is shutdown
TestData.skipCheckingUUIDsConsistentAcrossCluster = true;

(function() {
    'use strict';

    // Set the refresh period to 10 min to rule out races
    _setShellFailPoint({
        configureFailPoint: "modifyReplicaSetMonitorDefaultRefreshPeriod",
        mode: "alwaysOn",
        data: {
            period: 10 * 60,
        },
    });

    var dbName = "test";
    var collName = jsTest.name();

    function runTest(host, rst, waitForPrimary) {
        // We create a new connection to 'host' here instead of passing in the original connection.
        // This to work around the fact that connections created by ReplSetTest already have slaveOk
        // set on them, but we need a connection with slaveOk not set for this test.
        var conn = new Mongo(host);
        var coll = conn.getDB(dbName).getCollection(collName);
        assert(!coll.exists());
        assert.writeOK(coll.insert([{}, {}, {}, {}, {}]));
        var cursor = coll.find().batchSize(2);
        // Retrieve the first batch of results.
        cursor.next();
        cursor.next();
        assert.eq(0, cursor.objsLeftInBatch());
        var primary = rst.getPrimary();
        var secondary = rst.getSecondary();
        assert.throws(function() {
            primary.getDB("admin").runCommand({replSetStepDown: 60, force: true});
        });
        rst.waitForState(primary, ReplSetTest.State.SECONDARY);
        if (waitForPrimary) {
            rst.waitForState(secondary, ReplSetTest.State.PRIMARY);
        }
        // When the primary steps down, it closes all client connections. Since 'conn' may be a
        // direct connection to the primary and the shell doesn't automatically retry operations on
        // network errors, we run a dummy operation here to force the shell to reconnect.
        try {
            conn.getDB("admin").runCommand("ping");
        } catch (e) {
        }

        // Even though our connection doesn't have slaveOk set, we should still be able to iterate
        // our cursor and kill our cursor.
        assert(cursor.hasNext());
        assert.doesNotThrow(function() {
            cursor.close();
        });
    }

    // Test querying a replica set primary directly.
    var rst = new ReplSetTest({nodes: 1});
    rst.startSet();
    rst.initiate();
    runTest(rst.getPrimary().host, rst, false);
    rst.stopSet();

    rst = new ReplSetTest({nodes: 2});
    rst.startSet();
    rst.initiate();
    runTest(rst.getURL(), rst, true);
    rst.stopSet();

    // Test querying a replica set primary through mongos.
    var st = new ShardingTest({shards: 1, rs: {nodes: 2}, config: 2});
    rst = st.rs0;
    runTest(st.s0.host, rst, true);
    st.stop();
})();