Preserve stable snapshots and stabilize Electron e2e (#734)
This commit is contained in:
committed by
GitHub
Unverified
parent
34bbb039d3
commit
5a3da41562
96
tests/unit/models-page.test.tsx
Normal file
96
tests/unit/models-page.test.tsx
Normal file
@@ -0,0 +1,96 @@
|
||||
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);
|
||||
});
|
||||
});
|
||||
Reference in New Issue
Block a user