assertloom 0.0.0-stage → 0.2.0

package/dist/repair/repairLoop.js

@@ -0,0 +1,219 @@
+import { createHash } from "node:crypto";
+import { readFile, writeFile } from "node:fs/promises";
+import path from "node:path";
+import { promptTokens } from "../ai/pricing.js";
+import { buildRepairPrompt, REPAIR_SYSTEM_PROMPT } from "../ai/prompts.js";
+import { checkBody, renderTestFile } from "../generator/index.js";
+import { runJest } from "../runner/index.js";
+import { classifyTestFailure } from "./classifyFailure.js";
+import { RepairSchema } from "./schema.js";
+export const MAX_REPAIR_ATTEMPTS = 3;
+const MAX_TARGETS_PER_ATTEMPT = 15;
+const MAX_OUTPUT_TOKENS = 16_000;
+export function summarize(run) {
+    return { status: run.status, passed: run.passed, failed: run.failed, skipped: run.skipped, coverage: run.coverage };
+}
+const key = (functionName, name) => `${functionName}\u0000${name}`;
+function cloneSuites(suites) {
+    return suites.map((s) => ({ binding: s.binding, cases: s.cases.map((c) => ({ ...c })) }));
+}
+function outcomesByKey(run) {
+    return new Map(run.tests.map((t) => [key(t.suite, t.name), t]));
+}
+function findFailing(suites, run) {
+    const outcomes = outcomesByKey(run);
+    const failing = [];
+    suites.forEach((suite, suiteIndex) => suite.cases.forEach((testCase, caseIndex) => {
+        const outcome = outcomes.get(key(suite.binding.binding, testCase.name));
+        if (testCase.skipReason || outcome?.status !== "failed")
+            return;
+        failing.push({ suiteIndex, caseIndex, outcome, ...classifyTestFailure(outcome) });
+    }));
+    return failing;
+}
+function firstLine(text) {
+    return text.split("\n").find((line) => line.trim())?.trim() ?? "";
+}
+function activeCaseCount(suites) {
+    return suites.reduce((n, s) => n + s.cases.filter((c) => !c.skipReason).length, 0);
+}
+function sha256(text) {
+    return createHash("sha256").update(text).digest("hex");
+}
+/**
+ * Repairs failing generated tests in bounded rounds. Only the test file is ever written;
+ * a round that breaks a previously passing test is reverted and ends the loop.
+ */
+export async function repairTests(options) {
+    if (path.resolve(options.testPath) === path.resolve(options.sourcePath)) {
+        throw new Error("Refusing to repair: the test path is the source file.");
+    }
+    const maxAttempts = Math.max(0, Math.min(options.maxAttempts ?? MAX_REPAIR_ATTEMPTS, MAX_REPAIR_ATTEMPTS));
+    const runTests = options.runTests ?? ((testPath) => runJest({ testPath, sourcePath: options.sourcePath, allowNetwork: options.allowNetwork }));
+    const originalSource = await readFile(options.sourcePath, "utf8");
+    const originalHash = sha256(originalSource);
+    const render = (suites) => renderTestFile({
+        sourceFile: options.analysis.file,
+        preamble: options.preamble,
+        suites,
+        extension: options.extension,
+    });
+    let suites = cloneSuites(options.suites);
+    let content = render(suites);
+    let run = options.initialRun;
+    const attempts = [];
+    const usage = { inputTokens: 0, outputTokens: 0 };
+    let stopReason;
+    while (!stopReason && attempts.length < maxAttempts) {
+        if (run.infra) {
+            stopReason = "unrepairable";
+            break;
+        }
+        const failing = findFailing(suites, run);
+        if (failing.length === 0) {
+            stopReason = "all-passing";
+            break;
+        }
+        const targets = failing.filter((f) => f.repairable).slice(0, MAX_TARGETS_PER_ATTEMPT);
+        if (targets.length === 0) {
+            stopReason = "unrepairable";
+            break;
+        }
+        const repairTargets = targets.map((t) => {
+            const suite = suites[t.suiteIndex];
+            const testCase = suite.cases[t.caseIndex];
+            return {
+                binding: suite.binding,
+                testName: testCase.name,
+                body: testCase.body,
+                failureCategory: t.category,
+                message: t.outcome.failure?.message ?? "",
+            };
+        });
+        const prompt = buildRepairPrompt({
+            analysis: options.analysis,
+            code: options.promptSource,
+            targets: repairTargets,
+            attempt: attempts.length + 1,
+            mockSummary: options.mockSummary,
+        });
+        if (options.budget && !options.budget.allows({ inputTokens: promptTokens(REPAIR_SYSTEM_PROMPT, prompt), outputTokens: MAX_OUTPUT_TOKENS })) {
+            stopReason = "budget";
+            break;
+        }
+        const attempt = {
+            attempt: attempts.length + 1,
+            before: summarize(run),
+            requested: targets.length,
+            changes: [],
+            invalid: [],
+            regressed: false,
+            usage: { inputTokens: 0, outputTokens: 0 },
+        };
+        attempts.push(attempt);
+        const response = await options.provider.generateStructured({
+            system: REPAIR_SYSTEM_PROMPT,
+            prompt,
+            schema: RepairSchema,
+            schemaName: "assertloom_repairs",
+            maxOutputTokens: MAX_OUTPUT_TOKENS,
+        });
+        options.budget?.add(response.usage);
+        attempt.usage = response.usage;
+        usage.inputTokens += response.usage.inputTokens;
+        usage.outputTokens += response.usage.outputTokens;
+        const proposals = new Map(response.data.repairs.map((r) => [key(r.functionName, r.name), r]));
+        const candidate = cloneSuites(suites);
+        const removals = [];
+        for (const target of targets) {
+            const suite = candidate[target.suiteIndex];
+            const testCase = suite.cases[target.caseIndex];
+            const functionName = suite.binding.binding;
+            const proposal = proposals.get(key(functionName, testCase.name));
+            const invalid = (reason) => attempt.invalid.push({ functionName, name: testCase.name, reason });
+            if (!proposal) {
+                invalid("no repair returned");
+                continue;
+            }
+            if (proposal.action === "fix") {
+                const check = checkBody(proposal.body, options.extension, new Set(suites.map((s) => s.binding.binding)));
+                if ("error" in check) {
+                    invalid(`proposed body ${check.error}`);
+                    continue;
+                }
+                if (proposal.body.trim() === testCase.body.trim()) {
+                    invalid("proposed body is unchanged");
+                    continue;
+                }
+                testCase.body = proposal.body.trim();
+                testCase.async = check.async;
+            }
+            else if (proposal.action === "skip") {
+                testCase.skipReason = `suspected bug in the source: ${proposal.reason}`;
+            }
+            else {
+                removals.push({ suiteIndex: target.suiteIndex, caseIndex: target.caseIndex });
+            }
+            attempt.changes.push({ functionName, name: testCase.name, action: proposal.action, reason: proposal.reason });
+        }
+        for (const { suiteIndex, caseIndex } of removals.sort((a, b) => b.caseIndex - a.caseIndex)) {
+            candidate[suiteIndex].cases.splice(caseIndex, 1);
+        }
+        const nonEmpty = candidate.filter((s) => s.cases.length > 0);
+        if (attempt.changes.length === 0 || activeCaseCount(nonEmpty) === 0) {
+            stopReason = "no-valid-repairs";
+            break;
+        }
+        const candidateContent = render(nonEmpty);
+        await writeFile(options.testPath, candidateContent, "utf8");
+        const candidateRun = await runTests(options.testPath);
+        attempt.after = summarize(candidateRun);
+        const before = outcomesByKey(run);
+        const after = outcomesByKey(candidateRun);
+        const brokeAPassingTest = [...before].some(([testKey, outcome]) => outcome.status === "passed" && after.get(testKey)?.status === "failed");
+        if (candidateRun.infra || brokeAPassingTest) {
+            attempt.regressed = true;
+            await writeFile(options.testPath, content, "utf8");
+            stopReason = "regressed";
+            break;
+        }
+        suites = nonEmpty;
+        content = candidateContent;
+        run = candidateRun;
+    }
+    const gaveUp = [];
+    if (!run.infra) {
+        const remaining = findFailing(suites, run);
+        if (remaining.length > 0) {
+            for (const f of remaining) {
+                const testCase = suites[f.suiteIndex].cases[f.caseIndex];
+                const reason = f.category === "mock"
+                    ? "opens a real network connection; mock the dependency (--mock <module>) or rerun with --allow-network"
+                    : `still failing after repair (${f.category}): ${firstLine(f.outcome.failure?.message ?? "")}`;
+                testCase.skipReason = reason;
+                gaveUp.push({ functionName: suites[f.suiteIndex].binding.binding, name: testCase.name, action: "gave-up", reason });
+            }
+            content = render(suites);
+            await writeFile(options.testPath, content, "utf8");
+            run = await runTests(options.testPath);
+        }
+    }
+    stopReason ??= findFailing(suites, run).length || gaveUp.length ? "max-attempts" : "all-passing";
+    const currentSource = await readFile(options.sourcePath, "utf8");
+    const sourceUnchanged = sha256(currentSource) === originalHash;
+    if (!sourceUnchanged)
+        await writeFile(options.sourcePath, originalSource, "utf8");
+    return {
+        suites,
+        content,
+        run,
+        initial: summarize(options.initialRun),
+        final: summarize(run),
+        attempts,
+        gaveUp,
+        stopReason,
+        sourceUnchanged,
+        usage,
+    };
+}
+//# sourceMappingURL=repairLoop.js.map

