97 lines
2.5 KiB
TypeScript
97 lines
2.5 KiB
TypeScript
import { act, render } from '@testing-library/react';
|
|
import { afterEach, beforeEach, describe, expect, it, vi } from 'vitest';
|
|
import { Models } from '@/pages/Models/index';
|
|
|
|
const hostApiFetchMock = vi.fn();
|
|
const trackUiEventMock = vi.fn();
|
|
|
|
const { gatewayState, settingsState } = vi.hoisted(() => ({
|
|
gatewayState: {
|
|
status: { state: 'running', port: 18789, connectedAt: 1, pid: 1234 },
|
|
},
|
|
settingsState: {
|
|
devModeUnlocked: false,
|
|
},
|
|
}));
|
|
|
|
vi.mock('@/stores/gateway', () => ({
|
|
useGatewayStore: (selector: (state: typeof gatewayState) => unknown) => selector(gatewayState),
|
|
}));
|
|
|
|
vi.mock('@/stores/settings', () => ({
|
|
useSettingsStore: (selector: (state: typeof settingsState) => unknown) => selector(settingsState),
|
|
}));
|
|
|
|
vi.mock('@/lib/host-api', () => ({
|
|
hostApiFetch: (...args: unknown[]) => hostApiFetchMock(...args),
|
|
}));
|
|
|
|
vi.mock('@/lib/telemetry', () => ({
|
|
trackUiEvent: (...args: unknown[]) => trackUiEventMock(...args),
|
|
}));
|
|
|
|
vi.mock('@/components/settings/ProvidersSettings', () => ({
|
|
ProvidersSettings: () => null,
|
|
}));
|
|
|
|
vi.mock('@/components/common/FeedbackState', () => ({
|
|
FeedbackState: ({ title }: { title: string }) => <div>{title}</div>,
|
|
}));
|
|
|
|
vi.mock('react-i18next', () => ({
|
|
useTranslation: () => ({
|
|
t: (key: string, fallback?: string | { count?: number }) => {
|
|
if (typeof fallback === 'string') return fallback;
|
|
return key;
|
|
},
|
|
}),
|
|
}));
|
|
|
|
function createUsageEntry(totalTokens: number) {
|
|
return {
|
|
timestamp: '2026-04-01T12:00:00.000Z',
|
|
sessionId: `session-${totalTokens}`,
|
|
agentId: 'main',
|
|
model: 'gpt-5',
|
|
provider: 'openai',
|
|
inputTokens: totalTokens,
|
|
outputTokens: 0,
|
|
cacheReadTokens: 0,
|
|
cacheWriteTokens: 0,
|
|
totalTokens,
|
|
};
|
|
}
|
|
|
|
describe('Models page auto refresh', () => {
|
|
beforeEach(() => {
|
|
vi.useFakeTimers();
|
|
vi.clearAllMocks();
|
|
gatewayState.status = { state: 'running', port: 18789, connectedAt: 1, pid: 1234 };
|
|
Object.defineProperty(document, 'visibilityState', {
|
|
configurable: true,
|
|
value: 'visible',
|
|
});
|
|
hostApiFetchMock.mockResolvedValue([createUsageEntry(27)]);
|
|
});
|
|
|
|
afterEach(() => {
|
|
vi.useRealTimers();
|
|
});
|
|
|
|
it('refreshes token usage while the page stays open', async () => {
|
|
render(<Models />);
|
|
|
|
await act(async () => {
|
|
await Promise.resolve();
|
|
});
|
|
expect(hostApiFetchMock).toHaveBeenCalledTimes(1);
|
|
|
|
await act(async () => {
|
|
vi.advanceTimersByTime(15_000);
|
|
await Promise.resolve();
|
|
});
|
|
|
|
expect(hostApiFetchMock).toHaveBeenCalledTimes(2);
|
|
});
|
|
});
|