import { describe, expect, it } from 'bun:test'; import { codexResetCreditDetailsFor, codexResetCreditsFor, quotaRowsFor, } from '../src/services/quotaService'; describe('quotaRowsFor', () => { it('把 Codex 的已用百分比转换为剩余额度', () => { const rows = quotaRowsFor('codex', { rate_limit: { allowed: true, primary_window: { used_percent: 97, limit_window_seconds: 604800, reset_at: 1784698723, }, }, }); expect(rows).toHaveLength(1); expect(rows[0].label).toBe('Weekly limit'); expect(rows[0].remainingPercent).toBe(3); expect(rows[0].reset).toBeTruthy(); }); it('区分 Codex 的 5 小时、周和月限额', () => { const rows = quotaRowsFor('codex', { rate_limit: { primary_window: { used_percent: 10, limit_window_seconds: 18_000, }, secondary_window: { used_percent: 20, limit_window_seconds: 2_592_000, }, }, additional_rate_limits: [ { limit_name: 'Enhanced code review', rate_limit: { primary_window: { used_percent: 30, limit_window_seconds: 604_800, }, }, }, ], }); expect(rows.map((row) => row.label)).toEqual([ '5-hour limit', 'Monthly limit', 'Enhanced code review Weekly limit', ]); }); it('Team 次级窗口缺少时长时与新版一致按周窗口回退，不臆测月限额', () => { const rows = quotaRowsFor('codex', { plan_type: 'team', rate_limit: { primary_window: { used_percent: 10 }, secondary_window: { used_percent: 20 }, }, }); expect(rows.map((row) => row.label)).toEqual(['5-hour limit', 'Weekly limit']); }); it('支持反序窗口且只在有重置时间时推断已耗尽额度', () => { const rows = quotaRowsFor('codex', { rateLimit: { allowed: false, primaryWindow: { limitWindowSeconds: 604800 }, secondaryWindow: { limitWindowSeconds: 18000, resetAfterSeconds: 3600 }, }, }); expect(rows.map((row) => row.label)).toEqual(['5-hour limit', 'Weekly limit']); expect(rows.map((row) => row.remainingPercent)).toEqual([0, null]); expect(rows[0].resetAtMs).toBeGreaterThan(Date.now()); }); it('读取 Codex 可用重置额度', () => { expect(codexResetCreditsFor({ rate_limit_reset_credits: { available_count: '2' }, })).toBe(2); }); it('读取 Codex 重置次数和最早有效过期时间', () => { const result = codexResetCreditDetailsFor({ available_count: '2', credits: [ { id: 'later', reset_type: 'codex_rate_limits', status: 'available', expires_at: '2026-08-20T00:00:00Z', }, { id: 'earlier', reset_type: 'codex_rate_limits', status: 'available', expires_at: '2026-08-12T18:06:25Z', }, { id: 'used', reset_type: 'codex_rate_limits', status: 'used', expires_at: '2026-07-20T00:00:00Z', }, { id: 'expired', reset_type: 'codex_rate_limits', status: 'available', expires_at: '2026-07-15T00:00:00Z', }, ], }, Date.parse('2026-07-16T00:00:00Z')); expect(result).toEqual({ availableCount: 2, earliestExpiry: '2026-08-12T18:06:25Z', }); }); it('重置次数支持 applicable 字段、空列表，不把过期积分算入推断次数', () => { expect(codexResetCreditDetailsFor({ credits: [] })).toEqual({ availableCount: undefined, earliestExpiry: undefined }); expect(codexResetCreditDetailsFor({ applicableAvailableCount: '0', credits: [ { resetType: 'codex_rate_limits', status: 'available', expiresAt: '2030-01-01T00:00:00Z' }, { resetType: 'codex_rate_limits', status: 'available', expiresAt: '2020-01-01T00:00:00Z' }, { resetType: 'codex_rate_limits', status: 'available', expiresAt: 'invalid' }, { resetType: 'other', status: 'available', expiresAt: '2030-01-01T00:00:00Z' }, ], }, Date.parse('2026-01-01T00:00:00Z'))).toEqual({ availableCount: 1, applicableAvailableCount: 0, earliestExpiry: '2030-01-01T00:00:00Z', }); }); it('不会把数组、对象和布尔值当作百分比', () => { for (const used of [[], [25], {}, false, '', ' ', NaN, Infinity]) { expect(quotaRowsFor('codex', { rate_limit: { primary_window: { used_percent: used } } })[0].remainingPercent).toBeNull(); } }); it('不会把小于 1 的上游百分比错误放大 100 倍', () => { const codex = quotaRowsFor('codex', { rate_limit: { primary_window: { used_percent: 0.63 } }, }); const claude = quotaRowsFor('claude', { five_hour: { utilization: 0.44 }, }); expect(codex[0].remainingPercent).toBeCloseTo(99.37); expect(claude[0].remainingPercent).toBeCloseTo(99.56); }); it('Claude 优先活动的现代 Fable 窗口，并去除旧字段重复项', () => { const limit = (percent: unknown, active: boolean, name = 'Fable') => ({ kind: 'weekly_scoped', percent, is_active: active, scope: { model: { display_name: name } }, resets_at: '2030-01-01T00:00:00Z', }); const rows = quotaRowsFor('claude', { iguana_necktie: { utilization: 41 }, five_hour: { utilization: 10 }, limits: [null, limit(null, true), limit(12, false, 'Fable 5'), limit(64, true), limit(99, true, 'Sonnet')], }); expect(rows.map((row) => [row.label, row.remainingPercent])).toEqual([ ['5-hour window', 90], ['7-day Fable window', 36], ]); expect(rows[1].resetAtMs).toBe(Date.parse('2030-01-01T00:00:00Z')); }); it('Claude 无有效现代 Fable 数据时兼容旧字段', () => { const rows = quotaRowsFor('claude', { iguana_necktie: { utilization: 41 }, limits: [{ kind: 'weekly_scoped', percent: null, scope: { model: { display_name: 'Fable' } } }], }); expect(rows).toHaveLength(1); expect(rows[0]).toMatchObject({ label: '7-day Fable window', remainingPercent: 59 }); }); it('显示 Claude 已启用的额外用量', () => { const rows = quotaRowsFor('claude', { five_hour: { utilization: 20, resets_at: '2027-01-01T00:00:00Z' }, extra_usage: { is_enabled: true, monthly_limit: 5000, used_credits: 1250, utilization: 25, }, }); expect(rows.at(-1)).toMatchObject({ label: 'Extra usage', remainingPercent: 75, }); expect(rows.at(-1)?.detail).toContain('$12.50'); expect(rows.at(-1)?.detail).toContain('$50.00'); }); it('按剩余量计算 Kimi 和 Antigravity 百分比', () => { const kimi = quotaRowsFor('kimi', { usage: { used: 99.5, limit: 100 }, }); const antigravity = quotaRowsFor('antigravity', { groups: [ { displayName: 'Gemini', buckets: [{ remainingFraction: 0.03 }], }, ], }); expect(kimi[0].remainingPercent).toBeCloseTo(0.5); expect(antigravity[0].remainingPercent).toBe(3); }); it('Kimi 短期窗口排在周汇总前，识别 protobuf 时间单位且保留小数', () => { const rows = quotaRowsFor('kimi', { usage: { used: 99.5, limit: 100 }, limits: [ { window: { duration: 5, timeUnit: 'TIME_UNIT_HOUR' }, detail: { limit: 100, remaining: 80, reset_in: 3600 } }, { window: { duration: 1, time_unit: 'TIME_UNIT_WEEK' }, detail: { limit: 100, used: 20 } }, { window: { duration: 300, timeUnit: 'TIME_UNIT_MINUTE' }, detail: { limit: 100, used: 0 } }, { window: { duration: 90, timeUnit: 'TIME_UNIT_SECOND' }, detail: { limit: 100, used: 0 } }, ], }); expect(rows.map((row) => row.label)).toEqual(['5 hours window', '7 days window', '5 hours window', '90 seconds window', 'Weekly quota']); expect(rows.at(-1)?.remainingPercent).toBeCloseTo(0.5); expect(rows[0].remainingPercent).toBe(80); expect(rows[0].resetAtMs).toBeGreaterThan(Date.now()); }); it('Antigravity 把 5h 排在周窗口前，兼容百分数字符串且不修改响应', () => { const buckets = [{ window: 'weekly', remainingFraction: '40%' }, { window: 'five_hour', remaining_fraction: 0.8 }]; const rows = quotaRowsFor('antigravity', { groups: [{ displayName: 'Gemini', buckets }] }); expect(rows.map((row) => row.remainingPercent)).toEqual([80, 40]); expect(buckets[0].window).toBe('weekly'); }); it('Antigravity 兼容 body 包装的额度响应', () => { expect(quotaRowsFor('antigravity', { body: JSON.stringify({ groups: [{ buckets: [{ remainingFraction: 0.5 }] }] }) })[0].remainingPercent).toBe(50); }); it('区分 Antigravity 同一分组中的不同窗口', () => { const rows = quotaRowsFor('antigravity', { groups: [{ displayName: 'Gemini Pro', buckets: [ { window: '5h', remainingFraction: 0.8 }, { window: 'weekly', remainingFraction: 0.4 }, ], }], }); expect(rows.map((row) => row.label)).toEqual([ 'Gemini Pro · 5h', 'Gemini Pro · weekly', ]); }); it('xAI 仅返回产品用量时仍能解析，付费账号不伪造剩余额度', () => { expect(quotaRowsFor('xai', { config: { productUsage: [{ product: 'grok', usagePercent: 20 }] } })[1]) .toMatchObject({ label: 'grok', remainingPercent: 80 }); expect(quotaRowsFor('xai', { mode: 'paid-info' })[0]) .toMatchObject({ label: 'Paid API account', remainingPercent: null }); }); it('xAI 不从月账单借用周窗口的重置时间', () => { const rows = quotaRowsFor('xai', { weekly: { config: { creditUsagePercent: 25 } }, monthly: { config: { monthlyLimit: 1000, used: 500, billingPeriodEnd: '2030-01-01T00:00:00Z' } }, }); expect(rows[0].resetAtMs).toBeUndefined(); expect(rows[1].resetAtMs).toBe(Date.parse('2030-01-01T00:00:00Z')); }); it('合并 xAI 每周、月度和按量付费额度', () => { const rows = quotaRowsFor('xai', { weekly: { config: { currentPeriod: { type: 'weekly', end: '2027-01-01T00:00:00Z' }, creditUsagePercent: 25, productUsage: [{ product: 'grok-code', usagePercent: 40 }], }, }, monthly: { config: { monthlyLimit: { val: 1000 }, used: { val: 1200 }, onDemandCap: { val: 500 }, billingPeriodEnd: '2027-01-31T00:00:00Z', }, }, }); expect(rows.find((row) => row.label === 'Weekly quota')?.remainingPercent).toBe(75); expect(rows.find((row) => row.label === 'grok-code')?.remainingPercent).toBe(60); expect(rows.find((row) => row.label === 'Monthly included quota')?.remainingPercent).toBe(0); expect(rows.find((row) => row.label === 'On-demand quota')?.remainingPercent).toBe(60); }); it('xAI 保留 0 和 100 的边界值，缺失用量不显示全额余额', () => { expect(quotaRowsFor('xai', { config: { creditUsagePercent: 0 } })[0].remainingPercent).toBe(100); expect(quotaRowsFor('xai', { config: { creditUsagePercent: 100 } })[0].remainingPercent).toBe(0); expect(quotaRowsFor('xai', { config: { monthlyLimit: { val: 15000 }, used: { val: 0 } } })[0].remainingPercent).toBe(100); const unknown = quotaRowsFor('xai', { config: { monthlyLimit: 15000 } })[0]; expect(unknown.remainingPercent).toBeNull(); expect(unknown.detail).not.toBe('US$150.00 / US$150.00'); expect(quotaRowsFor('xai', { config: { monthlyLimit: 15000, used: 15000 } })[0].remainingPercent).toBe(0); }); it('xAI 按字段合并并保留活动周期，不由月账单补齐周重置时间', () => { const rows = quotaRowsFor('xai', { weekly: { config: { currentPeriod: { type: 'weekly' } } }, monthly: { config: { creditUsagePercent: 0, currentPeriod: { type: 'monthly', end: '2030-01-01T00:00:00Z' } } }, }); expect(rows).toHaveLength(1); expect(rows[0].remainingPercent).toBe(100); expect(rows[0].resetAtMs).toBeUndefined(); }); });