package/dist/repair/repairLoop.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"repairLoop.js","sourceRoot":"","sources":["../../src/repair/repairLoop.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AACzC,OAAO,EAAE,QAAQ,EAAE,SAAS,EAAE,MAAM,kBAAkB,CAAC;AACvD,OAAO,IAAI,MAAM,WAAW,CAAC;AAC7B,OAAO,EAAE,YAAY,EAAe,MAAM,kBAAkB,CAAC;AAC7D,OAAO,EAAE,iBAAiB,EAAE,oBAAoB,EAAqB,MAAM,kBAAkB,CAAC;AAE9F,OAAO,EAAE,SAAS,EAAE,cAAc,EAAuB,MAAM,uBAAuB,CAAC;AACvF,OAAO,EAAE,OAAO,EAAoC,MAAM,oBAAoB,CAAC;AAE/E,OAAO,EAAE,mBAAmB,EAAwB,MAAM,sBAAsB,CAAC;AACjF,OAAO,EAAE,YAAY,EAAqB,MAAM,aAAa,CAAC;AAE9D,MAAM,CAAC,MAAM,mBAAmB,GAAG,CAAC,CAAC;AACrC,MAAM,uBAAuB,GAAG,EAAE,CAAC;AACnC,MAAM,iBAAiB,GAAG,MAAM,CAAC;AAyEjC,MAAM,UAAU,SAAS,CAAC,GAAc;IACtC,OAAO,EAAE,MAAM,EAAE,GAAG,CAAC,MAAM,EAAE,MAAM,EAAE,GAAG,CAAC,MAAM,EAAE,MAAM,EAAE,GAAG,CAAC,MAAM,EAAE,OAAO,EAAE,GAAG,CAAC,OAAO,EAAE,QAAQ,EAAE,GAAG,CAAC,QAAQ,EAAE,CAAC;AACtH,CAAC;AAED,MAAM,GAAG,GAAG,CAAC,YAAoB,EAAE,IAAY,EAAE,EAAE,CAAC,GAAG,YAAY,SAAS,IAAI,EAAE,CAAC;AAEnF,SAAS,WAAW,CAAC,MAAwB;IAC3C,OAAO,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,EAAE,OAAO,EAAE,CAAC,CAAC,OAAO,EAAE,KAAK,EAAE,CAAC,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,EAAE,GAAG,CAAC,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;AAC5F,CAAC;AAED,SAAS,aAAa,CAAC,GAAc;IACnC,OAAO,IAAI,GAAG,CAAC,GAAG,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,KAAK,EAAE,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC;AAClE,CAAC;AAED,SAAS,WAAW,CAAC,MAAwB,EAAE,GAAc;IAC3D,MAAM,QAAQ,GAAG,aAAa,CAAC,GAAG,CAAC,CAAC;IACpC,MAAM,OAAO,GAAkB,EAAE,CAAC;IAClC,MAAM,CAAC,OAAO,CAAC,CAAC,KAAK,EAAE,UAAU,EAAE,EAAE,CACnC,KAAK,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,QAAQ,EAAE,SAAS,EAAE,EAAE;QAC1C,MAAM,OAAO,GAAG,QAAQ,CAAC,GAAG,CAAC,GAAG,CAAC,KAAK,CAAC,OAAO,CAAC,OAAO,EAAE,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC;QACxE,IAAI,QAAQ,CAAC,UAAU,IAAI,OAAO,EAAE,MAAM,KAAK,QAAQ;YAAE,OAAO;QAChE,OAAO,CAAC,IAAI,CAAC,EAAE,UAAU,EAAE,SAAS,EAAE,OAAO,EAAE,GAAG,mBAAmB,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;IACpF,CAAC,CAAC,CACH,CAAC;IACF,OAAO,OAAO,CAAC;AACjB,CAAC;AAED,SAAS,SAAS,CAAC,IAAY;IAC7B,OAAO,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC;AACpE,CAAC;AAED,SAAS,eAAe,CAAC,MAAwB;IAC/C,OAAO,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC;AACrF,CAAC;AAED,SAAS,MAAM,CAAC,IAAY;IAC1B,OAAO,UAAU,CAAC,QAAQ,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;AACzD,CAAC;AAED;;;GAGG;AACH,MAAM,CAAC,KAAK,UAAU,WAAW,CAAC,OAAsB;IACtD,IAAI,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,QAAQ,CAAC,KAAK,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,UAAU,CAAC,EAAE,CAAC;QACxE,MAAM,IAAI,KAAK,CAAC,uDAAuD,CAAC,CAAC;IAC3E,CAAC;IACD,MAAM,WAAW,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,GAAG,CAAC,OAAO,CAAC,WAAW,IAAI,mBAAmB,EAAE,mBAAmB,CAAC,CAAC,CAAC;IAC3G,MAAM,QAAQ,GACZ,OAAO,CAAC,QAAQ,IAAI,CAAC,CAAC,QAAgB,EAAE,EAAE,CAAC,OAAO,CAAC,EAAE,QAAQ,EAAE,UAAU,EAAE,OAAO,CAAC,UAAU,EAAE,YAAY,EAAE,OAAO,CAAC,YAAY,EAAE,CAAC,CAAC,CAAC;IACxI,MAAM,cAAc,GAAG,MAAM,QAAQ,CAAC,OAAO,CAAC,UAAU,EAAE,MAAM,CAAC,CAAC;IAClE,MAAM,YAAY,GAAG,MAAM,CAAC,cAAc,CAAC,CAAC;IAE5C,MAAM,MAAM,GAAG,CAAC,MAAwB,EAAE,EAAE,CAC1C,cAAc,CAAC;QACb,UAAU,EAAE,OAAO,CAAC,QAAQ,CAAC,IAAI;QACjC,QAAQ,EAAE,OAAO,CAAC,QAAQ;QAC1B,MAAM;QACN,SAAS,EAAE,OAAO,CAAC,SAAS;KAC7B,CAAC,CAAC;IAEL,IAAI,MAAM,GAAG,WAAW,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;IACzC,IAAI,OAAO,GAAG,MAAM,CAAC,MAAM,CAAC,CAAC;IAC7B,IAAI,GAAG,GAAG,OAAO,CAAC,UAAU,CAAC;IAC7B,MAAM,QAAQ,GAAoB,EAAE,CAAC;IACrC,MAAM,KAAK,GAAe,EAAE,WAAW,EAAE,CAAC,EAAE,YAAY,EAAE,CAAC,EAAE,CAAC;IAC9D,IAAI,UAAkC,CAAC;IAEvC,OAAO,CAAC,UAAU,IAAI,QAAQ,CAAC,MAAM,GAAG,WAAW,EAAE,CAAC;QACpD,IAAI,GAAG,CAAC,KAAK,EAAE,CAAC;YACd,UAAU,GAAG,cAAc,CAAC;YAC5B,MAAM;QACR,CAAC;QAED,MAAM,OAAO,GAAG,WAAW,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;QACzC,IAAI,OAAO,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACzB,UAAU,GAAG,aAAa,CAAC;YAC3B,MAAM;QACR,CAAC;QAED,MAAM,OAAO,GAAG,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,uBAAuB,CAAC,CAAC;QACtF,IAAI,OAAO,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACzB,UAAU,GAAG,cAAc,CAAC;YAC5B,MAAM;QACR,CAAC;QAED,MAAM,aAAa,GAAmB,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE;YACtD,MAAM,KAAK,GAAG,MAAM,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC;YACnC,MAAM,QAAQ,GAAG,KAAK,CAAC,KAAK,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC;YAC1C,OAAO;gBACL,OAAO,EAAE,KAAK,CAAC,OAAO;gBACtB,QAAQ,EAAE,QAAQ,CAAC,IAAI;gBACvB,IAAI,EAAE,QAAQ,CAAC,IAAI;gBACnB,eAAe,EAAE,CAAC,CAAC,QAAQ;gBAC3B,OAAO,EAAE,CAAC,CAAC,OAAO,CAAC,OAAO,EAAE,OAAO,IAAI,EAAE;aAC1C,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,MAAM,MAAM,GAAG,iBAAiB,CAAC;YAC/B,QAAQ,EAAE,OAAO,CAAC,QAAQ;YAC1B,IAAI,EAAE,OAAO,CAAC,YAAY;YAC1B,OAAO,EAAE,aAAa;YACtB,OAAO,EAAE,QAAQ,CAAC,MAAM,GAAG,CAAC;YAC5B,WAAW,EAAE,OAAO,CAAC,WAAW;SACjC,CAAC,CAAC;QACH,IAAI,OAAO,CAAC,MAAM,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,MAAM,CAAC,EAAE,WAAW,EAAE,YAAY,CAAC,oBAAoB,EAAE,MAAM,CAAC,EAAE,YAAY,EAAE,iBAAiB,EAAE,CAAC,EAAE,CAAC;YAC3I,UAAU,GAAG,QAAQ,CAAC;YACtB,MAAM;QACR,CAAC;QAED,MAAM,OAAO,GAAkB;YAC7B,OAAO,EAAE,QAAQ,CAAC,MAAM,GAAG,CAAC;YAC5B,MAAM,EAAE,SAAS,CAAC,GAAG,CAAC;YACtB,SAAS,EAAE,OAAO,CAAC,MAAM;YACzB,OAAO,EAAE,EAAE;YACX,OAAO,EAAE,EAAE;YACX,SAAS,EAAE,KAAK;YAChB,KAAK,EAAE,EAAE,WAAW,EAAE,CAAC,EAAE,YAAY,EAAE,CAAC,EAAE;SAC3C,CAAC;QACF,QAAQ,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QAEvB,MAAM,QAAQ,GAAG,MAAM,OAAO,CAAC,QAAQ,CAAC,kBAAkB,CAAC;YACzD,MAAM,EAAE,oBAAoB;YAC5B,MAAM;YACN,MAAM,EAAE,YAAY;YACpB,UAAU,EAAE,oBAAoB;YAChC,eAAe,EAAE,iBAAiB;SACnC,CAAC,CAAC;QACH,OAAO,CAAC,MAAM,EAAE,GAAG,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC;QACpC,OAAO,CAAC,KAAK,GAAG,QAAQ,CAAC,KAAK,CAAC;QAC/B,KAAK,CAAC,WAAW,IAAI,QAAQ,CAAC,KAAK,CAAC,WAAW,CAAC;QAChD,KAAK,CAAC,YAAY,IAAI,QAAQ,CAAC,KAAK,CAAC,YAAY,CAAC;QAElD,MAAM,SAAS,GAAG,IAAI,GAAG,CAAC,QAAQ,CAAC,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,YAAY,EAAE,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC;QAC9F,MAAM,SAAS,GAAG,WAAW,CAAC,MAAM,CAAC,CAAC;QACtC,MAAM,QAAQ,GAAqD,EAAE,CAAC;QAEtE,KAAK,MAAM,MAAM,IAAI,OAAO,EAAE,CAAC;YAC7B,MAAM,KAAK,GAAG,SAAS,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC;YAC3C,MAAM,QAAQ,GAAG,KAAK,CAAC,KAAK,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC;YAC/C,MAAM,YAAY,GAAG,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC;YAC3C,MAAM,QAAQ,GAAG,SAAS,CAAC,GAAG,CAAC,GAAG,CAAC,YAAY,EAAE,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC;YACjE,MAAM,OAAO,GAAG,CAAC,MAAc,EAAE,EAAE,CAAC,OAAO,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,YAAY,EAAE,IAAI,EAAE,QAAQ,CAAC,IAAI,EAAE,MAAM,EAAE,CAAC,CAAC;YAExG,IAAI,CAAC,QAAQ,EAAE,CAAC;gBACd,OAAO,CAAC,oBAAoB,CAAC,CAAC;gBAC9B,SAAS;YACX,CAAC;YAED,IAAI,QAAQ,CAAC,MAAM,KAAK,KAAK,EAAE,CAAC;gBAC9B,MAAM,KAAK,GAAG,SAAS,CAAC,QAAQ,CAAC,IAAI,EAAE,OAAO,CAAC,SAAS,EAAE,IAAI,GAAG,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC;gBACzG,IAAI,OAAO,IAAI,KAAK,EAAE,CAAC;oBACrB,OAAO,CAAC,iBAAiB,KAAK,CAAC,KAAK,EAAE,CAAC,CAAC;oBACxC,SAAS;gBACX,CAAC;gBACD,IAAI,QAAQ,CAAC,IAAI,CAAC,IAAI,EAAE,KAAK,QAAQ,CAAC,IAAI,CAAC,IAAI,EAAE,EAAE,CAAC;oBAClD,OAAO,CAAC,4BAA4B,CAAC,CAAC;oBACtC,SAAS;gBACX,CAAC;gBACD,QAAQ,CAAC,IAAI,GAAG,QAAQ,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC;gBACrC,QAAQ,CAAC,KAAK,GAAG,KAAK,CAAC,KAAK,CAAC;YAC/B,CAAC;iBAAM,IAAI,QAAQ,CAAC,MAAM,KAAK,MAAM,EAAE,CAAC;gBACtC,QAAQ,CAAC,UAAU,GAAG,gCAAgC,QAAQ,CAAC,MAAM,EAAE,CAAC;YAC1E,CAAC;iBAAM,CAAC;gBACN,QAAQ,CAAC,IAAI,CAAC,EAAE,UAAU,EAAE,MAAM,CAAC,UAAU,EAAE,SAAS,EAAE,MAAM,CAAC,SAAS,EAAE,CAAC,CAAC;YAChF,CAAC;YACD,OAAO,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,YAAY,EAAE,IAAI,EAAE,QAAQ,CAAC,IAAI,EAAE,MAAM,EAAE,QAAQ,CAAC,MAAM,EAAE,MAAM,EAAE,QAAQ,CAAC,MAAM,EAAE,CAAC,CAAC;QAChH,CAAC;QAED,KAAK,MAAM,EAAE,UAAU,EAAE,SAAS,EAAE,IAAI,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,SAAS,GAAG,CAAC,CAAC,SAAS,CAAC,EAAE,CAAC;YAC3F,SAAS,CAAC,UAAU,CAAC,CAAC,KAAK,CAAC,MAAM,CAAC,SAAS,EAAE,CAAC,CAAC,CAAC;QACnD,CAAC;QACD,MAAM,QAAQ,GAAG,SAAS,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;QAE7D,IAAI,OAAO,CAAC,OAAO,CAAC,MAAM,KAAK,CAAC,IAAI,eAAe,CAAC,QAAQ,CAAC,KAAK,CAAC,EAAE,CAAC;YACpE,UAAU,GAAG,kBAAkB,CAAC;YAChC,MAAM;QACR,CAAC;QAED,MAAM,gBAAgB,GAAG,MAAM,CAAC,QAAQ,CAAC,CAAC;QAC1C,MAAM,SAAS,CAAC,OAAO,CAAC,QAAQ,EAAE,gBAAgB,EAAE,MAAM,CAAC,CAAC;QAC5D,MAAM,YAAY,GAAG,MAAM,QAAQ,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;QACtD,OAAO,CAAC,KAAK,GAAG,SAAS,CAAC,YAAY,CAAC,CAAC;QAExC,MAAM,MAAM,GAAG,aAAa,CAAC,GAAG,CAAC,CAAC;QAClC,MAAM,KAAK,GAAG,aAAa,CAAC,YAAY,CAAC,CAAC;QAC1C,MAAM,iBAAiB,GAAG,CAAC,GAAG,MAAM,CAAC,CAAC,IAAI,CACxC,CAAC,CAAC,OAAO,EAAE,OAAO,CAAC,EAAE,EAAE,CAAC,OAAO,CAAC,MAAM,KAAK,QAAQ,IAAI,KAAK,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,MAAM,KAAK,QAAQ,CAC/F,CAAC;QAEF,IAAI,YAAY,CAAC,KAAK,IAAI,iBAAiB,EAAE,CAAC;YAC5C,OAAO,CAAC,SAAS,GAAG,IAAI,CAAC;YACzB,MAAM,SAAS,CAAC,OAAO,CAAC,QAAQ,EAAE,OAAO,EAAE,MAAM,CAAC,CAAC;YACnD,UAAU,GAAG,WAAW,CAAC;YACzB,MAAM;QACR,CAAC;QAED,MAAM,GAAG,QAAQ,CAAC;QAClB,OAAO,GAAG,gBAAgB,CAAC;QAC3B,GAAG,GAAG,YAAY,CAAC;IACrB,CAAC;IAED,MAAM,MAAM,GAAmB,EAAE,CAAC;IAClC,IAAI,CAAC,GAAG,CAAC,KAAK,EAAE,CAAC;QACf,MAAM,SAAS,GAAG,WAAW,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;QAC3C,IAAI,SAAS,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACzB,KAAK,MAAM,CAAC,IAAI,SAAS,EAAE,CAAC;gBAC1B,MAAM,QAAQ,GAAG,MAAM,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC;gBACzD,MAAM,MAAM,GACV,CAAC,CAAC,QAAQ,KAAK,MAAM;oBACnB,CAAC,CAAC,sGAAsG;oBACxG,CAAC,CAAC,+BAA+B,CAAC,CAAC,QAAQ,MAAM,SAAS,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,EAAE,OAAO,IAAI,EAAE,CAAC,EAAE,CAAC;gBACnG,QAAQ,CAAC,UAAU,GAAG,MAAM,CAAC;gBAC7B,MAAM,CAAC,IAAI,CAAC,EAAE,YAAY,EAAE,MAAM,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,OAAO,CAAC,OAAO,EAAE,IAAI,EAAE,QAAQ,CAAC,IAAI,EAAE,MAAM,EAAE,SAAS,EAAE,MAAM,EAAE,CAAC,CAAC;YACtH,CAAC;YACD,OAAO,GAAG,MAAM,CAAC,MAAM,CAAC,CAAC;YACzB,MAAM,SAAS,CAAC,OAAO,CAAC,QAAQ,EAAE,OAAO,EAAE,MAAM,CAAC,CAAC;YACnD,GAAG,GAAG,MAAM,QAAQ,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;QACzC,CAAC;IACH,CAAC;IAED,UAAU,KAAK,WAAW,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC,MAAM,IAAI,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,cAAc,CAAC,CAAC,CAAC,aAAa,CAAC;IAEjG,MAAM,aAAa,GAAG,MAAM,QAAQ,CAAC,OAAO,CAAC,UAAU,EAAE,MAAM,CAAC,CAAC;IACjE,MAAM,eAAe,GAAG,MAAM,CAAC,aAAa,CAAC,KAAK,YAAY,CAAC;IAC/D,IAAI,CAAC,eAAe;QAAE,MAAM,SAAS,CAAC,OAAO,CAAC,UAAU,EAAE,cAAc,EAAE,MAAM,CAAC,CAAC;IAElF,OAAO;QACL,MAAM;QACN,OAAO;QACP,GAAG;QACH,OAAO,EAAE,SAAS,CAAC,OAAO,CAAC,UAAU,CAAC;QACtC,KAAK,EAAE,SAAS,CAAC,GAAG,CAAC;QACrB,QAAQ;QACR,MAAM;QACN,UAAU;QACV,eAAe;QACf,KAAK;KACN,CAAC;AACJ,CAAC"}

