lzt404
e4d0059efa
- Update Claude, Codex, Kimi, Antigravity, and xAI compatibility - Display plan details, subscription expiry, and dynamic reset times - Deduplicate quota requests and serialize Codex resets - Require confirmation for potentially billable xAI health probes - Add quota parsing, request, and rendering regression tests
54 lines
2.7 KiB
TypeScript
54 lines
2.7 KiB
TypeScript
import { describe, expect, it } from 'bun:test';
|
|
import { antigravityProjectFor, codexMetadataFor, isPaidXaiFile } from '../src/services/quotaMetadata';
|
|
import { providerForFile } from '../src/services/quotaService';
|
|
|
|
const jwt = (payload: unknown) => `header.${Buffer.from(JSON.stringify(payload)).toString('base64url')}.signature`;
|
|
|
|
describe('quota credential metadata', () => {
|
|
it('识别 xAI 别名，保留原有供应商别名', () => {
|
|
for (const provider of ['x_ai', 'x-ai', 'grok', 'xai', 'XAI']) {
|
|
expect(providerForFile({ provider })).toBe('xai');
|
|
}
|
|
expect(providerForFile({ type: 'anthropic' })).toBe('claude');
|
|
expect(providerForFile({ type: 'anti_gravity' })).toBe('antigravity');
|
|
});
|
|
|
|
it('Codex 支持各层的 JWT、JSON 和对象令牌以及命名空间字段', () => {
|
|
const info = {
|
|
chatgpt_account_id: 'account-中文', chatgpt_plan_type: 'PRO',
|
|
chatgpt_subscription_active_until: '2030-01-01T00:00:00Z',
|
|
};
|
|
for (const token of [info, JSON.stringify(info), jwt({ 'https://api.openai.com/auth': info })]) {
|
|
for (const file of [{ id_token: token }, { metadata: { idToken: token } }, { attributes: { id_token: token } }]) {
|
|
expect(codexMetadataFor(file)).toEqual({
|
|
accountId: 'account-中文', plan: 'pro', subscriptionActiveUntil: '2030-01-01T00:00:00Z',
|
|
});
|
|
}
|
|
}
|
|
});
|
|
|
|
it('保留 Codex 直接 account_id 和订阅对象，安全忽略坏令牌', () => {
|
|
expect(codexMetadataFor({ id_token: 'bad.jwt', account_id: 'direct', metadata: { subscription: { active_until: 1900000000 } } }))
|
|
.toEqual({ accountId: 'direct', plan: undefined, subscriptionActiveUntil: '1900000000' });
|
|
expect(codexMetadataFor({ id_token: [], subscription_active_until: 0 }).subscriptionActiveUntil).toBeUndefined();
|
|
});
|
|
|
|
it('Antigravity 支持 metadata、attributes 和下载文件的项目字段', () => {
|
|
for (const file of [
|
|
{ project_id: 'p' }, { metadata: { projectId: 'p' } },
|
|
{ attributes: { gemini_virtual_project: 'p' } },
|
|
{ installed: { project_id: 'p' } }, { web: { projectId: 'p' } },
|
|
]) expect(antigravityProjectFor(file)).toBe('p');
|
|
expect(antigravityProjectFor({})).toBe('');
|
|
});
|
|
|
|
it('单独 using_api 或 paid 前缀不能证明 xAI 是付费账户', () => {
|
|
expect(isPaidXaiFile({ using_api: true })).toBe(false);
|
|
expect(isPaidXaiFile({ prefix: 'paid' })).toBe(false);
|
|
expect(isPaidXaiFile({ usingApi: 'YES', metadata: { prefix: 'PAID' } })).toBe(true);
|
|
expect(isPaidXaiFile({ metadata: { oauth: { access_token: jwt({ 'https://x.ai/tier': 1 }) } } })).toBe(true);
|
|
expect(isPaidXaiFile({ token: jwt({ tier: 0 }) })).toBe(false);
|
|
expect(isPaidXaiFile({ token: 'bad.jwt' })).toBe(false);
|
|
});
|
|
});
|