summaryrefslogtreecommitdiff
path: root/jstests/replsets/retryable_write_concern.js
blob: 60163db9685b495ae5214245adc36b5f1555d65f (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
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
/**
 * Tests for making sure that retried writes will wait properly for writeConcern.
 *
 * @tags: [uses_transactions]
 */
(function() {

    "use strict";

    load("jstests/libs/retryable_writes_util.js");
    load("jstests/libs/write_concern_util.js");

    if (!RetryableWritesUtil.storageEngineSupportsRetryableWrites(jsTest.options().storageEngine)) {
        jsTestLog("Retryable writes are not supported, skipping test");
        return;
    }

    const kNodes = 2;

    /**
     * Tests that a retryable write properly waits for writeConcern on retry. Takes an optional
     * 'setupFunc' that sets up the database state. 'setupFunc' accepts a connection to the
     * primary.
     */
    let runTest = function(priConn, secConn, cmd, dbName, setupFunc) {
        dbName = dbName || "test";
        jsTestLog(`Testing ${tojson(cmd)} on ${dbName}.`);

        // Send a dummy write to this connection so it will have the Client object initialized.
        let secondPriConn = new Mongo(priConn.host);
        let testDB2 = secondPriConn.getDB(dbName);
        assert.writeOK(testDB2.dummy.insert({x: 1}, {writeConcern: {w: kNodes}}));

        if (setupFunc) {
            setupFunc(priConn);
        }

        stopServerReplication(secConn);

        let testDB = priConn.getDB(dbName);
        checkWriteConcernTimedOut(testDB.runCommand(cmd));
        checkWriteConcernTimedOut(testDB2.runCommand(cmd));

        restartServerReplication(secConn);
    };

    let replTest = new ReplSetTest({nodes: kNodes});
    replTest.startSet({verbose: 1});
    replTest.initiate();

    let priConn = replTest.getPrimary();
    let secConn = replTest.getSecondary();

    let lsid = UUID();

    runTest(priConn, secConn, {
        insert: 'user',
        documents: [{_id: 10}, {_id: 30}],
        ordered: false,
        lsid: {id: lsid},
        txnNumber: NumberLong(34),
        writeConcern: {w: 'majority', wtimeout: 200},
    });

    runTest(priConn, secConn, {
        update: 'user',
        updates: [
            {q: {_id: 10}, u: {$inc: {x: 1}}},
        ],
        ordered: false,
        lsid: {id: lsid},
        txnNumber: NumberLong(35),
        writeConcern: {w: 'majority', wtimeout: 200},
    });

    runTest(priConn, secConn, {
        delete: 'user',
        deletes: [{q: {x: 1}, limit: 1}, {q: {y: 1}, limit: 1}],
        ordered: false,
        lsid: {id: lsid},
        txnNumber: NumberLong(36),
        writeConcern: {w: 'majority', wtimeout: 200},
    });

    runTest(priConn, secConn, {
        findAndModify: 'user',
        query: {_id: 60},
        update: {$inc: {x: 1}},
        new: true,
        upsert: true,
        lsid: {id: lsid},
        txnNumber: NumberLong(37),
        writeConcern: {w: 'majority', wtimeout: 200},
    });

    runTest(priConn,
            secConn,
            {
              commitTransaction: 1,
              lsid: {id: lsid},
              txnNumber: NumberLong(38),
              autocommit: false,
              writeConcern: {w: 'majority', wtimeout: 200},
            },
            'admin',
            function(conn) {
                assert.commandWorked(conn.getDB('test').runCommand({
                    insert: 'user',
                    documents: [{_id: 80}, {_id: 90}],
                    ordered: false,
                    lsid: {id: lsid},
                    txnNumber: NumberLong(38),
                    readConcern: {level: 'snapshot'},
                    autocommit: false,
                    startTransaction: true
                }));

            });

    replTest.stopSet();
})();