import {FixtureHelpers} from "jstests/libs/fixture_helpers.js";

/**
 * Asserts that releaseMemory command failed for a given cursor with the appropriate code.
 */
export function assertReleaseMemoryFailedWithCode(result, cursorId, codes) {
    if (!Array.isArray(codes)) {
        codes = [codes];
    }
    if (!result.hasOwnProperty("cursorsWithErrors")) {
        doassert("Command result does not contain 'cursorsWithErrors' field: " + tojson(result));
    }
    for (const releaseMemoryError of result.cursorsWithErrors) {
        if (releaseMemoryError.cursorId.compare(cursorId) === 0) {
            assert.contains(
                releaseMemoryError.status.code,
                codes,
                "the following release memory error contains an unexpected error code: " + tojson(releaseMemoryError),
            );
            return;
        }
    }
    doassert(
        "Cursor " +
            tojsononeline(cursorId) +
            " did not fail during release memory. Full command result: " +
            tojson(result),
    );
}

/**
 * Asserts that releaseMemory command worked for a given cursor.
 */
export function assertReleaseMemoryWorked(result, cursorId) {
    if (!result.hasOwnProperty("cursorsReleased")) {
        doassert("Command result does not contain 'cursorsReleased' field: " + tojson(result));
    }
    for (const releaseMemoryOk of result.cursorsReleased) {
        if (releaseMemoryOk.compare(cursorId) === 0) {
            return;
        }
    }
    doassert(
        "Releasing memory from cursor " +
            tojsononeline(cursorId) +
            " did not succeed during release memory. Full command result: " +
            tojson(result),
    );
}

/**
 * Accumulate metric from a server status
 */
export function accumulateServerStatusMetric(db, metricGetter) {
    return retryOnRetryableError(
        () => {
            let total = 0;
            FixtureHelpers.mapOnEachShardNode({
                db: db,
                func: (db) => {
                    const serverStatus = db.serverStatus();
                    if (!serverStatus.hasOwnProperty("metrics")) {
                        return;
                    }
                    total += metricGetter(serverStatus.metrics);
                },
            });
            return total;
        },
        10,
        100,
        [ErrorCodes.InterruptedDueToStorageChange],
    );
}

// Sets the mode of the simulateAvailableDiskSpace failpoint.
export function setAvailableDiskSpaceMode(db, mode, bytes = 450 * 1024 * 1024) {
    FixtureHelpers.runCommandOnEachPrimary({
        db: db,
        cmdObj: {configureFailPoint: "simulateAvailableDiskSpace", mode, "data": {bytes: bytes}},
    });
}