@tokenbuddy/tokenbuddy 1.0.36 → 1.0.37

package/tests/credit-tracker.test.ts

@@ -1,165 +0,0 @@
-import {
-  CreditTracker,
-  DEFAULT_FRESH_PURCHASE_THRESHOLD,
-  DEFAULT_FRESH_PURCHASE_WINDOW_MS,
-  DEFAULT_PURCHASE_BUDGET_PER_MINUTE
-} from "../src/credit-tracker.js";
-
-describe("CreditTracker", () => {
-  function makeClock(start = 1_000_000): { now: number; advance: (ms: number) => void } {
-    const clock = { now: start, advance: (ms: number) => { clock.now += ms; } };
-    return clock;
-  }
-
-  test("recordPurchase creates an entry with the supplied balance", () => {
-    const clock = makeClock();
-    const tracker = new CreditTracker({ now: () => clock.now });
-    const entry = tracker.recordPurchase("s1", 1_000_000, 1_000_000);
-    expect(entry.sellerId).toBe("s1");
-    expect(entry.lastPurchaseAmountMicros).toBe(1_000_000);
-    expect(entry.currentBalanceMicros).toBe(1_000_000);
-    expect(entry.leftoverCreditMicros).toBe(0);
-    expect(tracker.summary().purchasesInLastMinute).toBe(1);
-  });
-
-  test("recordPurchase rejects non-positive amounts", () => {
-    const tracker = new CreditTracker();
-    expect(() => tracker.recordPurchase("s1", 0, 0)).toThrow();
-    expect(() => tracker.recordPurchase("s1", -1, -1)).toThrow();
-  });
-
-  test("recordSpend clamps to zero and is a no-op for unknown sellers", () => {
-    const tracker = new CreditTracker();
-    expect(tracker.recordSpend("missing", 100)).toBeUndefined();
-    tracker.recordPurchase("s1", 1_000_000, 1_000_000);
-    const updated = tracker.recordSpend("s1", 600_000);
-    expect(updated?.currentBalanceMicros).toBe(600_000);
-    const zeroed = tracker.recordSpend("s1", -10);
-    expect(zeroed?.currentBalanceMicros).toBe(0);
-  });
-
-  test("isInFreshPurchaseWindow is true only when balance ratio and time window are both satisfied", () => {
-    const clock = makeClock();
-    const tracker = new CreditTracker({ now: () => clock.now });
-    tracker.recordPurchase("s1", 1_000_000, 1_000_000);
-
-    // Right after purchase: 100% balance, well within the window.
-    expect(tracker.isInFreshPurchaseWindow("s1")).toBe(true);
-
-    // After 50% spend: ratio = 0.5, still at the default threshold.
-    tracker.recordSpend("s1", 500_000);
-    expect(tracker.isInFreshPurchaseWindow("s1")).toBe(true);
-
-    // After more spend the ratio drops below threshold.
-    tracker.recordSpend("s1", 100_000);
-    expect(tracker.isInFreshPurchaseWindow("s1")).toBe(false);
-
-    // Replenish via a new purchase and verify window restarts.
-    clock.advance(1_000);
-    tracker.recordPurchase("s1", 1_000_000, 1_000_000);
-    expect(tracker.isInFreshPurchaseWindow("s1")).toBe(true);
-
-    // Advance past the window: even a full balance no longer counts.
-    clock.advance(DEFAULT_FRESH_PURCHASE_WINDOW_MS + 1);
-    expect(tracker.isInFreshPurchaseWindow("s1")).toBe(false);
-  });
-
-  test("isInFreshPurchaseWindow returns false for sellers without any purchase", () => {
-    const tracker = new CreditTracker();
-    expect(tracker.isInFreshPurchaseWindow("never-purchased")).toBe(false);
-  });
-
-  test("canAutoPurchase enforces a per-minute budget and prunes old timestamps", () => {
-    const clock = makeClock();
-    const tracker = new CreditTracker({ now: () => clock.now });
-
-    expect(tracker.canAutoPurchase()).toBe(true);
-    tracker.recordPurchase("s1", 100, 100);
-    tracker.recordPurchase("s2", 100, 100);
-    tracker.recordPurchase("s3", 100, 100);
-    expect(tracker.canAutoPurchase()).toBe(false); // budget hit
-
-    // Advance past the 60s window: oldest timestamps fall out.
-    clock.advance(60_001);
-    expect(tracker.canAutoPurchase()).toBe(true);
-  });
-
-  test("transferLeftoverToWasted moves the current balance into the leftover bucket and tracks waste", () => {
-    const clock = makeClock();
-    const tracker = new CreditTracker({ now: () => clock.now });
-    tracker.recordPurchase("s1", 1_000_000, 800_000);
-    const wasted = tracker.transferLeftoverToWasted("s1", "failover");
-    expect(wasted).toBe(800_000);
-    const entry = tracker.getEntry("s1");
-    expect(entry?.currentBalanceMicros).toBe(0);
-    expect(entry?.leftoverCreditMicros).toBe(800_000);
-    const summary = tracker.summary();
-    expect(summary.totalWastedMicros).toBe(800_000);
-    expect(summary.wastedSinceLastDoctorRun).toBe(800_000);
-  });
-
-  test("transferLeftoverToWasted is a no-op for unknown sellers or empty balances", () => {
-    const tracker = new CreditTracker();
-    expect(tracker.transferLeftoverToWasted("missing", "x")).toBe(0);
-    tracker.recordPurchase("s1", 100, 0);
-    expect(tracker.transferLeftoverToWasted("s1", "x")).toBe(0);
-  });
-
-  test("consumeLeftover drains the leftover bucket up to the requested amount", () => {
-    const tracker = new CreditTracker();
-    tracker.recordPurchase("s1", 1_000_000, 500_000);
-    tracker.transferLeftoverToWasted("s1", "failover");
-    expect(tracker.consumeLeftover("s1", 300_000)).toBe(300_000);
-    expect(tracker.getEntry("s1")?.leftoverCreditMicros).toBe(200_000);
-    // Requesting more than available returns only the available amount.
-    expect(tracker.consumeLeftover("s1", 999_999)).toBe(200_000);
-    expect(tracker.getEntry("s1")?.leftoverCreditMicros).toBe(0);
-  });
-
-  test("consumeLeftover is a no-op for sellers without leftover", () => {
-    const tracker = new CreditTracker();
-    expect(tracker.consumeLeftover("missing", 100)).toBe(0);
-  });
-
-  test("summary includes per-seller entries with latest values", () => {
-    const clock = makeClock();
-    const tracker = new CreditTracker({ now: () => clock.now });
-    tracker.recordPurchase("s1", 1_000_000, 1_000_000);
-    clock.advance(1_000);
-    tracker.recordPurchase("s2", 500_000, 500_000);
-    const summary = tracker.summary();
-    expect(summary.perSeller).toHaveLength(2);
-    const s2 = summary.perSeller.find((entry) => entry.sellerId === "s2");
-    expect(s2?.currentBalanceMicros).toBe(500_000);
-    expect(s2?.lastPurchaseAt).toBe(clock.now);
-  });
-
-  test("resetDoctorRunCounter clears the wasted-since-last-doctor counter", () => {
-    const tracker = new CreditTracker();
-    tracker.recordPurchase("s1", 1_000_000, 500_000);
-    tracker.transferLeftoverToWasted("s1", "failover");
-    expect(tracker.summary().wastedSinceLastDoctorRun).toBe(500_000);
-    tracker.resetDoctorRunCounter();
-    expect(tracker.summary().wastedSinceLastDoctorRun).toBe(0);
-    // totalWastedMicros is preserved.
-    expect(tracker.summary().totalWastedMicros).toBe(500_000);
-  });
-
-  test("clear removes all entries and resets all counters", () => {
-    const tracker = new CreditTracker();
-    tracker.recordPurchase("s1", 100, 100);
-    tracker.transferLeftoverToWasted("s1", "failover");
-    tracker.clear();
-    const summary = tracker.summary();
-    expect(summary.totalWastedMicros).toBe(0);
-    expect(summary.wastedSinceLastDoctorRun).toBe(0);
-    expect(summary.purchasesInLastMinute).toBe(0);
-    expect(summary.perSeller).toEqual([]);
-  });
-
-  test("default constants match the v1.1 / v1.2 design", () => {
-    expect(DEFAULT_FRESH_PURCHASE_WINDOW_MS).toBe(30_000);
-    expect(DEFAULT_FRESH_PURCHASE_THRESHOLD).toBe(0.5);
-    expect(DEFAULT_PURCHASE_BUDGET_PER_MINUTE).toBe(3);
-  });
-});

