summaryrefslogtreecommitdiff
path: root/jstests/noPassthrough/snapshot_cursor_shutdown_stepdown.js
blob: f73eb952043c75a6df560aef9610a6faab89396f (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
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
// Tests that stashed transaction resources are destroyed at shutdown and stepdown.
// @tags: [uses_transactions]
(function() {
    "use strict";

    const dbName = "test";
    const collName = "coll";

    //
    // Test that stashed transaction resources are destroyed at shutdown.
    //

    let rst = new ReplSetTest({nodes: 1});
    rst.startSet();
    rst.initiate();

    let primaryDB = rst.getPrimary().getDB(dbName);

    let session = primaryDB.getMongo().startSession();
    let sessionDB = session.getDatabase(dbName);

    for (let i = 0; i < 4; i++) {
        assert.commandWorked(sessionDB.coll.insert({_id: i}, {writeConcern: {w: "majority"}}));
    }

    // Create a snapshot read cursor.
    assert.commandWorked(sessionDB.runCommand({
        find: collName,
        batchSize: 2,
        readConcern: {level: "snapshot"},
        startTransaction: true,
        autocommit: false,
        txnNumber: NumberLong(0)
    }));

    // It should be possible to shut down the server without hanging. We must skip collection
    // validation, since this will hang.
    const signal = true;  // Use default kill signal.
    const forRestart = false;
    rst.stopSet(signal, forRestart, {skipValidation: true});

    function testStepdown(stepdownFunc) {
        rst = new ReplSetTest({nodes: 2});
        rst.startSet();
        rst.initiate();

        const primary = rst.getPrimary();
        const primaryDB = primary.getDB(dbName);

        const session = primaryDB.getMongo().startSession();
        const sessionDB = session.getDatabase(dbName);

        for (let i = 0; i < 4; i++) {
            assert.commandWorked(sessionDB.coll.insert({_id: i}, {writeConcern: {w: "majority"}}));
        }

        // Create a snapshot read cursor.
        const res = assert.commandWorked(sessionDB.runCommand({
            find: collName,
            batchSize: 2,
            readConcern: {level: "snapshot"},
            txnNumber: NumberLong(0),
            startTransaction: true,
            autocommit: false
        }));
        assert(res.hasOwnProperty("cursor"), tojson(res));
        assert(res.cursor.hasOwnProperty("id"), tojson(res));
        const cursorId = res.cursor.id;

        // It should be possible to step down the primary without hanging.
        stepdownFunc(rst);
        rst.waitForState(primary, ReplSetTest.State.SECONDARY);

        // Kill the cursor.
        assert.commandWorked(sessionDB.runCommand({killCursors: collName, cursors: [cursorId]}));
        rst.stopSet();
    }

    //
    // Test that stashed transaction resources are destroyed at stepdown triggered by
    // replSetStepDown.
    //
    function replSetStepDown(replSetTest) {
        assert.commandWorked(
            replSetTest.getPrimary().adminCommand({replSetStepDown: 60, force: true}));
    }
    testStepdown(replSetStepDown);

    //
    // Test that stashed transaction resources are destroyed at stepdown triggered by loss of
    // quorum.
    //
    function stepDownOnLossOfQuorum(replSetTest) {
        const secondary = rst.getSecondary();
        const secondaryId = rst.getNodeId(secondary);
        rst.stop(secondaryId);
    }
    testStepdown(stepDownOnLossOfQuorum);
})();