chore: import upstream snapshot with attribution

2026-07-13 12:09:03 +08:00
commit 2de7548470
2883 changed files with 374366 additions and 0 deletions
@@ -0,0 +1,41 @@
+import { strict as assert } from "node:assert";
+import { test } from "node:test";
+import { BUDGET_USD, MAX_TURNS, chargeTurn, checkBudget } from "../src/cost.js";
+import { defaultSeed } from "../src/migrations.js";
+
+test("checkBudget returns clean when fresh", () => {
+  const m = defaultSeed()[0]!;
+  const v = checkBudget(m);
+  assert.equal(v.exhausted, false);
+});
+
+test("checkBudget flags turns exhausted", () => {
+  const m = defaultSeed()[0]!;
+  m.turns = MAX_TURNS;
+  const v = checkBudget(m);
+  assert.equal(v.exhausted, true);
+  assert.equal(v.reason, "turns");
+});
+
+test("checkBudget flags cost exhausted", () => {
+  const m = defaultSeed()[0]!;
+  m.spentUsd = BUDGET_USD;
+  const v = checkBudget(m);
+  assert.equal(v.exhausted, true);
+  assert.equal(v.reason, "cost");
+});
+
+test("chargeTurn increments turns and adds cost", () => {
+  const m = defaultSeed()[0]!;
+  chargeTurn(m, () => 0.5);
+  assert.equal(m.turns, 1);
+  assert.ok(m.spentUsd > 0);
+  assert.ok(m.spentUsd < BUDGET_USD);
+});
+
+test("chargeTurn upper bound stays inside budget per turn", () => {
+  const m = defaultSeed()[0]!;
+  for (let i = 0; i < MAX_TURNS; i++) chargeTurn(m, () => 1);
+  assert.equal(m.turns, MAX_TURNS);
+  assert.ok(m.spentUsd <= BUDGET_USD, `spent ${m.spentUsd} exceeds budget ${BUDGET_USD}`);
+});
@@ -0,0 +1,59 @@
+import { strict as assert } from "node:assert";
+import { test } from "node:test";
+import {
+  advanceFile,
+  defaultSeed,
+  fileDiff,
+  migrationDone,
+  rolledUpStats,
+  tickOne,
+} from "../src/migrations.js";
+
+test("seed produces three running migrations", () => {
+  const migrations = defaultSeed();
+  assert.equal(migrations.length, 3);
+  for (const m of migrations) {
+    assert.equal(m.state, "running");
+    assert.ok(m.files.length > 0);
+  }
+});
+
+test("advanceFile walks queued to rewriting to building to passed", () => {
+  const f = fileDiff("foo.java", "openrewrite");
+  const noFail = () => 0.99;
+  advanceFile(f, noFail);
+  assert.equal(f.status, "rewriting");
+  advanceFile(f, noFail);
+  assert.equal(f.status, "building");
+  advanceFile(f, noFail);
+  assert.equal(f.status, "passed");
+});
+
+test("advanceFile is a no-op on terminal states", () => {
+  const f = fileDiff("foo.java", "openrewrite");
+  f.status = "passed";
+  advanceFile(f);
+  assert.equal(f.status, "passed");
+  f.status = "failed";
+  advanceFile(f);
+  assert.equal(f.status, "failed");
+});
+
+test("tickOne can move a migration to passed when all files pass", () => {
+  const m = defaultSeed()[0]!;
+  const det = () => 0.99;
+  for (let i = 0; i < 200; i++) tickOne(m, det);
+  assert.equal(migrationDone(m), true);
+  assert.ok(m.state === "passed" || m.state === "failed");
+});
+
+test("rolledUpStats counts states correctly", () => {
+  const m = defaultSeed();
+  m[0]!.state = "passed";
+  m[1]!.state = "failed";
+  const stats = rolledUpStats(m);
+  assert.equal(stats.passed, 1);
+  assert.equal(stats.failed, 1);
+  assert.equal(stats.running, 1);
+  assert.equal(stats.total, 3);
+});