@aliou/pi-synthetic 0.17.3 → 0.18.0

package/src/services/quota-store.test.ts

@@ -1,211 +0,0 @@
-import {
-  afterEach,
-  assert,
-  beforeEach,
-  describe,
-  expect,
-  it,
-  vi,
-} from "vitest";
-import type { QuotasResponse } from "../types/quotas";
-import { QuotaStore } from "./quota-store";
-
-beforeEach(() => {
-  vi.useFakeTimers();
-});
-
-afterEach(() => {
-  vi.useRealTimers();
-});
-
-describe("QuotaStore", () => {
-  const sampleQuotas: QuotasResponse = {
-    subscription: { limit: 100, requests: 5, renewsAt: "2026-01-01T00:00:00Z" },
-  };
-
-  describe("ingest", () => {
-    it("stores and emits API-sourced data", () => {
-      const store = new QuotaStore();
-      const received: QuotasResponse[] = [];
-      store.subscribe((snap) => received.push(snap.quotas));
-
-      const result = store.ingest(sampleQuotas, "api");
-
-      expect(result).toBe(true);
-      expect(store.getSnapshot()?.quotas).toBe(sampleQuotas);
-      expect(store.getSnapshot()?.source).toBe("api");
-      expect(received).toHaveLength(1);
-    });
-
-    it("stores and emits header-sourced data", () => {
-      const store = new QuotaStore();
-      const result = store.ingest(sampleQuotas, "header");
-
-      expect(result).toBe(true);
-      expect(store.getSnapshot()?.source).toBe("header");
-    });
-
-    it("throttles header ingestion within throttle window", () => {
-      const store = new QuotaStore();
-      store.ingest(sampleQuotas, "header");
-
-      // Within throttle window — should be dropped
-      const result = store.ingest(sampleQuotas, "header");
-      expect(result).toBe(false);
-
-      // Advance past throttle
-      vi.advanceTimersByTime(store.headerThrottleMs + 1);
-      const result2 = store.ingest(sampleQuotas, "header");
-      expect(result2).toBe(true);
-    });
-
-    it("does NOT throttle API ingestion", () => {
-      const store = new QuotaStore();
-      store.ingest(sampleQuotas, "api");
-
-      // API is never throttled
-      const result = store.ingest(sampleQuotas, "api");
-      expect(result).toBe(true);
-    });
-
-    it("header after API emit always goes through", () => {
-      const store = new QuotaStore();
-      store.ingest(sampleQuotas, "api");
-
-      // Header 1ms after API should not be blocked
-      vi.advanceTimersByTime(1);
-      const result = store.ingest(sampleQuotas, "header");
-      expect(result).toBe(true);
-    });
-
-    it("updates timestamp on each successful ingest", () => {
-      const store = new QuotaStore();
-      store.ingest(sampleQuotas, "api");
-      const snap1 = store.getSnapshot();
-      assert(snap1);
-      const t1 = snap1.updatedAt;
-
-      vi.advanceTimersByTime(10_000);
-      store.ingest(sampleQuotas, "api");
-      const snap2 = store.getSnapshot();
-      assert(snap2);
-      const t2 = snap2.updatedAt;
-
-      expect(t2).toBeGreaterThan(t1);
-    });
-  });
-
-  describe("subscribe", () => {
-    it("notifies subscribers on ingest", () => {
-      const store = new QuotaStore();
-      const calls: QuotasResponse[] = [];
-      store.subscribe((snap) => calls.push(snap.quotas));
-
-      store.ingest(sampleQuotas, "api");
-      expect(calls).toHaveLength(1);
-      expect(calls[0]).toBe(sampleQuotas);
-    });
-
-    it("does not notify on throttled ingest", () => {
-      const store = new QuotaStore();
-      const calls: QuotasResponse[] = [];
-      store.subscribe((snap) => calls.push(snap.quotas));
-
-      store.ingest(sampleQuotas, "header");
-      store.ingest(sampleQuotas, "header"); // throttled
-
-      expect(calls).toHaveLength(1);
-    });
-
-    it("unsubscribes when unsubscribe function is called", () => {
-      const store = new QuotaStore();
-      const calls: QuotasResponse[] = [];
-      const unsub = store.subscribe((snap) => calls.push(snap.quotas));
-
-      unsub();
-      store.ingest(sampleQuotas, "api");
-
-      expect(calls).toHaveLength(0);
-    });
-
-    it("supports multiple subscribers", () => {
-      const store = new QuotaStore();
-      const calls1: QuotasResponse[] = [];
-      const calls2: QuotasResponse[] = [];
-      store.subscribe((snap) => calls1.push(snap.quotas));
-      store.subscribe((snap) => calls2.push(snap.quotas));
-
-      store.ingest(sampleQuotas, "api");
-
-      expect(calls1).toHaveLength(1);
-      expect(calls2).toHaveLength(1);
-    });
-  });
-
-  describe("refreshFromApi", () => {
-    it("calls the fetcher and ingests the result", async () => {
-      const store = new QuotaStore();
-      const fetcher = vi.fn().mockResolvedValue(sampleQuotas);
-
-      const result = await store.refreshFromApi(fetcher);
-
-      assert(result);
-      expect(result.quotas).toBe(sampleQuotas);
-      expect(result.source).toBe("api");
-      expect(fetcher).toHaveBeenCalledOnce();
-    });
-
-    it("deduplicates concurrent calls", async () => {
-      const store = new QuotaStore();
-      let resolveFirst!: (v: QuotasResponse) => void;
-      const first = new Promise<QuotasResponse>((r) => (resolveFirst = r));
-      const fetcher = vi.fn().mockImplementation(() => first);
-
-      // Start two concurrent refreshes
-      const p1 = store.refreshFromApi(fetcher);
-      const p2 = store.refreshFromApi(fetcher);
-
-      // Only one fetcher call
-      expect(fetcher).toHaveBeenCalledOnce();
-      expect(store.isRefreshing).toBe(true);
-
-      // Resolve the fetch
-      resolveFirst(sampleQuotas);
-      await p1;
-      await p2;
-
-      expect(store.isRefreshing).toBe(false);
-    });
-
-    it("handles fetcher returning undefined", async () => {
-      const store = new QuotaStore();
-      const fetcher = vi.fn().mockResolvedValue(undefined);
-
-      const result = await store.refreshFromApi(fetcher);
-
-      expect(result).toBeUndefined();
-      expect(store.getSnapshot()).toBeUndefined();
-    });
-  });
-
-  describe("clear", () => {
-    it("resets all state", () => {
-      const store = new QuotaStore();
-      store.ingest(sampleQuotas, "api");
-
-      store.clear();
-
-      expect(store.getSnapshot()).toBeUndefined();
-    });
-
-    it("resets header throttle after clear", () => {
-      const store = new QuotaStore();
-      store.ingest(sampleQuotas, "header");
-      expect(store.ingest(sampleQuotas, "header")).toBe(false);
-
-      store.clear();
-
-      expect(store.ingest(sampleQuotas, "header")).toBe(true);
-    });
-  });
-});

