mongodb/jstests/libs/parallel_shell_helpers.js

/**
 * Allows passing arguments to the function executed by startParallelShell.
 */
export const funWithArgs = (fn, ...args) => {
    const result = `(${fn.toString()})(${args.map(x => tojson(x)).reduce((x, y) => `${x}, ${y}`)})`;
    if (fn.constructor.name === 'AsyncFunction') {
        return `await ${result}`;
    }

    return result;
};

/**
 * Internal function used by _doAssertCommandInParallelShell().
 */
export const _parallelShellRunCommand = function(
    dbName, cmdObj, expectedCode, checkResultFn, checkResultArgs) {
    const expectedCodeStr =
        (expectedCode === undefined ? '' : '; expectedCode: ' + tojson(expectedCode));
    jsTestLog('Starting command in parallel shell - host: ' + db.getMongo().host +
              '; db: ' + dbName + '; command: ' + tojson(cmdObj) + expectedCodeStr);
    const testDB = db.getSiblingDB(dbName);
    const result = testDB.runCommand(cmdObj);
    jsTestLog('Finished running command in parallel shell - host: ' + db.getMongo().host +
              '; db: ' + dbName + '; command: ' + tojson(cmdObj) + expectedCodeStr +
              '; result: ' + tojson(result));
    if (expectedCode === undefined) {
        assert.commandWorked(result);
    } else {
        assert.commandFailedWithCode(result, expectedCode);
    }
    if (!checkResultFn) {
        return;
    }
    jsTestLog('Checking command result in parallel shell - host: ' + db.getMongo().host +
              '; db: ' + dbName + '; command: ' + tojson(cmdObj) + expectedCodeStr +
              '; result: ' + tojson(result) + '; checkResultFn: ' + checkResultFn +
              '; checkResultArgs: ' + tojson(checkResultArgs));
    checkResultFn(result, checkResultArgs);
    jsTestLog('Successfully verified command result in parallel shell - host: ' +
              db.getMongo().host + '; db: ' + dbName + '; command: ' + tojson(cmdObj) +
              expectedCodeStr + '; result: ' + tojson(result));
};

/**
 * Internal function used by assertCommandWorkedInParallelShell() and
 * assertCommandFailedWithCodeInParallelShell().
 */
export const _doAssertCommandInParallelShell = function(
    conn, db, cmdObj, expectedCode, checkResultFn, checkResultArgs) {
    // Return joinable object to caller.
    return startParallelShell(funWithArgs(_parallelShellRunCommand,
                                          db.getName(),
                                          cmdObj,
                                          expectedCode,
                                          checkResultFn,
                                          checkResultArgs),
                              conn.port);
};

/**
 * Starts command in a parallel shell.
 * Provides similar behavior to assert.commandWorked().
 */
export const assertCommandWorkedInParallelShell = function(
    conn, db, cmdObj, checkResultFn, checkResultArgs) {
    return _doAssertCommandInParallelShell(conn, db, cmdObj, checkResultFn, checkResultArgs);
};

/**
 * Starts command in a parallel shell.
 * Provides similar behavior to assert.commandFailedWithCode().
 */
export const assertCommandFailedWithCodeInParallelShell = function(
    conn, db, cmdObj, expectedCode, checkResultFn, checkResultArgs) {
    assert(expectedCode,
           'expected error code(s) must be provided to run command in parallel shell - host: ' +
               db.getMongo().host + '; db: ' + db.getName() + '; command: ' + tojson(cmdObj));
    return _doAssertCommandInParallelShell(
        conn, db, cmdObj, expectedCode, checkResultFn, checkResultArgs);
};