package/dist/repair/schema.d.ts

@@ -0,0 +1,17 @@
+import { z } from "zod";
+export declare const REPAIR_ACTIONS: readonly ["fix", "skip", "remove"];
+export declare const RepairSchema: z.ZodObject<{
+    repairs: z.ZodArray<z.ZodObject<{
+        functionName: z.ZodString;
+        name: z.ZodString;
+        action: z.ZodEnum<{
+            fix: "fix";
+            remove: "remove";
+            skip: "skip";
+        }>;
+        body: z.ZodString;
+        reason: z.ZodString;
+    }, z.core.$strip>>;
+}, z.core.$strip>;
+export type RepairResponse = z.infer<typeof RepairSchema>;
+export type RepairAction = (typeof REPAIR_ACTIONS)[number];

package/dist/repair/schema.js

@@ -0,0 +1,14 @@
+import { z } from "zod";
+export const REPAIR_ACTIONS = ["fix", "skip", "remove"];
+export const RepairSchema = z.object({
+    repairs: z.array(z.object({
+        functionName: z.string().describe("Binding name of the function, exactly as given"),
+        name: z.string().describe("Test name, exactly as given"),
+        action: z
+            .enum(REPAIR_ACTIONS)
+            .describe("fix: rewrite the body; skip: the test is right and the source looks buggy; remove: the test cannot be made reliable"),
+        body: z.string().describe("The corrected test body for fix; an empty string otherwise"),
+        reason: z.string().describe("One sentence explaining the failure and the change"),
+    })),
+});
+//# sourceMappingURL=schema.js.map

