summaryrefslogtreecommitdiff
path: root/jstests/noPassthroughWithMongod/top_drop.js
blob: 17949e59ff63a5102e888022702e4100c8fef634 (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
/**
 * Checks that top removes entries after dropping a collection or database.
 * TODO(SERVER-21167): Move this test from noPassthrough to core.
 */
(function() {
    "use strict";

    let topDB = db.getSiblingDB("topdrop");
    assert.commandWorked(topDB.dropDatabase());

    // Asserts that the output of top contains exactly these collection entries for topDB.
    function checkTopEntries(expectedEntries) {
        let res = topDB.adminCommand("top");
        assert.commandWorked(res, "Failed to run the top command");

        let entriesInTop = Object.keys(res.totals).filter(function(ns) {
            // This filter only includes non-system collections in our test database.
            const dbPrefix = topDB.getName() + ".";
            const systemCollectionPrefix = "system.";
            return ns.startsWith(dbPrefix) && !ns.startsWith(dbPrefix + systemCollectionPrefix);
        });
        let expectedEntryNames = expectedEntries.map(function(coll) {
            return coll.getFullName();
        });

        const entriesAreEqual = friendlyEqual(entriesInTop.sort(), expectedEntryNames.sort());
        if (!entriesAreEqual) {
            // TODO(SERVER-26750): This block can be removed once SERVER-26750 is resolved.
            jsTest.log("Warning: expected to see " + tojson(expectedEntryNames) +
                       " in top, but got " + tojson(entriesInTop));

            assert.lt(expectedEntryNames.length,
                      entriesInTop.length,
                      "Fewer entries in top than expected; got " + tojson(entriesInTop) +
                          " but expected " + tojson(expectedEntryNames) + "\nFull top output:\n" +
                          tojson(res.totals));

            // We allow an unexpected entry in top if the insert counter has been cleared. This is
            // probably due to a background job releasing an AutoGetCollectionForReadCommand for
            // that collection.
            entriesInTop.forEach(function(coll) {
                if (expectedEntryNames.includes(coll)) {
                    return;
                }

                let topStats = res.totals[coll];
                assert.eq(0,
                          res.totals[coll].insert.count,
                          coll + " has unexpected insert entries in top. Full top output:\n" +
                              tojson(res.totals));
            });
        }
    }

    // Create a few entries in top.
    assert.writeOK(topDB.coll1.insert({}));
    assert.writeOK(topDB.coll2.insert({}));
    assert.writeOK(topDB.coll3.insert({}));
    checkTopEntries([topDB.coll1, topDB.coll2, topDB.coll3]);

    // Check that dropping a collection removes that collection but leaves the others.
    assert.commandWorked(topDB.runCommand({drop: "coll3"}));
    checkTopEntries([topDB.coll1, topDB.coll2]);

    // Check that dropping the database removes the remaining collections.
    assert.commandWorked(topDB.dropDatabase());
    checkTopEntries([]);

    // Check that top doesn't keep state about non-existent collections.
    assert.commandWorked(topDB.dropDatabase());
    topDB.foo.find().itcount();
    topDB.baz.update({}, {$set: {a: 1}});
    topDB.bar.remove({});

    let res = topDB.adminCommand("top");
    checkTopEntries([]);

    assert.commandWorked(topDB.dropDatabase());
}());