package/src/services/quota-warnings.test.ts

@@ -1,393 +0,0 @@
-import {
-  afterEach,
-  assert,
-  beforeEach,
-  describe,
-  expect,
-  it,
-  vi,
-} from "vitest";
-import type { QuotasResponse } from "../types/quotas";
-import { assessWindow, type QuotaWindow } from "../utils/quotas-severity";
-import { type NotifyFn, QuotaWarningNotifier } from "./quota-warnings";
-
-beforeEach(() => {
-  vi.useFakeTimers();
-});
-
-afterEach(() => {
-  vi.useRealTimers();
-});
-
-describe("QuotaWarningNotifier", () => {
-  const baseQuotas: QuotasResponse = {
-    weeklyTokenLimit: {
-      nextRegenAt: new Date(Date.now() + 6 * 24 * 3600 * 1000).toISOString(),
-      percentRemaining: 90,
-      maxCredits: "$10.00",
-      remainingCredits: "$9.00",
-      nextRegenCredits: "$0.50",
-    },
-    rollingFiveHourLimit: {
-      nextTickAt: new Date(Date.now() + 2.5 * 3600 * 1000).toISOString(),
-      tickPercent: 10,
-      remaining: 90,
-      max: 100,
-      limited: false,
-    },
-    search: {
-      hourly: {
-        limit: 100,
-        requests: 10,
-        renewsAt: new Date(Date.now() + 30 * 60 * 1000).toISOString(),
-      },
-    },
-    freeToolCalls: {
-      limit: 100,
-      requests: 5,
-      renewsAt: new Date(Date.now() + 12 * 3600 * 1000).toISOString(),
-    },
-  };
-
-  describe("shouldNotify", () => {
-    it("notifies on first time seeing a window at risk", () => {
-      const notifier = new QuotaWarningNotifier();
-      expect(notifier.shouldNotify("Credits / week", "warning")).toBe(true);
-      expect(notifier.shouldNotify("Requests / 5h", "high")).toBe(true);
-      expect(notifier.shouldNotify("Search / hour", "critical")).toBe(true);
-    });
-
-    it("notifies on severity escalation", () => {
-      const notifier = new QuotaWarningNotifier();
-      notifier.markNotified("Credits / week", "warning");
-      expect(notifier.shouldNotify("Credits / week", "high")).toBe(true);
-
-      notifier.markNotified("Requests / 5h", "high");
-      expect(notifier.shouldNotify("Requests / 5h", "critical")).toBe(true);
-    });
-
-    it("notifies on skip from none to any risk level", () => {
-      const notifier = new QuotaWarningNotifier();
-      notifier.markNotified("Test", "none");
-      expect(notifier.shouldNotify("Test", "warning")).toBe(true);
-    });
-
-    it("does not notify on same severity for warning within cooldown", () => {
-      const notifier = new QuotaWarningNotifier();
-      notifier.markNotified("Credits / week", "warning");
-      expect(notifier.shouldNotify("Credits / week", "warning")).toBe(false);
-    });
-
-    it("does notify on warning after cooldown elapsed", () => {
-      const notifier = new QuotaWarningNotifier();
-      notifier.markNotified("Credits / week", "warning");
-
-      vi.advanceTimersByTime(60 * 60 * 1000 + 1);
-
-      expect(notifier.shouldNotify("Credits / week", "warning")).toBe(true);
-    });
-
-    it("does not notify on downgrade to warning", () => {
-      const notifier = new QuotaWarningNotifier();
-      notifier.markNotified("Credits / week", "high");
-      expect(notifier.shouldNotify("Credits / week", "warning")).toBe(false);
-    });
-
-    it("does notify on downgrade to high (no cooldown)", () => {
-      const notifier = new QuotaWarningNotifier();
-      notifier.markNotified("Requests / 5h", "critical");
-      expect(notifier.shouldNotify("Requests / 5h", "high")).toBe(true);
-    });
-
-    it("always notifies for high severity (no cooldown)", () => {
-      const notifier = new QuotaWarningNotifier();
-      notifier.markNotified("Credits / week", "high");
-      expect(notifier.shouldNotify("Credits / week", "high")).toBe(true);
-    });
-
-    it("always notifies for critical severity (no cooldown)", () => {
-      const notifier = new QuotaWarningNotifier();
-      notifier.markNotified("Credits / week", "critical");
-      expect(notifier.shouldNotify("Credits / week", "critical")).toBe(true);
-    });
-  });
-
-  describe("markNotified", () => {
-    it("tracks severity per window key", () => {
-      const notifier = new QuotaWarningNotifier();
-      notifier.markNotified("Credits / week", "warning");
-      expect(notifier.shouldNotify("Credits / week", "warning")).toBe(false);
-
-      // Different key is independent
-      expect(notifier.shouldNotify("Requests / 5h", "warning")).toBe(true);
-    });
-
-    it("allows re-notification after escalation then downgrade then re-escalation", () => {
-      const notifier = new QuotaWarningNotifier();
-      notifier.markNotified("Test", "high");
-      expect(notifier.shouldNotify("Test", "warning")).toBe(false);
-      expect(notifier.shouldNotify("Test", "high")).toBe(true);
-    });
-  });
-
-  describe("clearAlertState", () => {
-    it("resets all alert state so windows notify again", () => {
-      const notifier = new QuotaWarningNotifier();
-      notifier.markNotified("Credits / week", "warning");
-      expect(notifier.shouldNotify("Credits / week", "warning")).toBe(false);
-
-      notifier.clearAlertState();
-
-      expect(notifier.shouldNotify("Credits / week", "warning")).toBe(true);
-    });
-  });
-
-  describe("findHighRiskWindows", () => {
-    it("returns empty for low-usage quotas", () => {
-      const notifier = new QuotaWarningNotifier();
-      const risks = notifier.findHighRiskWindows(baseQuotas);
-      expect(risks).toHaveLength(0);
-    });
-
-    it("finds windows with high usage", () => {
-      const notifier = new QuotaWarningNotifier();
-      const quotas: QuotasResponse = {
-        ...baseQuotas,
-        rollingFiveHourLimit: {
-          nextTickAt: new Date(Date.now() + 2.5 * 3600 * 1000).toISOString(),
-          tickPercent: 10,
-          remaining: 5,
-          max: 100,
-          limited: false,
-        },
-      };
-      const risks = notifier.findHighRiskWindows(quotas);
-      const fiveHourRisk = risks.find(
-        (r) => r.window.label === "Requests / 5h",
-      );
-      assert(fiveHourRisk, "fiveHourRisk should exist");
-      expect(fiveHourRisk.assessment.severity).toBe("high");
-    });
-
-    it("finds limited windows even with low usage", () => {
-      const notifier = new QuotaWarningNotifier();
-      const quotas: QuotasResponse = {
-        ...baseQuotas,
-        rollingFiveHourLimit: {
-          nextTickAt: new Date(Date.now() + 2.5 * 3600 * 1000).toISOString(),
-          tickPercent: 10,
-          remaining: 95,
-          max: 100,
-          limited: true,
-        },
-      };
-      const risks = notifier.findHighRiskWindows(quotas);
-      const fiveHourRisk = risks.find(
-        (r) => r.window.label === "Requests / 5h",
-      );
-      assert(fiveHourRisk, "fiveHourRisk should exist");
-      expect(fiveHourRisk.assessment.severity).toBe("critical");
-    });
-
-    it("returns empty for quotas with no windows", () => {
-      const notifier = new QuotaWarningNotifier();
-      expect(notifier.findHighRiskWindows({})).toHaveLength(0);
-    });
-  });
-
-  describe("formatWarningMessage", () => {
-    it("formats single window warning", () => {
-      const notifier = new QuotaWarningNotifier();
-      const w: QuotaWindow = {
-        label: "Requests / 5h",
-        usedPercent: 92,
-        resetsAt: new Date(Date.now() + 2 * 3600 * 1000),
-        windowSeconds: 5 * 3600,
-        usedValue: 92,
-        limitValue: 100,
-        showPace: false,
-      };
-      const assessment = assessWindow(w);
-      const msg = notifier.formatWarningMessage([{ window: w, assessment }]);
-      expect(msg).toContain("Synthetic quota warning:");
-      expect(msg).toContain("Requests / 5h");
-      expect(msg).toContain("92% used");
-      expect(msg).toContain("projected");
-    });
-
-    it("formats multiple windows", () => {
-      const notifier = new QuotaWarningNotifier();
-      const w1: QuotaWindow = {
-        label: "Credits / week",
-        usedPercent: 85,
-        resetsAt: new Date(Date.now() + 6 * 24 * 3600 * 1000),
-        windowSeconds: 7 * 24 * 3600,
-        usedValue: 85,
-        limitValue: 100,
-        showPace: true,
-        paceScale: 1 / 7,
-      };
-      const w2: QuotaWindow = {
-        label: "Requests / 5h",
-        usedPercent: 92,
-        resetsAt: new Date(Date.now() + 2 * 3600 * 1000),
-        windowSeconds: 5 * 3600,
-        usedValue: 92,
-        limitValue: 100,
-        showPace: false,
-      };
-      const msg = notifier.formatWarningMessage([
-        { window: w1, assessment: assessWindow(w1) },
-        { window: w2, assessment: assessWindow(w2) },
-      ]);
-      expect(msg).toContain("Credits / week");
-      expect(msg).toContain("Requests / 5h");
-      const lines = msg.split("\n");
-      expect(lines).toHaveLength(3); // header + 2 windows
-    });
-
-    it("includes severity label for non-none severities", () => {
-      const notifier = new QuotaWarningNotifier();
-      const w: QuotaWindow = {
-        label: "Requests / 5h",
-        usedPercent: 92,
-        resetsAt: new Date(Date.now() + 2 * 3600 * 1000),
-        windowSeconds: 5 * 3600,
-        usedValue: 92,
-        limitValue: 100,
-        showPace: false,
-      };
-      const msg = notifier.formatWarningMessage([
-        { window: w, assessment: assessWindow(w) },
-      ]);
-      expect(msg).toMatch(/\(high\)/);
-    });
-  });
-
-  describe("evaluate", () => {
-    it("does not notify for low-usage quotas", () => {
-      const notifier = new QuotaWarningNotifier();
-      const calls: Array<[string, string]> = [];
-      const notify: NotifyFn = (msg, lvl) => calls.push([msg, lvl]);
-
-      notifier.evaluate(baseQuotas, false, notify);
-      expect(calls).toHaveLength(0);
-    });
-
-    it("notifies for high-usage quotas with skipAlreadyWarned=false", () => {
-      const notifier = new QuotaWarningNotifier();
-      const calls: Array<[string, string]> = [];
-      const notify: NotifyFn = (msg, lvl) => calls.push([msg, lvl]);
-
-      const highUsageQuotas: QuotasResponse = {
-        ...baseQuotas,
-        rollingFiveHourLimit: {
-          nextTickAt: new Date(Date.now() + 2.5 * 3600 * 1000).toISOString(),
-          tickPercent: 10,
-          remaining: 5,
-          max: 100,
-          limited: false,
-        },
-      };
-
-      notifier.evaluate(highUsageQuotas, false, notify);
-      expect(calls).toHaveLength(1);
-      expect(calls[0][0]).toContain("Synthetic quota warning");
-    });
-
-    it("does not re-notify on same severity with skipAlreadyWarned=true", () => {
-      const notifier = new QuotaWarningNotifier();
-      const calls: Array<[string, string]> = [];
-      const notify: NotifyFn = (msg, lvl) => calls.push([msg, lvl]);
-
-      // 85% used (no pace) → warning severity, which has cooldown
-      const warningQuotas: QuotasResponse = {
-        ...baseQuotas,
-        rollingFiveHourLimit: {
-          nextTickAt: new Date(Date.now() + 2.5 * 3600 * 1000).toISOString(),
-          tickPercent: 10,
-          remaining: 15,
-          max: 100,
-          limited: false,
-        },
-      };
-
-      notifier.evaluate(warningQuotas, true, notify);
-      expect(calls).toHaveLength(1);
-
-      // Same severity, same data — should not re-notify (warning has cooldown)
-      notifier.evaluate(warningQuotas, true, notify);
-      expect(calls).toHaveLength(1);
-    });
-
-    it("notifies on severity escalation even with skipAlreadyWarned=true", () => {
-      const notifier = new QuotaWarningNotifier();
-      const calls: Array<[string, string]> = [];
-      const notify: NotifyFn = (msg, lvl) => calls.push([msg, lvl]);
-
-      // 92% used (no pace) → high severity
-      const highQuotas: QuotasResponse = {
-        ...baseQuotas,
-        rollingFiveHourLimit: {
-          nextTickAt: new Date(Date.now() + 2.5 * 3600 * 1000).toISOString(),
-          tickPercent: 10,
-          remaining: 8,
-          max: 100,
-          limited: false,
-        },
-      };
-
-      notifier.evaluate(highQuotas, true, notify);
-      expect(calls).toHaveLength(1);
-
-      // Escalate to critical (limited)
-      const criticalQuotas: QuotasResponse = {
-        ...baseQuotas,
-        rollingFiveHourLimit: {
-          nextTickAt: new Date(Date.now() + 2.5 * 3600 * 1000).toISOString(),
-          tickPercent: 10,
-          remaining: 2,
-          max: 100,
-          limited: true,
-        },
-      };
-
-      notifier.evaluate(criticalQuotas, true, notify);
-      expect(calls).toHaveLength(2);
-      expect(calls[1][1]).toBe("error");
-    });
-  });
-
-  describe("notification flow (shouldNotify + markNotified integration)", () => {
-    it("notifies once on first warning, blocks repeat, notifies on escalation", () => {
-      const notifier = new QuotaWarningNotifier();
-      expect(notifier.shouldNotify("Credits / week", "warning")).toBe(true);
-      notifier.markNotified("Credits / week", "warning");
-
-      expect(notifier.shouldNotify("Credits / week", "warning")).toBe(false);
-
-      expect(notifier.shouldNotify("Credits / week", "high")).toBe(true);
-      notifier.markNotified("Credits / week", "high");
-
-      expect(notifier.shouldNotify("Credits / week", "high")).toBe(true);
-    });
-
-    it("allows re-notification after clear", () => {
-      const notifier = new QuotaWarningNotifier();
-      expect(notifier.shouldNotify("Credits / week", "warning")).toBe(true);
-      notifier.markNotified("Credits / week", "warning");
-      expect(notifier.shouldNotify("Credits / week", "warning")).toBe(false);
-
-      notifier.clearAlertState();
-
-      expect(notifier.shouldNotify("Credits / week", "warning")).toBe(true);
-    });
-
-    it("tracks windows independently", () => {
-      const notifier = new QuotaWarningNotifier();
-      notifier.markNotified("Credits / week", "warning");
-      expect(notifier.shouldNotify("Credits / week", "warning")).toBe(false);
-      expect(notifier.shouldNotify("Search / hour", "warning")).toBe(true);
-    });
-  });
-});