package/dist/repair/schema.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"schema.js","sourceRoot":"","sources":["../../src/repair/schema.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AAExB,MAAM,CAAC,MAAM,cAAc,GAAG,CAAC,KAAK,EAAE,MAAM,EAAE,QAAQ,CAAU,CAAC;AAEjE,MAAM,CAAC,MAAM,YAAY,GAAG,CAAC,CAAC,MAAM,CAAC;IACnC,OAAO,EAAE,CAAC,CAAC,KAAK,CACd,CAAC,CAAC,MAAM,CAAC;QACP,YAAY,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,gDAAgD,CAAC;QACnF,IAAI,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,6BAA6B,CAAC;QACxD,MAAM,EAAE,CAAC;aACN,IAAI,CAAC,cAAc,CAAC;aACpB,QAAQ,CAAC,qHAAqH,CAAC;QAClI,IAAI,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,4DAA4D,CAAC;QACvF,MAAM,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,oDAAoD,CAAC;KAClF,CAAC,CACH;CACF,CAAC,CAAC"}

package/dist/runner/coverage.d.ts

@@ -0,0 +1,5 @@
+import type { CoverageMetrics } from "./types.js";
+/** Resolves symlinks and Windows short names (`/var` → `/private/var`, `RUNNER~1` → `runneradmin`). */
+export declare function realPath(file: string): string;
+/** Reads coverage for one source file from Jest's `coverage-summary.json`. */
+export declare function readCoverage(summaryPath: string, sourcePath: string): Promise<CoverageMetrics | undefined>;

