import { red, stripColor } from "../fmt/colors.ts";import { buildMessage, diff, diffstr } from "./_diff.ts";import { format } from "./_format.ts";
const CAN_NOT_DISPLAY = "[Cannot display]";
export class AssertionError extends Error { override name = "AssertionError"; constructor(message: string) { super(message); }}
function isKeyedCollection(x: unknown): x is Set<unknown> { return [Symbol.iterator, "size"].every((k) => k in (x as Set<unknown>));}
export function equal(c: unknown, d: unknown): boolean { const seen = new Map(); return (function compare(a: unknown, b: unknown): boolean { if ( a && b && ((a instanceof RegExp && b instanceof RegExp) || (a instanceof URL && b instanceof URL)) ) { return String(a) === String(b); } if (a instanceof Date && b instanceof Date) { const aTime = a.getTime(); const bTime = b.getTime(); if (Number.isNaN(aTime) && Number.isNaN(bTime)) { return true; } return aTime === bTime; } if (typeof a === "number" && typeof b === "number") { return Number.isNaN(a) && Number.isNaN(b) || a === b; } if (Object.is(a, b)) { return true; } if (a && typeof a === "object" && b && typeof b === "object") { if (a && b && !constructorsEqual(a, b)) { return false; } if (a instanceof WeakMap || b instanceof WeakMap) { if (!(a instanceof WeakMap && b instanceof WeakMap)) return false; throw new TypeError("cannot compare WeakMap instances"); } if (a instanceof WeakSet || b instanceof WeakSet) { if (!(a instanceof WeakSet && b instanceof WeakSet)) return false; throw new TypeError("cannot compare WeakSet instances"); } if (seen.get(a) === b) { return true; } if (Object.keys(a || {}).length !== Object.keys(b || {}).length) { return false; } seen.set(a, b); if (isKeyedCollection(a) && isKeyedCollection(b)) { if (a.size !== b.size) { return false; }
let unmatchedEntries = a.size;
for (const [aKey, aValue] of a.entries()) { for (const [bKey, bValue] of b.entries()) { if ( (aKey === aValue && bKey === bValue && compare(aKey, bKey)) || (compare(aKey, bKey) && compare(aValue, bValue)) ) { unmatchedEntries--; break; } } }
return unmatchedEntries === 0; } const merged = { ...a, ...b }; for ( const key of [ ...Object.getOwnPropertyNames(merged), ...Object.getOwnPropertySymbols(merged), ] ) { type Key = keyof typeof merged; if (!compare(a && a[key as Key], b && b[key as Key])) { return false; } if (((key in a) && (!(key in b))) || ((key in b) && (!(key in a)))) { return false; } } if (a instanceof WeakRef || b instanceof WeakRef) { if (!(a instanceof WeakRef && b instanceof WeakRef)) return false; return compare(a.deref(), b.deref()); } return true; } return false; })(c, d);}
function constructorsEqual(a: object, b: object) { return a.constructor === b.constructor || a.constructor === Object && !b.constructor || !a.constructor && b.constructor === Object;}
export function assert(expr: unknown, msg = ""): asserts expr { if (!expr) { throw new AssertionError(msg); }}
type Falsy = false | 0 | 0n | "" | null | undefined;export function assertFalse(expr: unknown, msg = ""): asserts expr is Falsy { if (expr) { throw new AssertionError(msg); }}
export function assertEquals<T>(actual: T, expected: T, msg?: string) { if (equal(actual, expected)) { return; } let message = ""; const actualString = format(actual); const expectedString = format(expected); try { const stringDiff = (typeof actual === "string") && (typeof expected === "string"); const diffResult = stringDiff ? diffstr(actual as string, expected as string) : diff(actualString.split("\n"), expectedString.split("\n")); const diffMsg = buildMessage(diffResult, { stringDiff }).join("\n"); message = `Values are not equal:\n${diffMsg}`; } catch { message = `\n${red(CAN_NOT_DISPLAY)} + \n\n`; } if (msg) { message = msg; } throw new AssertionError(message);}
export function assertNotEquals<T>(actual: T, expected: T, msg?: string) { if (!equal(actual, expected)) { return; } let actualString: string; let expectedString: string; try { actualString = String(actual); } catch { actualString = "[Cannot display]"; } try { expectedString = String(expected); } catch { expectedString = "[Cannot display]"; } if (!msg) { msg = `actual: ${actualString} expected not to be: ${expectedString}`; } throw new AssertionError(msg);}
export function assertStrictEquals<T>( actual: unknown, expected: T, msg?: string,): asserts actual is T { if (Object.is(actual, expected)) { return; }
let message: string;
if (msg) { message = msg; } else { const actualString = format(actual); const expectedString = format(expected);
if (actualString === expectedString) { const withOffset = actualString .split("\n") .map((l) => ` ${l}`) .join("\n"); message = `Values have the same structure but are not reference-equal:\n\n${ red(withOffset) }\n`; } else { try { const stringDiff = (typeof actual === "string") && (typeof expected === "string"); const diffResult = stringDiff ? diffstr(actual as string, expected as string) : diff(actualString.split("\n"), expectedString.split("\n")); const diffMsg = buildMessage(diffResult, { stringDiff }).join("\n"); message = `Values are not strictly equal:\n${diffMsg}`; } catch { message = `\n${red(CAN_NOT_DISPLAY)} + \n\n`; } } }
throw new AssertionError(message);}
export function assertNotStrictEquals<T>( actual: T, expected: T, msg?: string,) { if (!Object.is(actual, expected)) { return; }
throw new AssertionError( msg ?? `Expected "actual" to be strictly unequal to: ${format(actual)}\n`, );}
export function assertAlmostEquals( actual: number, expected: number, tolerance = 1e-7, msg?: string,) { if (Object.is(actual, expected)) { return; } const delta = Math.abs(expected - actual); if (delta <= tolerance) { return; } const f = (n: number) => Number.isInteger(n) ? n : n.toExponential(); throw new AssertionError( msg ?? `actual: "${f(actual)}" expected to be close to "${f(expected)}": \delta "${f(delta)}" is greater than "${f(tolerance)}"`, );}
type AnyConstructor = new (...args: any[]) => any;type GetConstructorType<T extends AnyConstructor> = T extends new (...args: any) => infer C ? C : never;
export function assertInstanceOf<T extends AnyConstructor>( actual: unknown, expectedType: T, msg = "",): asserts actual is GetConstructorType<T> { if (!msg) { const expectedTypeStr = expectedType.name;
let actualTypeStr = ""; if (actual === null) { actualTypeStr = "null"; } else if (actual === undefined) { actualTypeStr = "undefined"; } else if (typeof actual === "object") { actualTypeStr = actual.constructor?.name ?? "Object"; } else { actualTypeStr = typeof actual; }
if (expectedTypeStr == actualTypeStr) { msg = `Expected object to be an instance of "${expectedTypeStr}".`; } else if (actualTypeStr == "function") { msg = `Expected object to be an instance of "${expectedTypeStr}" but was not an instanced object.`; } else { msg = `Expected object to be an instance of "${expectedTypeStr}" but was "${actualTypeStr}".`; } } assert(actual instanceof expectedType, msg);}
export function assertNotInstanceOf<A, T>( actual: A, unexpectedType: new (...args: any[]) => T, msg = `Expected object to not be an instance of "${typeof unexpectedType}"`,): asserts actual is Exclude<A, T> { assertFalse(actual instanceof unexpectedType, msg);}
export function assertExists<T>( actual: T, msg?: string,): asserts actual is NonNullable<T> { if (actual === undefined || actual === null) { if (!msg) { msg = `actual: "${actual}" expected to not be null or undefined`; } throw new AssertionError(msg); }}
export function assertStringIncludes( actual: string, expected: string, msg?: string,) { if (!actual.includes(expected)) { if (!msg) { msg = `actual: "${actual}" expected to contain: "${expected}"`; } throw new AssertionError(msg); }}
export function assertArrayIncludes<T>( actual: ArrayLike<T>, expected: ArrayLike<T>, msg?: string,) { const missing: unknown[] = []; for (let i = 0; i < expected.length; i++) { let found = false; for (let j = 0; j < actual.length; j++) { if (equal(expected[i], actual[j])) { found = true; break; } } if (!found) { missing.push(expected[i]); } } if (missing.length === 0) { return; } if (!msg) { msg = `actual: "${format(actual)}" expected to include: "${ format(expected) }"\nmissing: ${format(missing)}`; } throw new AssertionError(msg);}
export function assertMatch( actual: string, expected: RegExp, msg?: string,) { if (!expected.test(actual)) { if (!msg) { msg = `actual: "${actual}" expected to match: "${expected}"`; } throw new AssertionError(msg); }}
export function assertNotMatch( actual: string, expected: RegExp, msg?: string,) { if (expected.test(actual)) { if (!msg) { msg = `actual: "${actual}" expected to not match: "${expected}"`; } throw new AssertionError(msg); }}
export function assertObjectMatch( actual: Record<PropertyKey, any>, expected: Record<PropertyKey, unknown>,) { type loose = Record<PropertyKey, unknown>;
function filter(a: loose, b: loose) { const seen = new WeakMap(); return fn(a, b);
function fn(a: loose, b: loose): loose { if ((seen.has(a)) && (seen.get(a) === b)) { return a; } seen.set(a, b); const filtered = {} as loose; const entries = [ ...Object.getOwnPropertyNames(a), ...Object.getOwnPropertySymbols(a), ] .filter((key) => key in b) .map((key) => [key, a[key as string]]) as Array<[string, unknown]>; for (const [key, value] of entries) { if (Array.isArray(value)) { const subset = (b as loose)[key]; if (Array.isArray(subset)) { filtered[key] = fn({ ...value }, { ...subset }); continue; } } else if (value instanceof RegExp) { filtered[key] = value; continue; } else if (typeof value === "object") { const subset = (b as loose)[key]; if ((typeof subset === "object") && (subset)) { if ((value instanceof Map) && (subset instanceof Map)) { filtered[key] = new Map( [...value].filter(([k]) => subset.has(k)).map(( [k, v], ) => [k, typeof v === "object" ? fn(v, subset.get(k)) : v]), ); continue; } if ((value instanceof Set) && (subset instanceof Set)) { filtered[key] = new Set([...value].filter((v) => subset.has(v))); continue; } filtered[key] = fn(value as loose, subset as loose); continue; } } filtered[key] = value; } return filtered; } } return assertEquals( filter(actual, expected), filter(expected, expected), );}
export function fail(msg?: string): never { assert(false, `Failed assertion${msg ? `: ${msg}` : "."}`);}
export function assertIsError<E extends Error = Error>( error: unknown, ErrorClass?: new (...args: any[]) => E, msgIncludes?: string, msg?: string,): asserts error is E { if (error instanceof Error === false) { throw new AssertionError(`Expected "error" to be an Error object.`); } if (ErrorClass && !(error instanceof ErrorClass)) { msg = `Expected error to be instance of "${ErrorClass.name}", but was "${ typeof error === "object" ? error?.constructor?.name : "[not an object]" }"${msg ? `: ${msg}` : "."}`; throw new AssertionError(msg); } if ( msgIncludes && (!(error instanceof Error) || !stripColor(error.message).includes(stripColor(msgIncludes))) ) { msg = `Expected error message to include "${msgIncludes}", but got "${ error instanceof Error ? error.message : "[not an Error]" }"${msg ? `: ${msg}` : "."}`; throw new AssertionError(msg); }}
export function assertThrows( fn: () => unknown, msg?: string,): unknown;export function assertThrows<E extends Error = Error>( fn: () => unknown, ErrorClass: new (...args: any[]) => E, msgIncludes?: string, msg?: string,): E;export function assertThrows<E extends Error = Error>( fn: () => unknown, errorClassOrMsg?: | (new (...args: any[]) => E) | string, msgIncludesOrMsg?: string, msg?: string,): E | Error | unknown { let ErrorClass: (new (...args: any[]) => E) | undefined = undefined; let msgIncludes: string | undefined = undefined; let err;
if (typeof errorClassOrMsg !== "string") { if ( errorClassOrMsg === undefined || errorClassOrMsg.prototype instanceof Error || errorClassOrMsg.prototype === Error.prototype ) { ErrorClass = errorClassOrMsg as new (...args: any[]) => E; msgIncludes = msgIncludesOrMsg; } else { msg = msgIncludesOrMsg; } } else { msg = errorClassOrMsg; } let doesThrow = false; const msgToAppendToError = msg ? `: ${msg}` : "."; try { fn(); } catch (error) { if (ErrorClass) { if (error instanceof Error === false) { throw new AssertionError("A non-Error object was thrown."); } assertIsError( error, ErrorClass, msgIncludes, msg, ); } err = error; doesThrow = true; } if (!doesThrow) { msg = `Expected function to throw${msgToAppendToError}`; throw new AssertionError(msg); } return err;}
export function assertRejects( fn: () => PromiseLike<unknown>, msg?: string,): Promise<unknown>;export function assertRejects<E extends Error = Error>( fn: () => PromiseLike<unknown>, ErrorClass: new (...args: any[]) => E, msgIncludes?: string, msg?: string,): Promise<E>;export async function assertRejects<E extends Error = Error>( fn: () => PromiseLike<unknown>, errorClassOrMsg?: | (new (...args: any[]) => E) | string, msgIncludesOrMsg?: string, msg?: string,): Promise<E | Error | unknown> { let ErrorClass: (new (...args: any[]) => E) | undefined = undefined; let msgIncludes: string | undefined = undefined; let err;
if (typeof errorClassOrMsg !== "string") { if ( errorClassOrMsg === undefined || errorClassOrMsg.prototype instanceof Error || errorClassOrMsg.prototype === Error.prototype ) { ErrorClass = errorClassOrMsg as new (...args: any[]) => E; msgIncludes = msgIncludesOrMsg; } } else { msg = errorClassOrMsg; } let doesThrow = false; let isPromiseReturned = false; const msgToAppendToError = msg ? `: ${msg}` : "."; try { const possiblePromise = fn(); if ( possiblePromise && typeof possiblePromise === "object" && typeof possiblePromise.then === "function" ) { isPromiseReturned = true; await possiblePromise; } } catch (error) { if (!isPromiseReturned) { throw new AssertionError( `Function throws when expected to reject${msgToAppendToError}`, ); } if (ErrorClass) { if (error instanceof Error === false) { throw new AssertionError("A non-Error object was rejected."); } assertIsError( error, ErrorClass, msgIncludes, msg, ); } err = error; doesThrow = true; } if (!doesThrow) { throw new AssertionError( `Expected function to reject${msgToAppendToError}`, ); } return err;}
export function unimplemented(msg?: string): never { throw new AssertionError(msg || "unimplemented");}
export function unreachable(): never { throw new AssertionError("unreachable");}