73 lines
3.3 KiB
JavaScript
73 lines
3.3 KiB
JavaScript
|
import { describe, expect, it, vi } from 'vitest';
|
||
|
|
|
||
|
|
import { fetchCommandCodeUsage, fetchQuota, parseCommandCodeCredits } from './command-code.js';
|
||
|
|
|
||
|
|
const creditsPayload = {
|
||
|
|
credits: { monthlyCredits: 120, purchasedCredits: 30, freeCredits: 5 },
|
||
|
|
windowLimits: {
|
||
|
|
fiveHour: { used: 25, cap: 100, resetAt: 1_776_000_000 },
|
||
|
|
weekly: { used: 70, cap: 200, resetAt: 1_776_604_800 },
|
||
|
|
},
|
||
|
|
};
|
||
|
|
|
||
|
|
describe('Command Code quota provider', () => {
|
||
|
|
it('parses balances and rate-limit windows', () => {
|
||
|
|
const windows = parseCommandCodeCredits(creditsPayload);
|
||
|
|
expect(windows.monthly_credits).toMatchObject({ usedPercent: null, valueLabel: '120' });
|
||
|
|
expect(windows.purchased_credits).toMatchObject({ usedPercent: null, valueLabel: '30' });
|
||
|
|
expect(windows.free_credits).toMatchObject({ usedPercent: null, valueLabel: '5' });
|
||
|
|
expect(windows['5h']).toMatchObject({ usedPercent: 25, valueLabel: '25 / 100', resetAt: 1_776_000_000_000 });
|
||
|
|
expect(windows.weekly.usedPercent).toBe(35);
|
||
|
|
});
|
||
|
|
|
||
|
|
it('formats fractional credit values for display', () => {
|
||
|
|
const windows = parseCommandCodeCredits({
|
||
|
|
credits: { monthlyCredits: 69.7947070034 },
|
||
|
|
windowLimits: { fiveHour: { used: 0.2052929966, cap: 14 } },
|
||
|
|
});
|
||
|
|
expect(windows.monthly_credits.valueLabel).toBe('69.79');
|
||
|
|
expect(windows['5h'].valueLabel).toBe('0.21 / 14');
|
||
|
|
});
|
||
|
|
|
||
|
|
it('resolves the organization before fetching credits', async () => {
|
||
|
|
const requests = [];
|
||
|
|
const windows = await fetchCommandCodeUsage('secret', async (url, options) => {
|
||
|
|
requests.push({ url, options });
|
||
|
|
return new Response(JSON.stringify(url.endsWith('/alpha/whoami') ? { org: { id: 'org/a' } } : creditsPayload));
|
||
|
|
});
|
||
|
|
expect(requests.map(({ url }) => url)).toEqual([
|
||
|
|
'https://api.commandcode.ai/alpha/whoami',
|
||
|
|
'https://api.commandcode.ai/alpha/billing/credits?orgId=org%2Fa',
|
||
|
|
]);
|
||
|
|
expect(requests[0].options.headers.Authorization).toBe('Bearer secret');
|
||
|
|
expect(windows['5h'].usedPercent).toBe(25);
|
||
|
|
});
|
||
|
|
|
||
|
|
it('fetches account-scoped credits without orgId for personal accounts', async () => {
|
||
|
|
const urls = [];
|
||
|
|
await fetchCommandCodeUsage('secret', async (url) => {
|
||
|
|
urls.push(url);
|
||
|
|
return new Response(JSON.stringify(url.endsWith('/alpha/whoami') ? { user: { id: 'user-1' }, org: null } : creditsPayload));
|
||
|
|
});
|
||
|
|
expect(urls).toEqual([
|
||
|
|
'https://api.commandcode.ai/alpha/whoami',
|
||
|
|
'https://api.commandcode.ai/alpha/billing/credits',
|
||
|
|
]);
|
||
|
|
});
|
||
|
|
|
||
|
|
it('does not expose credentials in authentication errors', async () => {
|
||
|
|
await expect(fetchCommandCodeUsage('secret', async () => new Response('', { status: 401 }))).rejects.toThrow('authentication failed');
|
||
|
|
});
|
||
|
|
|
||
|
|
it('reads OAuth access credentials from the OpenCode auth file', async () => {
|
||
|
|
const fetchMock = vi.fn()
|
||
|
|
.mockResolvedValueOnce(new Response(JSON.stringify({ org: { id: 'org-1' } })))
|
||
|
|
.mockResolvedValueOnce(new Response(JSON.stringify(creditsPayload)));
|
||
|
|
vi.stubGlobal('fetch', fetchMock);
|
||
|
|
const result = await fetchQuota({ 'command-code': { type: 'oauth', access: 'test-token' } });
|
||
|
|
expect(result).toMatchObject({ providerId: 'command-code', ok: true, configured: true });
|
||
|
|
expect(fetchMock.mock.calls[0][1].headers.Authorization).toBe('Bearer test-token');
|
||
|
|
vi.unstubAllGlobals();
|
||
|
|
});
|
||
|
|
});
|