fix(quota): derive Codex windows from duration

2026-07-12 23:54:07 +03:00
parent afb368e11b
commit e247343423
3 changed files with 61 additions and 8 deletions
@@ -0,0 +1,48 @@
+import { afterEach, describe, expect, it, vi } from 'vitest';
+
+vi.mock('../../opencode/auth.js', () => ({
+  readAuthFile: () => ({ openai: { access: 'test-token' } }),
+}));
+
+import { fetchQuota } from './codex.js';
+
+afterEach(() => {
+  vi.unstubAllGlobals();
+});
+
+const mockUsage = (rateLimit) => {
+  vi.stubGlobal('fetch', vi.fn().mockResolvedValue({
+    ok: true,
+    json: async () => ({ rate_limit: rateLimit }),
+  }));
+};
+
+describe('Codex quota windows', () => {
+  it('labels a weekly-only primary window from its duration', async () => {
+    mockUsage({
+      primary_window: {
+        used_percent: 3,
+        limit_window_seconds: 604800,
+        reset_at: 1784491827,
+      },
+      secondary_window: null,
+    });
+
+    const result = await fetchQuota();
+
+    expect(result.usage.windows.weekly.usedPercent).toBe(3);
+    expect(result.usage.windows['5h']).toBeUndefined();
+  });
+
+  it('labels five-hour and weekly windows from their durations', async () => {
+    mockUsage({
+      primary_window: { used_percent: 10, limit_window_seconds: 18000 },
+      secondary_window: { used_percent: 20, limit_window_seconds: 604800 },
+    });
+
+    const result = await fetchQuota();
+
+    expect(result.usage.windows['5h'].usedPercent).toBe(10);
+    expect(result.usage.windows.weekly.usedPercent).toBe(20);
+  });
+});