@fast-check/vitest/lib/cjs/internals/TestBuilder.js

Property based testing for Vitest based on fast-check

"use strict";
Object.defineProperty(exports, "__esModule", { value: true });
exports.buildTest = buildTest;
const fast_check_1 = require("fast-check");
const TestWithPropRunnerBuilder_js_1 = require("./TestWithPropRunnerBuilder.js");
function adaptParametersForRecord(parameters, originalParamaters) {
    return {
        ...parameters,
        examples: parameters.examples !== undefined ? parameters.examples.map((example) => example[0]) : undefined,
        reporter: originalParamaters.reporter,
        asyncReporter: originalParamaters.asyncReporter,
    };
}
function adaptExecutionTreeForRecord(executionSummary) {
    return executionSummary.map((summary) => ({
        ...summary,
        value: summary.value[0],
        children: adaptExecutionTreeForRecord(summary.children),
    }));
}
function adaptRunDetailsForRecord(runDetails, originalParamaters) {
    const adaptedRunDetailsCommon = {
        ...runDetails,
        counterexample: runDetails.counterexample !== null ? runDetails.counterexample[0] : null,
        failures: runDetails.failures.map((failure) => failure[0]),
        executionSummary: adaptExecutionTreeForRecord(runDetails.executionSummary),
        runConfiguration: adaptParametersForRecord(runDetails.runConfiguration, originalParamaters),
    };
    return adaptedRunDetailsCommon;
}
function buildTestProp(testFn, fc) {
    return (arbitraries, params) => {
        if (Array.isArray(arbitraries)) {
            return (testName, prop, timeout) => (0, TestWithPropRunnerBuilder_js_1.buildTestWithPropRunner)(testFn, testName, arbitraries, prop, params, timeout, fc);
        }
        return (testName, prop, timeout) => {
            const recordArb = (0, fast_check_1.record)(arbitraries);
            const recordParams = params !== undefined
                ? {
                    ...params,
                    examples: params.examples !== undefined ? params.examples.map((example) => [example]) : undefined,
                    reporter: params.reporter !== undefined
                        ?
                            (runDetails) => params.reporter(adaptRunDetailsForRecord(runDetails, params))
                        : undefined,
                    asyncReporter: params.asyncReporter !== undefined
                        ?
                            (runDetails) => params.asyncReporter(adaptRunDetailsForRecord(runDetails, params))
                        : undefined,
                }
                : undefined;
            (0, TestWithPropRunnerBuilder_js_1.buildTestWithPropRunner)(testFn, testName, [recordArb], (value) => prop(value), recordParams, timeout, fc);
        };
    };
}
function buildTest(testFn, fc, ancestors = new Set()) {
    let atLeastOneExtra = false;
    const extraKeys = {};
    for (const unsafeKey of Object.getOwnPropertyNames(testFn)) {
        const key = unsafeKey;
        if (!ancestors.has(key) && typeof testFn[key] === 'function') {
            atLeastOneExtra = true;
            extraKeys[key] = key !== 'each' ? buildTest(testFn[key], fc, new Set([...ancestors, key])) : testFn[key];
        }
    }
    if (!atLeastOneExtra) {
        return testFn;
    }
    const enrichedTestFn = (...args) => testFn(...args);
    if ('each' in testFn) {
        extraKeys['prop'] = buildTestProp(testFn, fc);
    }
    return Object.assign(enrichedTestFn, extraKeys);
}