jstests/replsets/drop_collections_two_phase_step_down.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
65
66
67
68

/**
 * Test to ensure that a primary does not erroneously commit a two phase collection drop if it
 * steps down and back up again before the drop has been committed.
 *
 * Test steps:
 *
 * 1. Pause oplog application on secondary.
 * 2. Drop collection 'collToDrop' on the primary.
 * 3. Force primary to step down.
 * 4. Wait for primary to step up again.
 * 5. Make sure that collection 'collToDrop' has not been physically dropped on the primary.
 * 6. Resume oplog application on secondary and make sure collection drop is eventually committed.
 */

(function() {
    "use strict";

    load("jstests/replsets/libs/two_phase_drops.js");  // For TwoPhaseDropCollectionTest.

    // Set up a two phase drop test.
    let testName = "drop_collection_two_phase_step_down";
    let dbName = testName;
    let collName = "collToDrop";
    let twoPhaseDropTest = new TwoPhaseDropCollectionTest(testName, dbName);

    // Initialize replica set.
    let replTest = twoPhaseDropTest.initReplSet();

    // Create the collection that will be dropped.
    twoPhaseDropTest.createCollection(collName);

    // PREPARE collection drop.
    twoPhaseDropTest.prepareDropCollection(collName);

    // Step primary down using {force: true} and wait for the same node to become primary again.
    // We use {force: true} because the current secondary has oplog application disabled and will
    // not be able to take over as primary.
    try {
        const primary = replTest.getPrimary();
        const primaryId = replTest.getNodeId(primary);

        // Force step down primary.
        jsTestLog('Stepping down primary ' + primary.host + ' with {force: true}.');
        const stepDownResult = assert.throws(function() {
            // The amount of time the node has to wait before becoming primary again.
            const stepDownSecs = 1;
            primary.adminCommand({replSetStepDown: stepDownSecs, force: true});
        });
        let errMsg = 'Expected exception from stepping down primary ' + primary.host + ': ' +
            tojson(stepDownResult);
        assert(isNetworkError(stepDownResult), errMsg);

        // Wait for the node that stepped down to regain PRIMARY status.
        jsTestLog('Waiting for node ' + primary.host + ' to become primary again');
        assert.eq(replTest.nodes[primaryId], replTest.getPrimary());

        jsTestLog('Node ' + primary.host + ' is now PRIMARY again. Checking if drop-pending' +
                  ' collection still exists.');
        assert(twoPhaseDropTest.collectionIsPendingDrop(collName),
               'After stepping down and back up again, the primary ' + primary.host +
                   ' removed drop-pending collection unexpectedly');

        // COMMIT collection drop.
        twoPhaseDropTest.commitDropCollection(collName);
    } finally {
        twoPhaseDropTest.stop();
    }
}());