chore: import upstream snapshot with attribution
This commit is contained in:
@@ -0,0 +1,110 @@
|
||||
/**
|
||||
* tests/unit/ui/comboFlowModel-breakers.test.ts
|
||||
*
|
||||
* TDD for U1b — enrichRunWithBreakers: overlays REAL circuit-breaker state
|
||||
* (from GET /api/monitoring/health → providerHealth[provider]) onto a combo run's
|
||||
* targets, so the cascade can show "CB: OPEN · retry 41s" instead of only the
|
||||
* error-string heuristic.
|
||||
* Run: node --import tsx/esm --test tests/unit/ui/comboFlowModel-breakers.test.ts
|
||||
*/
|
||||
|
||||
import { describe, it } from "node:test";
|
||||
import assert from "node:assert/strict";
|
||||
|
||||
import {
|
||||
enrichRunWithBreakers,
|
||||
comboRunToFlow,
|
||||
type ComboRunModel,
|
||||
type TargetNodeModel,
|
||||
} from "../../../src/app/(dashboard)/dashboard/combos/live/comboFlowModel.ts";
|
||||
|
||||
function mkRun(
|
||||
targets: Array<Partial<TargetNodeModel> & { provider: string }>
|
||||
): ComboRunModel {
|
||||
return {
|
||||
comboName: "c",
|
||||
strategy: "priority",
|
||||
outcome: "running",
|
||||
startedAt: 0,
|
||||
targets: targets.map((t, i) => ({
|
||||
targetIndex: i,
|
||||
provider: t.provider,
|
||||
model: t.model ?? "m",
|
||||
state: t.state ?? "idle",
|
||||
...t,
|
||||
})),
|
||||
};
|
||||
}
|
||||
|
||||
describe("enrichRunWithBreakers", () => {
|
||||
it("returns null for a null run", () => {
|
||||
assert.equal(enrichRunWithBreakers(null, {}), null);
|
||||
});
|
||||
|
||||
it("returns the same run reference when no health map is provided", () => {
|
||||
const run = mkRun([{ provider: "openai" }]);
|
||||
assert.equal(enrichRunWithBreakers(run, null), run);
|
||||
assert.equal(enrichRunWithBreakers(run, undefined), run);
|
||||
});
|
||||
|
||||
it("attaches cbState + retryAfterMs for an OPEN provider breaker only", () => {
|
||||
const run = mkRun([{ provider: "openai" }, { provider: "anthropic" }]);
|
||||
const out = enrichRunWithBreakers(run, {
|
||||
openai: { state: "OPEN", retryAfterMs: 41000 },
|
||||
});
|
||||
assert.ok(out);
|
||||
assert.equal(out.targets[0].cbState, "OPEN");
|
||||
assert.equal(out.targets[0].cbRetryAfterMs, 41000);
|
||||
assert.equal(out.targets[1].cbState, undefined, "healthy/absent provider gets no badge");
|
||||
});
|
||||
|
||||
it("does not attach a badge for a CLOSED (healthy) breaker", () => {
|
||||
const run = mkRun([{ provider: "openai" }]);
|
||||
const out = enrichRunWithBreakers(run, { openai: { state: "CLOSED", retryAfterMs: 0 } });
|
||||
assert.equal(out?.targets[0].cbState, undefined);
|
||||
});
|
||||
|
||||
it("surfaces HALF_OPEN and DEGRADED states (case-insensitive)", () => {
|
||||
const run = mkRun([{ provider: "a" }, { provider: "b" }]);
|
||||
const out = enrichRunWithBreakers(run, {
|
||||
a: { state: "half_open", retryAfterMs: 5000 },
|
||||
b: { state: "DEGRADED" },
|
||||
});
|
||||
assert.equal(out?.targets[0].cbState, "HALF_OPEN");
|
||||
assert.equal(out?.targets[1].cbState, "DEGRADED");
|
||||
});
|
||||
|
||||
it("does not mutate the input run (pure)", () => {
|
||||
const run = mkRun([{ provider: "openai" }]);
|
||||
enrichRunWithBreakers(run, { openai: { state: "OPEN", retryAfterMs: 1 } });
|
||||
assert.equal(run.targets[0].cbState, undefined, "original run must be untouched");
|
||||
});
|
||||
|
||||
it("strips a stale cbState when the breaker recovers to CLOSED", () => {
|
||||
const run = mkRun([{ provider: "openai", cbState: "OPEN", cbRetryAfterMs: 9 }]);
|
||||
const out = enrichRunWithBreakers(run, { openai: { state: "CLOSED" } });
|
||||
assert.equal(out?.targets[0].cbState, undefined);
|
||||
assert.equal(out?.targets[0].cbRetryAfterMs, undefined);
|
||||
});
|
||||
|
||||
it("ignores unknown breaker state strings", () => {
|
||||
const run = mkRun([{ provider: "openai" }]);
|
||||
const out = enrichRunWithBreakers(run, { openai: { state: "WEIRD" } });
|
||||
assert.equal(out?.targets[0].cbState, undefined);
|
||||
});
|
||||
|
||||
it("comboRunToFlow carries cbState/cbRetryAfterMs into the target node data", () => {
|
||||
const run = mkRun([{ provider: "openai" }, { provider: "anthropic" }]);
|
||||
const enriched = enrichRunWithBreakers(run, {
|
||||
openai: { state: "OPEN", retryAfterMs: 41000 },
|
||||
});
|
||||
const { nodes } = comboRunToFlow(enriched as ComboRunModel);
|
||||
|
||||
const target0 = nodes.find((n) => n.id === "target-0");
|
||||
assert.equal(target0?.data.cbState, "OPEN");
|
||||
assert.equal(target0?.data.cbRetryAfterMs, 41000);
|
||||
|
||||
const target1 = nodes.find((n) => n.id === "target-1");
|
||||
assert.equal(target1?.data.cbState, undefined, "healthy provider node has no cbState");
|
||||
});
|
||||
});
|
||||
Reference in New Issue
Block a user