chromatic-lattice/tests/utils.ts

import Decimal, { DecimalSource, format } from "util/bignum";
import { Mock, expect, vi } from "vitest";

interface CustomMatchers<R = unknown> {
    compare_tolerance(expected: DecimalSource, tolerance?: number): R;
    toLogError(): R;
}

declare global {
    // eslint-disable-next-line @typescript-eslint/no-namespace
    namespace Vi {
        // eslint-disable-next-line @typescript-eslint/no-empty-interface
        interface Assertion extends CustomMatchers {}
        // eslint-disable-next-line @typescript-eslint/no-empty-interface
        interface AsymmetricMatchersContaining extends CustomMatchers {}
    }
}

expect.extend({
    compare_tolerance(received: DecimalSource, expected: DecimalSource, tolerance?: number) {
        const { isNot } = this;
        let pass = false;
        if (!Decimal.isFinite(expected)) {
            pass = !Decimal.isFinite(received);
        } else if (Decimal.isNaN(expected)) {
            pass = Decimal.isNaN(received);
        } else {
            pass = Decimal.eq_tolerance(received, expected, tolerance);
        }
        return {
            // do not alter your "pass" based on isNot. Vitest does it for you
            pass,
            message: () =>
                `Expected ${received} to${
                    (isNot as boolean) ? " not" : ""
                } be close to ${expected}`,
            expected: format(expected),
            actual: format(received)
        };
    },
    toLogError(received: () => unknown) {
        const { isNot } = this;
        console.error = vi.fn();
        received();
        const calls = (
            console.error as unknown as Mock<
                Parameters<typeof console.error>,
                ReturnType<typeof console.error>
            >
        ).mock.calls.length;
        const pass = calls >= 1;
        vi.restoreAllMocks();
        return {
            pass,
            message: () =>
                `Expected ${received} to ${(isNot as boolean) ? " not" : ""} log an error`,
            expected: "1+",
            actual: calls
        };
    }
});