package/dist/runner/coverage.js

@@ -0,0 +1,42 @@
+import { realpathSync } from "node:fs";
+import { readFile } from "node:fs/promises";
+import path from "node:path";
+function percent(metric) {
+    if (!metric || metric.total === 0 || typeof metric.pct !== "number")
+        return null;
+    return metric.pct;
+}
+/** Resolves symlinks and Windows short names (`/var` → `/private/var`, `RUNNER~1` → `runneradmin`). */
+export function realPath(file) {
+    try {
+        return realpathSync.native(path.resolve(file));
+    }
+    catch {
+        return path.resolve(file);
+    }
+}
+function samePath(a, b) {
+    const normalize = (p) => (process.platform === "win32" ? realPath(p).toLowerCase() : realPath(p));
+    return normalize(a) === normalize(b);
+}
+/** Reads coverage for one source file from Jest's `coverage-summary.json`. */
+export async function readCoverage(summaryPath, sourcePath) {
+    let summary;
+    try {
+        summary = JSON.parse(await readFile(summaryPath, "utf8"));
+    }
+    catch {
+        return undefined;
+    }
+    const key = Object.keys(summary).find((file) => file !== "total" && samePath(file, sourcePath));
+    if (!key)
+        return undefined;
+    const entry = summary[key];
+    return {
+        lines: percent(entry.lines),
+        statements: percent(entry.statements),
+        functions: percent(entry.functions),
+        branches: percent(entry.branches),
+    };
+}
+//# sourceMappingURL=coverage.js.map

package/dist/runner/coverage.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"coverage.js","sourceRoot":"","sources":["../../src/runner/coverage.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,EAAE,MAAM,SAAS,CAAC;AACvC,OAAO,EAAE,QAAQ,EAAE,MAAM,kBAAkB,CAAC;AAC5C,OAAO,IAAI,MAAM,WAAW,CAAC;AAU7B,SAAS,OAAO,CAAC,MAA0B;IACzC,IAAI,CAAC,MAAM,IAAI,MAAM,CAAC,KAAK,KAAK,CAAC,IAAI,OAAO,MAAM,CAAC,GAAG,KAAK,QAAQ;QAAE,OAAO,IAAI,CAAC;IACjF,OAAO,MAAM,CAAC,GAAG,CAAC;AACpB,CAAC;AAED,uGAAuG;AACvG,MAAM,UAAU,QAAQ,CAAC,IAAY;IACnC,IAAI,CAAC;QACH,OAAO,YAAY,CAAC,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC;IACjD,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;IAC5B,CAAC;AACH,CAAC;AAED,SAAS,QAAQ,CAAC,CAAS,EAAE,CAAS;IACpC,MAAM,SAAS,GAAG,CAAC,CAAS,EAAE,EAAE,CAAC,CAAC,OAAO,CAAC,QAAQ,KAAK,OAAO,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,WAAW,EAAE,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC;IAC1G,OAAO,SAAS,CAAC,CAAC,CAAC,KAAK,SAAS,CAAC,CAAC,CAAC,CAAC;AACvC,CAAC;AAED,8EAA8E;AAC9E,MAAM,CAAC,KAAK,UAAU,YAAY,CAAC,WAAmB,EAAE,UAAkB;IACxE,IAAI,OAAwB,CAAC;IAC7B,IAAI,CAAC;QACH,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC,MAAM,QAAQ,CAAC,WAAW,EAAE,MAAM,CAAC,CAAoB,CAAC;IAC/E,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,MAAM,GAAG,GAAG,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,KAAK,OAAO,IAAI,QAAQ,CAAC,IAAI,EAAE,UAAU,CAAC,CAAC,CAAC;IAChG,IAAI,CAAC,GAAG;QAAE,OAAO,SAAS,CAAC;IAE3B,MAAM,KAAK,GAAG,OAAO,CAAC,GAAG,CAAC,CAAC;IAC3B,OAAO;QACL,KAAK,EAAE,OAAO,CAAC,KAAK,CAAC,KAAK,CAAC;QAC3B,UAAU,EAAE,OAAO,CAAC,KAAK,CAAC,UAAU,CAAC;QACrC,SAAS,EAAE,OAAO,CAAC,KAAK,CAAC,SAAS,CAAC;QACnC,QAAQ,EAAE,OAAO,CAAC,KAAK,CAAC,QAAQ,CAAC;KAClC,CAAC;AACJ,CAAC"}

