import { describe, expect, it, vi, beforeEach } from 'vitest' import { flushPromises, mount } from '@vue/test-utils' import AccountUsageCell from '../AccountUsageCell.vue' import type { Account } from '@/types' const { getUsage } = vi.hoisted(() => ({ getUsage: vi.fn() })) vi.mock('@/api/admin', () => ({ adminAPI: { accounts: { getUsage } } })) vi.mock('vue-i18n', async () => { const actual = await vi.importActual('vue-i18n') return { ...actual, useI18n: () => ({ t: (key: string) => key }) } }) function makeAccount(overrides: Partial): Account { return { id: 1, name: 'account', platform: 'antigravity', type: 'oauth', proxy_id: null, concurrency: 1, priority: 1, status: 'active', error_message: null, last_used_at: null, expires_at: null, auto_pause_on_expired: true, created_at: '2026-03-15T00:00:00Z', updated_at: '2026-03-15T00:00:00Z', schedulable: true, rate_limited_at: null, rate_limit_reset_at: null, overload_until: null, temp_unschedulable_until: null, temp_unschedulable_reason: null, session_window_start: null, session_window_end: null, session_window_status: null, ...overrides, } } describe('AccountUsageCell', () => { beforeEach(() => { getUsage.mockReset() Object.defineProperty(window, 'matchMedia', { writable: true, value: vi.fn().mockImplementation(() => ({ matches: true, media: '(min-width: 768px)', onchange: null, addListener: vi.fn(), removeListener: vi.fn(), addEventListener: vi.fn(), removeEventListener: vi.fn(), dispatchEvent: vi.fn(), })) }) }) it('renders eligible Ollama Cloud state and forwards query updates', async () => { const wrapper = mount(AccountUsageCell, { props: { account: makeAccount({ id: 9001, platform: 'openai', type: 'apikey', ollama_cloud_usage: { account_id: 9001, eligible: true, configured: true, auto_refresh_enabled: true, encryption_key_configured: true, snapshot: { status: 'ok', last_attempt_at: '2026-07-23T00:00:00Z', next_refresh_at: '2026-07-23T01:00:00Z', data: { five_hour: { used_percent: 12 }, seven_day: { used_percent: 34 } } } } }) }, global: { stubs: { OllamaCloudUsageCell: { props: ['account'], emits: ['updated'], template: '' }, UsageProgressBar: true, AccountQuotaInfo: true } } }) expect(wrapper.get('[data-test="embedded-ollama"]').text()).toBe('12') expect(getUsage).not.toHaveBeenCalled() await wrapper.get('[data-test="embedded-ollama"]').trigger('click') const updatedAccount = wrapper.emitted('account-updated')?.[0]?.[0] expect(updatedAccount?.id).toBe(9001) expect(updatedAccount?.ollama_cloud_usage?.auto_refresh_enabled).toBe(false) }) it('Antigravity 图片用量会聚合新旧 image 模型', async () => { getUsage.mockResolvedValue({ antigravity_quota: { 'gemini-2.5-flash-image': { utilization: 45, reset_time: '2026-03-01T11:00:00Z' }, 'gemini-3.1-flash-image': { utilization: 20, reset_time: '2026-03-01T10:00:00Z' }, 'gemini-3-pro-image': { utilization: 70, reset_time: '2026-03-01T09:00:00Z' } } }) const wrapper = mount(AccountUsageCell, { props: { account: makeAccount({ id: 1001, platform: 'antigravity', type: 'oauth', extra: {} }) }, global: { stubs: { UsageProgressBar: { props: ['label', 'utilization', 'resetsAt', 'color'], template: '
{{ label }}|{{ utilization }}|{{ resetsAt }}
' }, AccountQuotaInfo: true } } }) await flushPromises() expect(wrapper.text()).toContain('admin.accounts.usageWindow.gemini3Image|70|2026-03-01T09:00:00Z') }) it('Antigravity 会显示 AI Credits 余额信息', async () => { getUsage.mockResolvedValue({ ai_credits: [ { credit_type: 'GOOGLE_ONE_AI', amount: 25, minimum_balance: 5 } ] }) const wrapper = mount(AccountUsageCell, { props: { account: makeAccount({ id: 1002, platform: 'antigravity', type: 'oauth', extra: {} }) }, global: { stubs: { UsageProgressBar: true, AccountQuotaInfo: true } } }) await flushPromises() expect(wrapper.text()).toContain('admin.accounts.aiCreditsBalance') expect(wrapper.text()).toContain('25') }) it('OpenAI OAuth 快照已过期时首屏会重新请求 usage', async () => { getUsage.mockResolvedValue({ five_hour: { utilization: 15, resets_at: '2026-03-08T12:00:00Z', remaining_seconds: 3600, window_stats: { requests: 3, tokens: 300, cost: 0.03, standard_cost: 0.03, user_cost: 0.03 } }, seven_day: { utilization: 77, resets_at: '2026-03-13T12:00:00Z', remaining_seconds: 3600, window_stats: { requests: 3, tokens: 300, cost: 0.03, standard_cost: 0.03, user_cost: 0.03 } } }) const wrapper = mount(AccountUsageCell, { props: { account: makeAccount({ id: 2000, platform: 'openai', type: 'oauth', extra: { codex_usage_updated_at: '2026-03-07T00:00:00Z', codex_5h_used_percent: 12, codex_5h_reset_at: '2026-03-08T12:00:00Z', codex_7d_used_percent: 34, codex_7d_reset_at: '2026-03-13T12:00:00Z' } }) }, global: { stubs: { UsageProgressBar: { props: ['label', 'utilization', 'resetsAt', 'windowStats', 'color'], template: '
{{ label }}|{{ utilization }}|{{ windowStats?.tokens }}
' }, AccountQuotaInfo: true } } }) await flushPromises() expect(getUsage).toHaveBeenCalledWith(2000) expect(wrapper.text()).toContain('5h|15|300') expect(wrapper.text()).toContain('7d|77|300') }) it('OpenAI OAuth 有 codex 快照时仍然使用 /usage API 数据渲染', async () => { getUsage.mockResolvedValue({ five_hour: { utilization: 18, resets_at: '2099-03-07T12:00:00Z', remaining_seconds: 3600, window_stats: { requests: 9, tokens: 900, cost: 0.09, standard_cost: 0.09, user_cost: 0.09 } }, seven_day: { utilization: 36, resets_at: '2099-03-13T12:00:00Z', remaining_seconds: 3600, window_stats: { requests: 9, tokens: 900, cost: 0.09, standard_cost: 0.09, user_cost: 0.09 } } }) const wrapper = mount(AccountUsageCell, { props: { account: makeAccount({ id: 2001, platform: 'openai', type: 'oauth', extra: { codex_usage_updated_at: '2099-03-07T10:00:00Z', codex_5h_used_percent: 12, codex_5h_reset_at: '2099-03-07T12:00:00Z', codex_7d_used_percent: 34, codex_7d_reset_at: '2099-03-13T12:00:00Z' } }) }, global: { stubs: { UsageProgressBar: { props: ['label', 'utilization', 'resetsAt', 'windowStats', 'color'], template: '
{{ label }}|{{ utilization }}|{{ windowStats?.tokens }}
' }, AccountQuotaInfo: true } } }) await flushPromises() expect(getUsage).toHaveBeenCalledWith(2001) // 单一数据源:始终使用 /usage API 返回值,忽略 codex 快照 expect(wrapper.text()).toContain('5h|18|900') expect(wrapper.text()).toContain('7d|36|900') }) it('OpenAI OAuth 有现成快照时,手动刷新信号会触发 usage 重拉', async () => { getUsage.mockResolvedValue({ five_hour: { utilization: 18, resets_at: '2099-03-07T12:00:00Z', remaining_seconds: 3600, window_stats: { requests: 9, tokens: 900, cost: 0.09, standard_cost: 0.09, user_cost: 0.09 } }, seven_day: { utilization: 36, resets_at: '2099-03-13T12:00:00Z', remaining_seconds: 3600, window_stats: { requests: 9, tokens: 900, cost: 0.09, standard_cost: 0.09, user_cost: 0.09 } } }) const wrapper = mount(AccountUsageCell, { props: { account: makeAccount({ id: 2010, platform: 'openai', type: 'oauth', extra: { codex_usage_updated_at: '2099-03-07T10:00:00Z', codex_5h_used_percent: 12, codex_5h_reset_at: '2099-03-07T12:00:00Z', codex_7d_used_percent: 34, codex_7d_reset_at: '2099-03-13T12:00:00Z' }, rate_limit_reset_at: null }), manualRefreshToken: 0 }, global: { stubs: { UsageProgressBar: { props: ['label', 'utilization', 'resetsAt', 'windowStats', 'color'], template: '
{{ label }}|{{ utilization }}|{{ windowStats?.tokens }}
' }, AccountQuotaInfo: true } } }) await flushPromises() // mount 时已经拉取一次 expect(getUsage).toHaveBeenCalledTimes(1) await wrapper.setProps({ manualRefreshToken: 1 }) await flushPromises() // 手动刷新再拉一次 expect(getUsage).toHaveBeenCalledTimes(2) expect(getUsage).toHaveBeenCalledWith(2010) // 单一数据源:始终使用 /usage API 值 expect(wrapper.text()).toContain('5h|18|900') }) it('OpenAI OAuth 在无 codex 快照时会回退显示 usage 接口窗口', async () => { getUsage.mockResolvedValue({ five_hour: { utilization: 0, resets_at: null, remaining_seconds: 0, window_stats: { requests: 2, tokens: 27700, cost: 0.06, standard_cost: 0.06, user_cost: 0.06 } }, seven_day: { utilization: 0, resets_at: null, remaining_seconds: 0, window_stats: { requests: 2, tokens: 27700, cost: 0.06, standard_cost: 0.06, user_cost: 0.06 } } }) const wrapper = mount(AccountUsageCell, { props: { account: makeAccount({ id: 2002, platform: 'openai', type: 'oauth', extra: {} }) }, global: { stubs: { UsageProgressBar: { props: ['label', 'utilization', 'resetsAt', 'windowStats', 'color'], template: '
{{ label }}|{{ utilization }}|{{ windowStats?.tokens }}
' }, AccountQuotaInfo: true } } }) await flushPromises() expect(getUsage).toHaveBeenCalledWith(2002) expect(wrapper.text()).toContain('5h|0|27700') expect(wrapper.text()).toContain('7d|0|27700') }) it('OpenAI OAuth 在行数据刷新但仍无 codex 快照时会重新拉取 usage', async () => { getUsage .mockResolvedValueOnce({ five_hour: { utilization: 0, resets_at: null, remaining_seconds: 0, window_stats: { requests: 1, tokens: 100, cost: 0.01, standard_cost: 0.01, user_cost: 0.01 } }, seven_day: null }) .mockResolvedValueOnce({ five_hour: { utilization: 0, resets_at: null, remaining_seconds: 0, window_stats: { requests: 2, tokens: 200, cost: 0.02, standard_cost: 0.02, user_cost: 0.02 } }, seven_day: null }) const wrapper = mount(AccountUsageCell, { props: { account: makeAccount({ id: 2003, platform: 'openai', type: 'oauth', updated_at: '2026-03-07T10:00:00Z', extra: {} }) }, global: { stubs: { UsageProgressBar: { props: ['label', 'utilization', 'resetsAt', 'windowStats', 'color'], template: '
{{ label }}|{{ utilization }}|{{ windowStats?.tokens }}
' }, AccountQuotaInfo: true } } }) await flushPromises() expect(wrapper.text()).toContain('5h|0|100') expect(getUsage).toHaveBeenCalledTimes(1) await wrapper.setProps({ account: { id: 2003, platform: 'openai', type: 'oauth', updated_at: '2026-03-07T10:01:00Z', extra: {} } }) await flushPromises() expect(getUsage).toHaveBeenCalledTimes(2) expect(wrapper.text()).toContain('5h|0|200') }) it('OpenAI 重置响应更新账号行时不会额外拉取 usage', async () => { getUsage.mockResolvedValue({ five_hour: { utilization: 0, resets_at: null, remaining_seconds: 0 }, seven_day: null }) const account = makeAccount({ id: 2004, platform: 'openai', type: 'oauth', updated_at: '2026-03-07T10:00:00Z', extra: {} }) const wrapper = mount(AccountUsageCell, { props: { account }, global: { stubs: { UsageProgressBar: true, AccountQuotaInfo: true, OpenAIQuotaResetCell: { props: ['account'], emits: ['account-updated'], template: '