githubmirror/webpack

test/setupTestFramework.js
147 строк4 KB
hai-x
test: generate snapshots per test case (#20675)
23 мар 2026, 15:19
23 мар 2026, 15:19a475512
"use strict";

const { toMatchSnapshot } = require("jest-snapshot");
const { getActiveSnapshotState } = require("./harness/snapshot");

expect.extend({
	toBeTypeOf(received, expected) {
		const objType = typeof received;
		const pass = objType === expected;

		const message = pass
			? () =>
					`${this.utils.matcherHint(".not.toBeTypeOf")}\n\n` +
					"Expected value to not be (using typeof):\n" +
					`  ${this.utils.printExpected(expected)}\n` +
					"Received:\n" +
					`  ${this.utils.printReceived(objType)}`
			: () =>
					`${this.utils.matcherHint(".toBeTypeOf")}\n\n` +
					"Expected value to be (using typeof):\n" +
					`  ${this.utils.printExpected(expected)}\n` +
					"Received:\n" +
					`  ${this.utils.printReceived(objType)}`;

		return { message, pass };
	},
	toEndWith(received, expected) {
		const pass = typeof received === "string" && received.endsWith(expected);

		const message = pass
			? () =>
					`${this.utils.matcherHint(".not.toEndWith")}\n\n` +
					"Expected value to not end with:\n" +
					`  ${this.utils.printExpected(expected)}\n` +
					"Received:\n" +
					`  ${this.utils.printReceived(received)}`
			: () =>
					`${this.utils.matcherHint(".toEndWith")}\n\n` +
					"Expected value to end with:\n" +
					`  ${this.utils.printExpected(expected)}\n` +
					"Received:\n" +
					`  ${this.utils.printReceived(received)}`;

		return { message, pass };
	},
	toMatchSnapshot(...args) {
		const snapshotState = getActiveSnapshotState();

		if (!snapshotState) {
			return toMatchSnapshot.apply(this, args);
		}

		const snapshotContext = Object.create(this);
		snapshotContext.snapshotState = snapshotState;
		return toMatchSnapshot.apply(snapshotContext, args);
	}
});

if (process.env.ALTERNATIVE_SORT) {
	const oldSort = Array.prototype.sort;

	// eslint-disable-next-line no-extend-native
	Array.prototype.sort = function sort(cmp) {
		oldSort.call(this, cmp);
		if (cmp) {
			for (let i = 1; i < this.length; i++) {
				if (cmp(this[i - 1], this[i]) === 0) {
					let j = i + 1;
					for (; j < this.length; j++) {
						if (cmp(this[j - 1], this[j]) !== 0) {
							break;
						}
					}
					for (let x = i - 1, y = j - 1; x < y; x++, y--) {
						const temp = this[x];
						this[x] = this[y];
						this[y] = temp;
					}
					i = j;
				}
			}
		}
		return this;
	};
}

// Setup debugging info for tests
if (process.env.DEBUG_INFO) {
	const addDebugInfo = (it) => (name, fn, timeout) => {
		if (fn.length === 0) {
			it(
				name,
				() => {
					process.stdout.write(`START1 ${name}\n`);
					try {
						const promise = fn();
						if (promise && promise.then) {
							return promise.then(
								(r) => {
									process.stdout.write(`DONE OK ${name}\n`);
									return r;
								},
								(err) => {
									process.stdout.write(`DONE FAIL ${name}\n`);
									throw err;
								}
							);
						}

						process.stdout.write(`DONE OK ${name}\n`);
					} catch (err) {
						process.stdout.write(`DONE FAIL ${name}\n`);
						throw err;
					}
				},
				timeout
			);
		} else {
			it(
				name,
				(done) => {
					process.stdout.write(`START2 ${name}\n`);
					return fn((err) => {
						if (err) {
							process.stdout.write(`DONE FAIL ${name}\n`);
						} else {
							process.stdout.write(`DONE OK ${name}\n`);
						}
						return done(err);
					});
				},
				timeout
			);
		}
	};

	// eslint-disable-next-line no-global-assign
	it = addDebugInfo(it);
}

// cspell:word wabt
// Workaround for a memory leak in wabt
// It leaks an Error object on construction
// so it leaks the whole stack trace
require("wast-loader");
process.removeAllListeners("uncaughtException");
process.removeAllListeners("unhandledRejection");