Files
OmniRoute/tests/unit/usage-bailian-split.test.ts
MumuTW f0b08f95c3 chore(usage): decompose services/usage.ts into per-provider usage/* leaves (999 → 253) (#8545)
* chore(usage): extract crof, nanogpt, qoder, opencode, deepseek, bailian, vertex, xiaomi-mimo, xai, github usage fetchers into usage/* leaves

Decompose services/usage.ts (god-file phase 1): move the remaining per-provider
usage fetcher/parser logic into co-located leaves under open-sse/services/usage/
so usage.ts becomes a thin dispatcher (imports + USAGE_FETCHER_PROVIDERS +
getUsageForProvider switch + __testing re-exports).

New leaves (each a pure data transform or independent fetcher, no orchestration):
  - usage/github.ts    getGitHubUsage, formatGitHubQuotaSnapshot, inferGitHubPlanName, shouldDisplayGitHubQuota
  - usage/crof.ts      getCrofUsage
  - usage/nanogpt.ts   getNanoGptUsage
  - usage/qoder.ts     getQoderUsage, parseQoderUserStatusUsage
  - usage/opencode.ts  getOpencodeUsage
  - usage/deepseek.ts  getDeepseekUsage
  - usage/bailian.ts   getBailianCodingPlanUsage
  - usage/vertex.ts    getVertexUsage
  - usage/xiaomi-mimo.ts getXiaomiMimoUsage
  - usage/xai.ts       getXaiUsage

usage.ts re-exports parseQoderUserStatusUsage (named) and threads every helper
the existing __testing contract exposes (usage-utils / usage-service-hardening /
qoder-usage-quota / xiaomi-mimo-selftrack / xai-usage / vertex-spend suites read
them from services/usage). External importers unchanged.

open-sse/services/usage.ts: 1065 -> 256 lines (cap 800).
npm run check:file-size: OK. typecheck:core: OK. eslint: clean. check:cycles: OK.

Added characterization tests (tests/unit/usage-<provider>-split.test.ts) that
import each leaf directly and pin its export surface + key edges, mirroring the
existing usage-quota-core-split / usage-scalars-split pattern.

* docs(changelog): add fragment for this PR
2026-07-26 03:52:19 -03:00

60 lines
2.2 KiB
TypeScript

// Characterization of the services/usage.ts bailian split (god-file decomposition): the Bailian
// (Alibaba Token Plan) triple-window fetcher (getBailianCodingPlanUsage) moved into
// services/usage/bailian.ts so usage.ts stays a thin dispatcher. Behavior-preserving move — this
// locks the export surface and the worst-case window selection; the full triple-window matrix
// is covered via getUsageForProvider in bailian-usage.test.ts.
import { test } from "node:test";
import assert from "node:assert/strict";
const B = await import("../../open-sse/services/usage/bailian.ts");
test("module exposes getBailianCodingPlanUsage", () => {
assert.equal(typeof B.getBailianCodingPlanUsage, "function");
});
test("getBailianCodingPlanUsage picks the most restrictive window as the surfaced quota", async () => {
const originalFetch = globalThis.fetch;
globalThis.fetch = async () =>
new Response(
JSON.stringify({
code: "Success",
data: {
codingPlanInstanceInfos: [
{
planName: "Qwen3 Coder Next",
codingPlanQuotaInfo: {
per5HourUsedQuota: 60,
per5HourTotalQuota: 100,
per5HourQuotaNextRefreshTime: 1718304000,
perWeekUsedQuota: 80,
perWeekTotalQuota: 100,
perWeekQuotaNextRefreshTime: 1718563200,
perBillMonthUsedQuota: 40,
perBillMonthTotalQuota: 100,
perBillMonthQuotaNextRefreshTime: 1719772800,
},
},
],
},
}),
{ status: 200, headers: { "Content-Type": "application/json" } }
);
try {
const r = (await B.getBailianCodingPlanUsage("conn", "sk-test", { consoleApiKey: "ck" })) as {
plan?: string;
used?: number;
total?: number;
remaining?: number;
remainingPercentage?: number;
unlimited?: boolean;
};
assert.equal(r.plan, "Alibaba Token Plan");
assert.equal(r.unlimited, false);
// weekly 80% is the most restrictive → used/total = 0.8
assert.equal(r.used! / r.total!, 0.8);
assert.equal(r.remainingPercentage, 20);
} finally {
globalThis.fetch = originalFetch;
}
});