59 lines
2.3 KiB
TypeScript
59 lines
2.3 KiB
TypeScript
import { fireEvent, render, screen, waitFor } from '@testing-library/react';
|
|
import { beforeEach, describe, expect, it, vi } from 'vitest';
|
|
import { TuningApp } from './TuningApp';
|
|
|
|
beforeEach(() => {
|
|
localStorage.clear();
|
|
sessionStorage.clear();
|
|
vi.unstubAllGlobals();
|
|
});
|
|
|
|
describe('TuningApp', () => {
|
|
it('连接调参服务并显示 DeepSeek 能力与新建表单', async () => {
|
|
const fetchMock = vi.fn((input: string | URL | Request, _init?: RequestInit) => {
|
|
void _init;
|
|
const url = String(input);
|
|
if (url.endsWith('/api/tuning/capabilities'))
|
|
return Promise.resolve(
|
|
new Response(
|
|
JSON.stringify({
|
|
ready: true,
|
|
configured: true,
|
|
apiKeyConfigured: true,
|
|
frameworkInstalled: true,
|
|
model: 'deepseek-v4-flash',
|
|
baseUrl: 'https://api.deepseek.com',
|
|
}),
|
|
{ status: 200, headers: { 'Content-Type': 'application/json' } },
|
|
),
|
|
);
|
|
return Promise.resolve(
|
|
new Response(JSON.stringify({ sessions: [] }), {
|
|
status: 200,
|
|
headers: { 'Content-Type': 'application/json' },
|
|
}),
|
|
);
|
|
});
|
|
vi.stubGlobal('fetch', fetchMock);
|
|
const { container } = render(<TuningApp />);
|
|
expect(container.firstElementChild).toHaveClass('h-full', 'overflow-y-auto');
|
|
fireEvent.change(screen.getByLabelText('访问令牌（仅当前标签页）'), {
|
|
target: { value: 'training-secret' },
|
|
});
|
|
fireEvent.click(screen.getByRole('button', { name: '连接/刷新' }));
|
|
expect(await screen.findByText(/deepseek-v4-flash/)).toBeInTheDocument();
|
|
expect(screen.getByText('新建 Unitree-Go2-Flat 调参 Session')).toBeInTheDocument();
|
|
const trialCount = screen.getByLabelText('调参次数（候选配置数）');
|
|
expect(trialCount).toHaveValue(12);
|
|
fireEvent.change(trialCount, { target: { value: '36' } });
|
|
expect(trialCount).toHaveValue(36);
|
|
await waitFor(() => expect(fetchMock).toHaveBeenCalledTimes(2));
|
|
for (const call of fetchMock.mock.calls) {
|
|
expect(String(call[0])).not.toContain('training-secret');
|
|
const init = call[1];
|
|
expect(init).toBeDefined();
|
|
expect(new Headers(init?.headers).get('Authorization')).toBe('Bearer training-secret');
|
|
}
|
|
});
|
|
});
|