Something went wrong. Try again.
Something went wrong. Try again.
3.9 kB · 106 lines
TypeScript
at main
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107import { test } from "node:test"import assert from "node:assert/strict"import { recordCodexUsage, getCachedCodexUsage, parseCodexUsagePayload } from "./codex-usage"
test("recordCodexUsage caches primary window from x-codex-* headers", () => { recordCodexUsage("codex", { "x-codex-primary-used-percent": "45.5", "x-codex-primary-window-minutes": "300", "x-codex-primary-reset-at": "1750000000", "x-codex-secondary-used-percent": "12", "x-codex-secondary-window-minutes": "10080", }) const usage = getCachedCodexUsage("codex") assert.ok(usage) assert.equal(usage!.percent, 45.5 / 100) assert.equal(usage!.display, "5h 46% 7d 12%") assert.equal((usage!.stats as any).primary.used_percent, 45.5) assert.equal((usage!.stats as any).secondary.used_percent, 12) assert.equal(typeof (usage!.stats as any).age_ms, "number") recordCodexUsage("codex:weekly", { "x-codex-primary-used-percent": "80", "x-codex-primary-window-minutes": "10080", }) assert.equal(getCachedCodexUsage("codex:weekly")!.display, "7d 80%") recordCodexUsage("codex:unknown-window", { "x-codex-primary-used-percent": "33" }) assert.equal(getCachedCodexUsage("codex:unknown-window")!.display, "33%")})
test("codex rate-limit snapshots are isolated by caller cache key", () => { recordCodexUsage("codex:key-a", { "x-codex-primary-used-percent": "10" }) recordCodexUsage("codex:key-b", { "x-codex-primary-used-percent": "80" }) assert.equal(getCachedCodexUsage("codex:key-a")?.used, 10) assert.equal(getCachedCodexUsage("codex:key-b")?.used, 80)})
test("comparable Codex snapshots return per-request plan percentage deltas", () => { assert.deepEqual( recordCodexUsage("codex:account", { "x-codex-primary-used-percent": "10.2", "x-codex-primary-window-minutes": "300", "x-codex-primary-reset-at": "1800000000", }), {}, ) const delta = recordCodexUsage("codex:account", { "x-codex-primary-used-percent": "10.7", "x-codex-primary-window-minutes": "300", "x-codex-primary-reset-at": "1800000000", }) assert.ok(delta.primary !== undefined && Math.abs(delta.primary - 0.5) < 1e-9) assert.deepEqual( recordCodexUsage("codex:account", { "x-codex-primary-used-percent": "0.1", "x-codex-primary-window-minutes": "300", "x-codex-primary-reset-at": "1800000300", }), {}, )})
test("live Codex usage uses the most exhausted window and exposes additional meters", () => { const usage = parseCodexUsagePayload( { plan_type: "pro", rate_limit: { allowed: true, limit_reached: false, primary_window: { used_percent: 20, limit_window_seconds: 18_000, reset_at: 2_000_000_000 }, secondary_window: { used_percent: 80, limit_window_seconds: 604_800, reset_at: 2_000_100_000 }, }, additional_rate_limits: [ { limit_name: "GPT-5.3-Codex-Spark", metered_feature: "spark", rate_limit: { allowed: false, limit_reached: true, primary_window: { used_percent: 100, reset_at: 2_000_200_000 }, }, }, ], }, 1_900_000_000_000, ) assert.equal(usage?.percent, 0.8) assert.equal(usage?.stats.allowed, true) assert.equal((usage?.stats.additional_rate_limits as any[])[0].metered_feature, "spark")})
test("credit overage keeps a plan usable after the nominal windows reach 100 percent", () => { const usage = parseCodexUsagePayload( { rate_limit: { allowed: false, limit_reached: true, primary_window: { used_percent: 100, reset_at: 2_000_000_000 }, }, credits: { has_credits: true, overage_limit_reached: false }, spend_control: { reached: false }, }, 1_900_000_000_000, ) assert.equal(usage?.percent, 1) assert.equal(usage?.stats.allowed, true) assert.equal(usage?.stats.credit_overage, true)})