jstests/noPassthrough/dropdatabase_respect_maxtimems.js


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

/*
 * Tests that dropDatabase respects maxTimeMS.
 * @tags: [requires_replication, uses_transactions]
 */
(function() {
    const rst = ReplSetTest({nodes: 1});
    rst.startSet();
    rst.initiate();

    const adminDB = rst.getPrimary().getDB("admin");
    const txnDB = rst.getPrimary().getDB("txn");
    const dropDB = rst.getPrimary().getDB("drop");

    (function assertColletionDropCanBeInterrupted() {
        assert.commandWorked(txnDB.foo.insert({}));
        assert.commandWorked(dropDB.bar.insert({}));
        const session = txnDB.getMongo().startSession({causalConsistency: false});
        const sessionDB = session.getDatabase("txn");
        session.startTransaction();
        assert.commandWorked(sessionDB.foo.insert({}));
        assert.commandFailedWithCode(dropDB.runCommand({dropDatabase: 1, maxTimeMS: 100}),
                                     ErrorCodes.MaxTimeMSExpired);

        assert.commandWorked(session.commitTransaction_forTesting());
        session.endSession();
    })();

    (function assertDatabaseDropCanBeInterrupted() {
        assert.commandWorked(txnDB.foo.insert({}));
        assert.commandWorked(dropDB.bar.insert({}));

        assert.commandWorked(rst.getPrimary().adminCommand(
            {configureFailPoint: "dropDatabaseHangAfterAllCollectionsDrop", mode: "alwaysOn"}));

        // This will get blocked by the failpoint when collection drop phase finishes.
        let dropDatabaseShell = startParallelShell(
            "assert.commandFailedWithCode(db.getSiblingDB(\"drop\").runCommand({dropDatabase: 1, maxTimeMS: 5000}), ErrorCodes.MaxTimeMSExpired);",
            rst.getPrimary().port);

        assert.soon(function() {
            const sessionFilter = {active: true, "command.dropDatabase": 1};
            const res = adminDB.aggregate([{$currentOp: {}}, {$match: sessionFilter}]);
            return res.hasNext();
        }, "Timeout waiting for dropDatabase to start");

        const session = txnDB.getMongo().startSession({causalConsistency: false});
        const sessionDB = session.getDatabase("txn");
        session.startTransaction();
        assert.commandWorked(sessionDB.foo.insert({}));

        // dropDatabase now gets unblocked by the failpoint but will immediately
        // get blocked by acquiring the GlobalWrite lock for dropping the database.
        assert.commandWorked(rst.getPrimary().adminCommand(
            {configureFailPoint: "dropDatabaseHangAfterAllCollectionsDrop", mode: "off"}));

        // This should timeout.
        dropDatabaseShell();

        assert.commandWorked(session.commitTransaction_forTesting());
        session.endSession();
    })();

    rst.stopSet();
})();