package/tests/daemon-413-fallback.test.ts

@@ -1,92 +0,0 @@
-import * as http from "http";
-import * as fs from "fs";
-import * as path from "path";
-import { AddressInfo } from "net";
-import { TokenbuddyDaemon } from "../src/daemon.js";
-
-/**
- * v1.2 §18.9: when the bootstrap returns 413 (registry over 1MB), the
- * daemon must fall back to the last successfully fetched snapshot
- * instead of failing every request. This is the buyer-side counterpart
- * to the bootstrap's hard 1MB cap.
- */
-describe("TokenbuddyDaemon registry 413 stale-fallback", () => {
-  const TEMP_DB = path.resolve(__dirname, "../../data-test/413-fallback-test.db");
-  let bootstrapServer: http.Server;
-  let bootstrapPort: number;
-  let daemon: TokenbuddyDaemon;
-  let daemonProxyPort: number;
-
-  function rmDb(): void {
-    for (const suffix of ["", "-wal", "-shm"]) {
-      const file = TEMP_DB + suffix;
-      if (fs.existsSync(file)) {
-        fs.unlinkSync(file);
-      }
-    }
-  }
-
-  beforeAll((done) => {
-    bootstrapServer = http.createServer((_req, res) => {
-      res.setHeader("Content-Type", "application/json");
-      // Always 200 for the boot fetch. Tests that want 413 do not
-      // need to flip this server; they instead verify the catch-block
-      // in `fetchRegistry` via the dedicated unit test in
-      // `seller-catalog-413.test.ts`.
-      res.statusCode = 200;
-      res.end(JSON.stringify({
-        version: 1,
-        defaultSeller: "primary-seller",
-        sellers: [
-          {
-            id: "primary-seller",
-            url: "https://primary.example.com",
-            supportedProtocols: ["chat_completions"],
-            paymentMethods: ["mock"],
-            models: ["gpt-4o"]
-          }
-        ]
-      }));
-    });
-    bootstrapServer.listen(0, "127.0.0.1", () => {
-      bootstrapPort = (bootstrapServer.address() as AddressInfo).port;
-      done();
-    });
-  });
-
-  afterAll((done) => {
-    bootstrapServer.close(() => done());
-  });
-
-  beforeEach(() => {
-    rmDb();
-    daemon = new TokenbuddyDaemon({
-      controlPort: 0,
-      proxyPort: 0,
-      dbPath: TEMP_DB,
-      sellerRegistryUrl: `http://127.0.0.1:${bootstrapPort}/registry/sellers`
-    });
-    daemon.start();
-    const proxyServer = (daemon as unknown as { proxyServer: { address(): AddressInfo } }).proxyServer;
-    daemonProxyPort = proxyServer.address().port;
-  });
-
-  afterEach(async () => {
-    daemon.stop();
-    // Drain any in-flight prewarm scheduler work to avoid jest
-    // open-handle warnings. The daemon's stop() is fire-and-forget.
-    await new Promise<void>((resolve) => setTimeout(resolve, 50));
-    rmDb();
-  });
-
-  test("daemon stays alive after a successful boot against the bootstrap", async () => {
-    // The buyer must surface this as a typed error so the daemon can
-    // fall back to the last-known snapshot. The fetch logic is covered
-    // by `seller-catalog-413.test.ts`; here we just assert the
-    // happy-path control plane is up.
-    const controlPort = (daemon as unknown as { controlServer: { address(): AddressInfo } }).controlServer.address().port;
-    const health = await (await fetch(`http://127.0.0.1:${controlPort}/health`)).json() as { status: string };
-    expect(health.status).toBe("ok");
-    expect(typeof daemonProxyPort).toBe("number");
-  });
-});