mongo/jstests/replsets/drop_db.js

/**
 * Ensures that the dropDatabase shell helper accepts a write concern as its first argument.
 *
 * @tags: [requires_replication]
 */
import {ReplSetTest} from "jstests/libs/replsettest.js";

function checkWriteConcern(testFn, checkFn) {
    const mongoRunCommandOriginal = Mongo.prototype.runCommand;

    const sentinel = {};
    let cmdObjSeen = sentinel;

    Mongo.prototype.runCommand = function runCommandSpy(dbName, cmdObj, options) {
        cmdObjSeen = cmdObj;
        return mongoRunCommandOriginal.apply(this, arguments);
    };

    try {
        assert.doesNotThrow(testFn);
    } finally {
        Mongo.prototype.runCommand = mongoRunCommandOriginal;
    }

    if (cmdObjSeen == sentinel) {
        throw new Error("Mongo.prototype.runCommand() was never called: " + testFn.toString());
    }

    checkFn(cmdObjSeen);
}

const rst = new ReplSetTest({nodes: 2});
rst.startSet();
rst.initiate();

const dbName = "dbDrop";
const collName = "coll";
const primaryDB = rst.getPrimary().getDB(dbName);

primaryDB.createCollection(collName);
checkWriteConcern(
    () => assert.commandWorked(primaryDB.dropDatabase({w: "majority"})),
    (cmdObj) => {
        assert.eq(cmdObj.writeConcern, {w: "majority"});
    },
);

primaryDB.createCollection(collName);
checkWriteConcern(
    () => assert.commandWorked(primaryDB.dropDatabase({w: 1})),
    (cmdObj) => {
        assert.eq(cmdObj.writeConcern, {w: 1});
    },
);

primaryDB.createCollection(collName);
checkWriteConcern(
    () => assert.commandFailedWithCode(primaryDB.dropDatabase({w: 45}), ErrorCodes.UnsatisfiableWriteConcern),
    (cmdObj) => {
        assert.eq(cmdObj.writeConcern, {w: 45});
    },
);

rst.stopSet();