paperclip/ui/src/pages/StatusCards/format.test.ts

142 lines
5.8 KiB
TypeScript

import { describe, expect, it } from "vitest";
import type { StatusCardUpdate } from "@paperclipai/shared";
import {
estimateStatusCardCost,
rollupUpdates,
rollupUpdatesToday,
} from "./format";
function update(overrides: Partial<StatusCardUpdate>): StatusCardUpdate {
return {
id: "00000000-0000-0000-0000-000000000000",
cardId: "00000000-0000-0000-0000-000000000001",
kind: "full",
trigger: "manual",
generationIssueId: null,
runId: null,
changes: [],
inputTokens: 1000,
outputTokens: 500,
costCents: 2,
model: null,
queryVersion: 1,
changeSummary: null,
startedAt: NOW.toISOString(),
finishedAt: null,
status: "ok",
error: null,
...overrides,
};
}
// A fixed instant rather than the real clock, so these fixtures mean the same
// thing on every machine and at every hour. `rollupUpdatesToday` filters on the
// *UTC* day boundary, and building from `new Date()` fails against that in two
// separate ways — see `iso` below. The function takes `now` for exactly this
// reason, and the sibling test at the bottom of this file already passes one.
const NOW = new Date("2026-07-23T12:00:00.000Z");
function iso(daysAgo: number): string {
const d = new Date(NOW);
d.setUTCDate(d.getUTCDate() - daysAgo);
// Noon UTC, so a row is unambiguously inside the UTC day it belongs to.
//
// Built on the *local* day instead, `iso(0)` misses in both directions. West
// of UTC it lands in the previous UTC day for the stretch between UTC
// midnight and local midnight — about seven hours a day at UTC-7 — and east
// of UTC+12 "today at local noon" is already yesterday in UTC outright.
// Either way today's rows drop out of a filter that means to count them. A
// run crossing midnight UTC splits the same way, with `iso(0)` and the
// function's default `now` landing on different days.
d.setUTCHours(12, 0, 0, 0);
return d.toISOString();
}
describe("rollupUpdates (lifetime)", () => {
it("sums the whole ledger and excludes compile rows from the update count", () => {
const rollup = rollupUpdates([
update({ kind: "full", inputTokens: 3000, outputTokens: 1000, costCents: 3, startedAt: iso(0) }),
update({ kind: "compile", inputTokens: 500, outputTokens: 100, costCents: 1, startedAt: iso(0) }),
update({ kind: "incremental", inputTokens: 1000, outputTokens: 200, costCents: 2, startedAt: iso(5) }),
]);
// full + incremental = 2 updates (compile excluded from count)
expect(rollup.updateCount).toBe(2);
// tokens/cost still include the compile spend
expect(rollup.totalTokens).toBe(3000 + 1000 + 500 + 100 + 1000 + 200);
expect(rollup.totalCostCents).toBe(3 + 1 + 2);
});
});
describe("rollupUpdatesToday", () => {
it("only counts updates started today and drops older ledger rows", () => {
const rollup = rollupUpdatesToday([
update({ kind: "full", inputTokens: 2000, outputTokens: 600, costCents: 3, startedAt: iso(0) }),
update({ kind: "compile", inputTokens: 400, outputTokens: 100, costCents: 1, startedAt: iso(0) }),
// yesterday + last week — must not be counted as "today"
update({ kind: "full", inputTokens: 9999, outputTokens: 9999, costCents: 99, startedAt: iso(1) }),
update({ kind: "incremental", inputTokens: 9999, outputTokens: 9999, costCents: 99, startedAt: iso(7) }),
], NOW);
// Only today's full rebuild counts as an update (compile excluded).
expect(rollup.updateCount).toBe(1);
// Today's tokens/cost include today's compile but not older days.
expect(rollup.totalTokens).toBe(2000 + 600 + 400 + 100);
expect(rollup.totalCostCents).toBe(3 + 1);
});
it("uses the UTC day boundary used by the server token cap", () => {
const now = new Date("2026-07-23T01:00:00.000Z");
const rollup = rollupUpdatesToday([
update({ startedAt: "2026-07-23T00:30:00.000Z", inputTokens: 200, outputTokens: 50 }),
update({ startedAt: "2026-07-22T23:30:00.000Z", inputTokens: 900, outputTokens: 100 }),
], now);
expect(rollup.updateCount).toBe(1);
expect(rollup.totalTokens).toBe(250);
});
});
describe("estimateStatusCardCost", () => {
it("manual mode = a single per-refresh rebuild", () => {
const est = estimateStatusCardCost({ mode: "manual", triggers: {} as never });
expect(est.primary).toContain("per refresh");
expect(est.note).toMatch(/only cost/i);
});
it("interval mode scales with the interval and reacts to the daily token cap", () => {
const uncapped = estimateStatusCardCost({ mode: "interval", intervalMinutes: 60, triggers: {} as never });
// 24h / 60min = 24 updates/day upper bound
expect(uncapped.primary).toContain("Up to ~24 updates/day");
expect(uncapped.primary).toContain("every 60 min");
const capped = estimateStatusCardCost({
mode: "interval",
intervalMinutes: 60,
dailyTokenCap: 10_000,
triggers: {} as never,
});
// 10_000 / 2_000 est tokens = 5 updates before the cap bites
expect(capped.primary).toContain("Up to ~5 updates/day");
expect(capped.note).toMatch(/daily token cap/i);
});
it("reactive mode reports the per-hour ceiling", () => {
const est = estimateStatusCardCost({ mode: "reactive", maxUpdatesPerHour: 4, debounceSeconds: 60, triggers: {} as never });
expect(est.primary).toContain("up to 4/hour");
// 24h * 4/hr = 96 updates/day upper bound
expect(est.primary).toContain("Up to ~96 updates/day");
});
it("interval mode honours the active-hours window", () => {
const est = estimateStatusCardCost({
mode: "interval",
intervalMinutes: 60,
activeHours: { start: "08:00", end: "20:00", timezone: "UTC" },
triggers: {} as never,
});
// 12h window / 60min = 12 updates/day
expect(est.primary).toContain("Up to ~12 updates/day");
expect(est.primary).toContain("during active hours");
});
});