Files
OmniRoute/tests/unit/usage-events.test.ts
Raxxoor 68b71bc826 fix(usage): normalize Antigravity and agy provider quotas (#3604)
Split-out PR B from #3584. Normalizes Antigravity/agy provider quotas: prefers retrieveUserQuota for live consumption, falls back to fetchAvailableModels and local usage_history, sanitizes cached Provider Limits so retired upstream IDs are not re-exposed, and schedules a deduplicated post-usage refresh. Maintainer follow-up: decoupled the post-usage refresh via a lightweight usageEvents bus (usageHistory no longer dynamic-imports providerLimits) so it does not pull the executors/translator graph into the typecheck-core surface — typecheck:core stays at 0. Integrated into release/v3.8.21. Thanks @dhaern!
2026-06-11 00:32:51 -03:00

59 lines
1.5 KiB
TypeScript

import test from "node:test";
import assert from "node:assert/strict";
import { emitUsageRecorded, onUsageRecorded } from "../../src/lib/usage/usageEvents.ts";
test("emitUsageRecorded delivers provider + connectionId to subscribers", () => {
const seen: Array<[string, string]> = [];
const off = onUsageRecorded((provider, connectionId) => seen.push([provider, connectionId]));
try {
emitUsageRecorded("antigravity", "conn-1");
assert.deepEqual(seen, [["antigravity", "conn-1"]]);
} finally {
off();
}
});
test("emitUsageRecorded no-ops when provider or connectionId is missing", () => {
let calls = 0;
const off = onUsageRecorded(() => {
calls += 1;
});
try {
emitUsageRecorded(null, "conn-1");
emitUsageRecorded("agy", "");
emitUsageRecorded(undefined, undefined);
assert.equal(calls, 0);
} finally {
off();
}
});
test("onUsageRecorded unsubscribe stops further delivery", () => {
let calls = 0;
const off = onUsageRecorded(() => {
calls += 1;
});
emitUsageRecorded("agy", "conn-2");
off();
emitUsageRecorded("agy", "conn-2");
assert.equal(calls, 1);
});
test("a throwing listener does not break emit for others", () => {
let reached = false;
const offBad = onUsageRecorded(() => {
throw new Error("boom");
});
const offGood = onUsageRecorded(() => {
reached = true;
});
try {
assert.doesNotThrow(() => emitUsageRecorded("antigravity", "conn-3"));
assert.equal(reached, true);
} finally {
offBad();
offGood();
}
});