package/dist/runner/index.d.ts

@@ -0,0 +1,5 @@
+export { runJest, sanitizedEnv, DEFAULT_RUN_TIMEOUT_MS, type RunOptions } from "./jestRunner.js";
+export { detectJestSetup, findProjectRoot } from "./project.js";
+export { detectFramework, detectVitestSetup } from "./vitest.js";
+export { NETWORK_BLOCKED_CODE, NETWORK_BLOCKED_MESSAGE } from "./networkGuard.js";
+export type { CoverageMetrics, InfraFailure, RunResult, TestOutcome } from "./types.js";

package/dist/runner/index.js

@@ -0,0 +1,5 @@
+export { runJest, sanitizedEnv, DEFAULT_RUN_TIMEOUT_MS } from "./jestRunner.js";
+export { detectJestSetup, findProjectRoot } from "./project.js";
+export { detectFramework, detectVitestSetup } from "./vitest.js";
+export { NETWORK_BLOCKED_CODE, NETWORK_BLOCKED_MESSAGE } from "./networkGuard.js";
+//# sourceMappingURL=index.js.map

package/dist/runner/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/runner/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,OAAO,EAAE,YAAY,EAAE,sBAAsB,EAAmB,MAAM,iBAAiB,CAAC;AACjG,OAAO,EAAE,eAAe,EAAE,eAAe,EAAE,MAAM,cAAc,CAAC;AAChE,OAAO,EAAE,eAAe,EAAE,iBAAiB,EAAE,MAAM,aAAa,CAAC;AACjE,OAAO,EAAE,oBAAoB,EAAE,uBAAuB,EAAE,MAAM,mBAAmB,CAAC"}

package/dist/runner/jestConfig.d.ts

@@ -0,0 +1,11 @@
+import type { JestSetup } from "./project.js";
+export interface RunFiles {
+    testPath: string;
+    sourcePath: string;
+    coverageDir: string;
+}
+/** Matches exactly this test file by its full path, on any OS path separator. */
+export declare function testRegexFor(testPath: string): string;
+export declare function defaultJestConfig(root: string, files: RunFiles): Record<string, unknown>;
+/** Writes the Jest config for one run into `runDir` and returns its path. */
+export declare function writeRunConfig(setup: JestSetup, files: RunFiles, runDir: string): Promise<string>;

package/dist/runner/jestConfig.js

@@ -0,0 +1,103 @@
+import { readFile, writeFile } from "node:fs/promises";
+import { createRequire } from "node:module";
+import path from "node:path";
+import { pathToFileURL } from "node:url";
+const require = createRequire(import.meta.url);
+function escapeRegex(text) {
+    return text.replace(/[.*+?^${}()|[\]\\]/g, "\\$&");
+}
+/** Matches exactly this test file by its full path, on any OS path separator. */
+export function testRegexFor(testPath) {
+    return `^${testPath.split(/[\\/]/).map(escapeRegex).join("[\\\\/]")}$`;
+}
+function coverageGlob(rootDir, sourcePath) {
+    return path.relative(rootDir, sourcePath).replaceAll("\\", "/");
+}
+export function defaultJestConfig(root, files) {
+    const babelOptions = {
+        babelrc: false,
+        configFile: false,
+        presets: [require.resolve("@babel/preset-typescript"), [require.resolve("@babel/preset-react"), { runtime: "automatic" }]],
+        plugins: [require.resolve("@babel/plugin-transform-modules-commonjs")],
+    };
+    return {
+        rootDir: root,
+        roots: [...new Set([path.dirname(files.testPath), path.dirname(files.sourcePath)])],
+        testEnvironment: "node",
+        testRegex: testRegexFor(files.testPath),
+        transform: { "\\.[cm]?[jt]sx?$": [require.resolve("babel-jest"), babelOptions] },
+        moduleFileExtensions: ["ts", "tsx", "mts", "cts", "js", "jsx", "mjs", "cjs", "json", "node"],
+        moduleNameMapper: { "^(\\.{1,2}/.*)\\.js$": "$1" },
+        collectCoverageFrom: [coverageGlob(root, files.sourcePath)],
+        coverageProvider: "v8",
+        coverageReporters: ["json-summary"],
+        coverageDirectory: files.coverageDir,
+    };
+}
+/** Loads the project's own config at run time and overrides only test selection and coverage. */
+function wrapperModule(baseConfigExpression, baseDir, files) {
+    const overrides = {
+        testRegex: testRegexFor(files.testPath),
+        coverageProvider: "v8",
+        coverageReporters: ["json-summary"],
+        coverageDirectory: files.coverageDir,
+    };
+    return `import path from "node:path";
+
+const baseDir = ${JSON.stringify(baseDir)};
+const sourcePath = ${JSON.stringify(files.sourcePath)};
+const testPath = ${JSON.stringify(files.testPath)};
+
+const ignores = (pattern, rootDir) => {
+  try {
+    // Jest rewrites "/" in these patterns to the OS separator, so check both spellings of the path.
+    const regex = new RegExp(pattern.replaceAll("<rootDir>", rootDir));
+    return regex.test(testPath) || regex.test(testPath.split(path.sep).join("/"));
+  } catch {
+    return false;
+  }
+};
+
+export default async () => {
+  let config = ${baseConfigExpression};
+  if (typeof config === "function") config = await config();
+  config = { ...config };
+  if (config.projects) throw new Error("Assertloom: Jest \`projects\` configurations are not supported yet.");
+  delete config.testMatch;
+  const rootDir = config.rootDir ? path.resolve(baseDir, config.rootDir) : baseDir;
+  if (Array.isArray(config.roots)) config.roots = [...config.roots, path.dirname(testPath)];
+  if (Array.isArray(config.testPathIgnorePatterns)) {
+    config.testPathIgnorePatterns = config.testPathIgnorePatterns.filter((p) => !ignores(p, rootDir));
+  }
+  return {
+    ...config,
+    rootDir,
+    ...${JSON.stringify(overrides)},
+    collectCoverageFrom: [path.relative(rootDir, sourcePath).split(path.sep).join("/")],
+  };
+};
+`;
+}
+/** Writes the Jest config for one run into `runDir` and returns its path. */
+export async function writeRunConfig(setup, files, runDir) {
+    if (setup.config.kind === "default") {
+        const file = path.join(runDir, "jest.config.json");
+        await writeFile(file, JSON.stringify(defaultJestConfig(setup.root, files), null, 2));
+        return file;
+    }
+    const baseDir = setup.config.kind === "file" ? path.dirname(setup.config.path) : setup.root;
+    let baseConfigExpression;
+    if (setup.config.kind === "package-json") {
+        baseConfigExpression = JSON.stringify(setup.config.config);
+    }
+    else if (setup.config.path.endsWith(".json")) {
+        baseConfigExpression = JSON.stringify(JSON.parse(await readFile(setup.config.path, "utf8")));
+    }
+    else {
+        baseConfigExpression = `(await import(${JSON.stringify(pathToFileURL(setup.config.path).href)})).default`;
+    }
+    const file = path.join(runDir, "jest.config.mjs");
+    await writeFile(file, wrapperModule(baseConfigExpression, baseDir, files));
+    return file;
+}
+//# sourceMappingURL=jestConfig.js.map

package/dist/runner/jestConfig.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"jestConfig.js","sourceRoot":"","sources":["../../src/runner/jestConfig.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,SAAS,EAAE,MAAM,kBAAkB,CAAC;AACvD,OAAO,EAAE,aAAa,EAAE,MAAM,aAAa,CAAC;AAC5C,OAAO,IAAI,MAAM,WAAW,CAAC;AAC7B,OAAO,EAAE,aAAa,EAAE,MAAM,UAAU,CAAC;AAGzC,MAAM,OAAO,GAAG,aAAa,CAAC,OAAO,IAAI,CAAC,GAAG,CAAC,CAAC;AAQ/C,SAAS,WAAW,CAAC,IAAY;IAC/B,OAAO,IAAI,CAAC,OAAO,CAAC,qBAAqB,EAAE,MAAM,CAAC,CAAC;AACrD,CAAC;AAED,iFAAiF;AACjF,MAAM,UAAU,YAAY,CAAC,QAAgB;IAC3C,OAAO,IAAI,QAAQ,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,GAAG,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC;AACzE,CAAC;AAED,SAAS,YAAY,CAAC,OAAe,EAAE,UAAkB;IACvD,OAAO,IAAI,CAAC,QAAQ,CAAC,OAAO,EAAE,UAAU,CAAC,CAAC,UAAU,CAAC,IAAI,EAAE,GAAG,CAAC,CAAC;AAClE,CAAC;AAED,MAAM,UAAU,iBAAiB,CAAC,IAAY,EAAE,KAAe;IAC7D,MAAM,YAAY,GAAG;QACnB,OAAO,EAAE,KAAK;QACd,UAAU,EAAE,KAAK;QACjB,OAAO,EAAE,CAAC,OAAO,CAAC,OAAO,CAAC,0BAA0B,CAAC,EAAE,CAAC,OAAO,CAAC,OAAO,CAAC,qBAAqB,CAAC,EAAE,EAAE,OAAO,EAAE,WAAW,EAAE,CAAC,CAAC;QAC1H,OAAO,EAAE,CAAC,OAAO,CAAC,OAAO,CAAC,0CAA0C,CAAC,CAAC;KACvE,CAAC;IAEF,OAAO;QACL,OAAO,EAAE,IAAI;QACb,KAAK,EAAE,CAAC,GAAG,IAAI,GAAG,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,QAAQ,CAAC,EAAE,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC;QACnF,eAAe,EAAE,MAAM;QACvB,SAAS,EAAE,YAAY,CAAC,KAAK,CAAC,QAAQ,CAAC;QACvC,SAAS,EAAE,EAAE,kBAAkB,EAAE,CAAC,OAAO,CAAC,OAAO,CAAC,YAAY,CAAC,EAAE,YAAY,CAAC,EAAE;QAChF,oBAAoB,EAAE,CAAC,IAAI,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,IAAI,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,MAAM,EAAE,MAAM,CAAC;QAC5F,gBAAgB,EAAE,EAAE,sBAAsB,EAAE,IAAI,EAAE;QAClD,mBAAmB,EAAE,CAAC,YAAY,CAAC,IAAI,EAAE,KAAK,CAAC,UAAU,CAAC,CAAC;QAC3D,gBAAgB,EAAE,IAAI;QACtB,iBAAiB,EAAE,CAAC,cAAc,CAAC;QACnC,iBAAiB,EAAE,KAAK,CAAC,WAAW;KACrC,CAAC;AACJ,CAAC;AAED,iGAAiG;AACjG,SAAS,aAAa,CAAC,oBAA4B,EAAE,OAAe,EAAE,KAAe;IACnF,MAAM,SAAS,GAAG;QAChB,SAAS,EAAE,YAAY,CAAC,KAAK,CAAC,QAAQ,CAAC;QACvC,gBAAgB,EAAE,IAAI;QACtB,iBAAiB,EAAE,CAAC,cAAc,CAAC;QACnC,iBAAiB,EAAE,KAAK,CAAC,WAAW;KACrC,CAAC;IAEF,OAAO;;kBAES,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC;qBACpB,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,UAAU,CAAC;mBAClC,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,QAAQ,CAAC;;;;;;;;;;;;;iBAahC,oBAAoB;;;;;;;;;;;;;SAa5B,IAAI,CAAC,SAAS,CAAC,SAAS,CAAC;;;;CAIjC,CAAC;AACF,CAAC;AAED,6EAA6E;AAC7E,MAAM,CAAC,KAAK,UAAU,cAAc,CAAC,KAAgB,EAAE,KAAe,EAAE,MAAc;IACpF,IAAI,KAAK,CAAC,MAAM,CAAC,IAAI,KAAK,SAAS,EAAE,CAAC;QACpC,MAAM,IAAI,GAAG,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,kBAAkB,CAAC,CAAC;QACnD,MAAM,SAAS,CAAC,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,iBAAiB,CAAC,KAAK,CAAC,IAAI,EAAE,KAAK,CAAC,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC;QACrF,OAAO,IAAI,CAAC;IACd,CAAC;IAED,MAAM,OAAO,GAAG,KAAK,CAAC,MAAM,CAAC,IAAI,KAAK,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,IAAI,CAAC;IAC5F,IAAI,oBAA4B,CAAC;IACjC,IAAI,KAAK,CAAC,MAAM,CAAC,IAAI,KAAK,cAAc,EAAE,CAAC;QACzC,oBAAoB,GAAG,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;IAC7D,CAAC;SAAM,IAAI,KAAK,CAAC,MAAM,CAAC,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC,EAAE,CAAC;QAC/C,oBAAoB,GAAG,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM,QAAQ,CAAC,KAAK,CAAC,MAAM,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC,CAAC,CAAC;IAC/F,CAAC;SAAM,CAAC;QACN,oBAAoB,GAAG,iBAAiB,IAAI,CAAC,SAAS,CAAC,aAAa,CAAC,KAAK,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC;IAC5G,CAAC;IAED,MAAM,IAAI,GAAG,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,iBAAiB,CAAC,CAAC;IAClD,MAAM,SAAS,CAAC,IAAI,EAAE,aAAa,CAAC,oBAAoB,EAAE,OAAO,EAAE,KAAK,CAAC,CAAC,CAAC;IAC3E,OAAO,IAAI,CAAC;AACd,CAAC"}

package/dist/runner/jestRunner.d.ts

@@ -0,0 +1,17 @@
+import type { RunResult } from "./types.js";
+import type { TestFramework } from "../types/index.js";
+export declare const DEFAULT_RUN_TIMEOUT_MS = 120000;
+export declare const DEFAULT_TEST_TIMEOUT_MS = 5000;
+export interface RunOptions {
+    testPath: string;
+    sourcePath: string;
+    timeoutMs?: number;
+    testTimeoutMs?: number;
+    /** Let tests open network connections. Off by default. */
+    allowNetwork?: boolean;
+    /** Default: jest. */
+    framework?: TestFramework;
+}
+/** Environment for untrusted test code: drops anything that looks like a credential. */
+export declare function sanitizedEnv(env?: NodeJS.ProcessEnv): NodeJS.ProcessEnv;
+export declare function runJest(options: RunOptions): Promise<RunResult>;

package/dist/runner/jestRunner.js

@@ -0,0 +1,154 @@
+import { spawn } from "node:child_process";
+import { mkdtemp, readFile, rm, writeFile } from "node:fs/promises";
+import os from "node:os";
+import path from "node:path";
+import { readCoverage, realPath } from "./coverage.js";
+import { writeRunConfig } from "./jestConfig.js";
+import { writeNetworkGuard } from "./networkGuard.js";
+import { detectJestSetup, jestCanStartIn } from "./project.js";
+import { cleanMessage, parseJestReport } from "./results.js";
+import { describeVitestSetup, detectVitestSetup, vitestArgs, writeVitestRunConfig } from "./vitest.js";
+export const DEFAULT_RUN_TIMEOUT_MS = 120_000;
+export const DEFAULT_TEST_TIMEOUT_MS = 5_000;
+const OUTPUT_TAIL_CHARS = 8_000;
+const SECRET_ENV = /KEY|TOKEN|SECRET|PASSWORD|PASSWD|CREDENTIAL|AUTH|SESSION|COOKIE/i;
+/** Environment for untrusted test code: drops anything that looks like a credential. */
+export function sanitizedEnv(env = process.env) {
+    const clean = {};
+    for (const [name, value] of Object.entries(env)) {
+        if (!SECRET_ENV.test(name))
+            clean[name] = value;
+    }
+    return { ...clean, NODE_ENV: "test", CI: "true", FORCE_COLOR: "0" };
+}
+function describeSetup(setup) {
+    const jest = setup.jestSource === "project" ? "project Jest" : "bundled Jest";
+    const config = setup.config.kind === "default"
+        ? "Assertloom default config"
+        : setup.config.kind === "file"
+            ? path.basename(setup.config.path)
+            : `"jest" in package.json`;
+    return `${jest}, ${config}`;
+}
+function runProcess(command, args, options) {
+    return new Promise((resolve, reject) => {
+        const child = spawn(command, args, { cwd: options.cwd, env: options.env, windowsHide: true });
+        let output = "";
+        let timedOut = false;
+        const append = (chunk) => {
+            output = (output + chunk).slice(-OUTPUT_TAIL_CHARS);
+        };
+        child.stdout.setEncoding("utf8");
+        child.stderr.setEncoding("utf8");
+        child.stdout.on("data", append);
+        child.stderr.on("data", append);
+        const timer = setTimeout(() => {
+            timedOut = true;
+            child.kill("SIGKILL");
+        }, options.timeoutMs);
+        child.on("error", (error) => {
+            clearTimeout(timer);
+            reject(error);
+        });
+        child.on("close", (code) => {
+            clearTimeout(timer);
+            resolve({ code, timedOut, output });
+        });
+    });
+}
+export async function runJest(options) {
+    // Jest reports real paths, so configure it with them too.
+    const testPath = realPath(options.testPath);
+    const sourcePath = realPath(options.sourcePath);
+    const timeoutMs = options.timeoutMs ?? DEFAULT_RUN_TIMEOUT_MS;
+    const testTimeoutMs = options.testTimeoutMs ?? DEFAULT_TEST_TIMEOUT_MS;
+    const vitest = options.framework === "vitest" ? detectVitestSetup(sourcePath) : undefined;
+    const jest = vitest ? undefined : detectJestSetup(sourcePath);
+    const originalSource = await readFile(sourcePath);
+    const runDir = await mkdtemp(path.join(os.tmpdir(), "assertloom-run-"));
+    const started = Date.now();
+    const network = options.allowNetwork ? "network allowed" : "network blocked";
+    const described = vitest ? describeVitestSetup(vitest) : describeSetup(jest);
+    const base = { tests: [], passed: 0, failed: 0, skipped: 0, total: 0, setup: `${described}, ${network}` };
+    let result;
+    try {
+        result = await execute();
+        return result;
+    }
+    finally {
+        await rm(runDir, { recursive: true, force: true });
+        const current = await readFile(sourcePath).catch(() => undefined);
+        if (!current?.equals(originalSource)) {
+            await writeFile(sourcePath, originalSource);
+            if (result)
+                result.sourceRestored = true;
+        }
+    }
+    async function execute() {
+        const coverageDir = path.join(runDir, "coverage");
+        const resultsPath = path.join(runDir, "results.json");
+        const files = { testPath, sourcePath, coverageDir };
+        const env = sanitizedEnv();
+        if (!options.allowNetwork) {
+            env.NODE_OPTIONS = [env.NODE_OPTIONS, await writeNetworkGuard(runDir)].filter(Boolean).join(" ");
+        }
+        let args;
+        let cwd;
+        if (vitest) {
+            args = vitestArgs(vitest, await writeVitestRunConfig(vitest, { ...files, resultsPath, testTimeoutMs }, runDir));
+            cwd = vitest.root;
+        }
+        else {
+            const setup = jest;
+            args = [
+                setup.jestBin,
+                "--config",
+                await writeRunConfig(setup, files, runDir),
+                "--runInBand",
+                "--ci",
+                "--no-watchman",
+                "--json",
+                "--outputFile",
+                resultsPath,
+                "--coverage",
+                // Open handles left by the code under test would otherwise keep Jest alive until the timeout.
+                "--forceExit",
+                `--testTimeout=${testTimeoutMs}`,
+            ];
+            // runDir holds the generated config, so Jest can always find one there.
+            cwd = jestCanStartIn(setup.root) ? setup.root : runDir;
+        }
+        const { timedOut, output } = await runProcess(process.execPath, args, { cwd, timeoutMs, env });
+        const durationMs = Date.now() - started;
+        let report;
+        try {
+            report = JSON.parse(await readFile(resultsPath, "utf8"));
+        }
+        catch {
+            if (timedOut) {
+                return { ...base, status: "error", durationMs, infra: { kind: "timeout", message: `Jest did not finish within ${timeoutMs / 1000}s.` } };
+            }
+            return { ...base, status: "error", durationMs, infra: { kind: "crash", message: cleanMessage(output) || "Jest exited without writing results." } };
+        }
+        const { tests, infra } = parseJestReport(report);
+        const passed = tests.filter((t) => t.status === "passed").length;
+        const failed = tests.filter((t) => t.status === "failed").length;
+        const skipped = tests.filter((t) => t.status === "skipped").length;
+        if (!infra && tests.length === 0) {
+            return { ...base, status: "error", durationMs, infra: { kind: "suite-error", message: cleanMessage(output) || "Jest found no tests to run." } };
+        }
+        return {
+            ...base,
+            status: infra ? "error" : failed > 0 ? "failed" : "passed",
+            tests,
+            passed,
+            failed,
+            skipped,
+            total: tests.length,
+            infra,
+            coverage: infra ? undefined : await readCoverage(path.join(coverageDir, "coverage-summary.json"), sourcePath),
+            durationMs,
+        };
+    }
+}
+//# sourceMappingURL=jestRunner.js.map