151 lines
11 KiB
TypeScript
151 lines
11 KiB
TypeScript
import { act, cleanup, fireEvent, render, screen, waitFor } from '@testing-library/react';
|
|
import { afterEach, beforeEach, describe, expect, it, vi } from 'vitest';
|
|
import { TeacherChatPanel } from '@/pages/Chat/TeacherChatPanel';
|
|
import type { TeacherDefinition, TeacherTopic, TeacherRequest } from '../../shared/coding-teacher';
|
|
|
|
const api = vi.hoisted(() => ({ catalog: vi.fn(), config: vi.fn(), conversation: vi.fn(), sendConversation: vi.fn(),
|
|
list: vi.fn(), read: vi.fn(), legacyHistory: vi.fn(), legacyRead: vi.fn(), create: vi.fn(), send: vi.fn(), events: vi.fn(), updateDiscussion: vi.fn(), cancel: vi.fn(), seenConversation: vi.fn() }));
|
|
const identity = vi.hoisted(() => ({ id: 0 }));
|
|
vi.mock('@/lib/coding-teacher', () => ({ teacherApi: api, legacyTopicBase: (base: string) => base,
|
|
teacherTopicsPath: (project: string) => '/legacy/' + project }));
|
|
vi.mock('@/stores/auth', () => ({ useAuthStore: (selector: (state: unknown) => unknown) => selector({ user: { userId: 'kid-' + identity.id } }) }));
|
|
const definition: TeacherDefinition = { schema_version: 1, config_id: 'agent-a', teacher_id: 'teacher',
|
|
name: '设计老师', description: '', avatar_id: 'avatar-01', welcome_message: '聊聊吧', suggested_questions: [],
|
|
system_prompt: '', skills: [], model: { model_id: 'model', reasoning_choice: { mode: 'default' } },
|
|
limits: { max_input_tokens: 8000, max_output_tokens: 1000 } };
|
|
const agent = { teacher_id: 'agent-a', version: 1, definition, is_default: true };
|
|
function request(id: string, text = '旧的问题'): TeacherRequest {
|
|
return { id, text, references: [], createdAt: id, sourceCursor: { seq: 0, workerGeneration: 0 },
|
|
sourceCapturedAt: 'now', includedSourceMessageIds: [], omittedMessages: 0, status: 'completed', response: '原来的回答',
|
|
projectId: 'project-a', projectName: '天气', sourceConversationId: 'source-a', teacherVersion: 1 };
|
|
}
|
|
function topic(requests = [request('1')]): TeacherTopic {
|
|
return { schemaVersion: 1, revision: 1, id: 'chat', accountId: 'kid', projectId: 'project-a', sourceConversationId: 'project',
|
|
definition, version: 1, createdAt: 'now', updatedAt: 'now', requests,
|
|
conversation: { agentId: 'agent-a', segmentTurns: 1, discussions: {} } };
|
|
}
|
|
let stream: EventTarget;
|
|
beforeEach(() => {
|
|
vi.resetAllMocks(); identity.id++; localStorage.clear(); stream = new EventTarget();
|
|
api.config.mockResolvedValue({ enabled: true, definition });
|
|
api.catalog.mockResolvedValue({ items: [agent] });
|
|
api.conversation.mockResolvedValue({ topic: topic(), before: null });
|
|
api.list.mockResolvedValue({ items: [], lastSelectedTopicId: null });
|
|
api.events.mockResolvedValue(Object.assign(stream, { close: vi.fn() }));
|
|
api.seenConversation.mockImplementation(async (_agent, requestId) => ({ ...topic(), conversation: { ...topic().conversation!, seenThrough: requestId } }));
|
|
api.sendConversation.mockImplementation(async (_agent, input) => ({ ...topic(), revision: 2,
|
|
requests: [...topic().requests, { ...request(input.requestId, input.text), projectId: input.projectId, sourceConversationId: input.sourceConversationId }] }));
|
|
});
|
|
afterEach(cleanup);
|
|
const props = { projectId: 'project-a', projectName: '天气', sourceId: 'source-a', selectedAgent: agent };
|
|
async function mount() { let view!: ReturnType<typeof render>; await act(async () => { view = render(<TeacherChatPanel {...props} />); }); return view; }
|
|
|
|
describe('continuous agent panel', () => {
|
|
it('retries incomplete prose only in its original project/source and keeps current drafts and one send identity', async () => {
|
|
const original = { ...request('broken', '现在项目是什么阶段？'), response: '它早就不是', replyIncomplete: true,
|
|
references: [{ kind: 'code' as const, text: '原项目代码' }] };
|
|
api.conversation.mockResolvedValue({ topic: topic([original]), before: null });
|
|
api.seenConversation.mockImplementation(async () => topic([original]));
|
|
let finish!: (value: TeacherTopic) => void;
|
|
api.sendConversation.mockImplementation(() => new Promise<TeacherTopic>(resolve => { finish = resolve; }));
|
|
const view = await mount();
|
|
expect(screen.getByRole('status')).toHaveTextContent('这次回复未能完整显示');
|
|
expect(screen.queryByTestId('teacher-reply')).toBeNull();
|
|
expect(api.sendConversation).not.toHaveBeenCalled();
|
|
fireEvent.change(screen.getByRole('textbox'), { target: { value: '仍在编辑的草稿' } });
|
|
view.rerender(<TeacherChatPanel {...props} projectId="project-b" sourceId="source-b" />);
|
|
expect(screen.getByRole('button', { name: '重新回答' })).toBeDisabled();
|
|
view.rerender(<TeacherChatPanel {...props} sourceId="another-source" />);
|
|
expect(screen.getByRole('button', { name: '重新回答' })).toBeDisabled();
|
|
view.rerender(<TeacherChatPanel {...props} />);
|
|
const retry = screen.getByRole('button', { name: '重新回答' });
|
|
fireEvent.click(retry); fireEvent.click(retry);
|
|
expect(api.sendConversation).toHaveBeenCalledTimes(1);
|
|
const sent = api.sendConversation.mock.calls[0][1];
|
|
expect(sent).toEqual({ requestId: expect.any(String), projectId: 'project-a', sourceConversationId: 'source-a',
|
|
text: original.text, references: original.references });
|
|
expect(sent.requestId).not.toBe(original.id);
|
|
fireEvent.change(screen.getByRole('textbox'), { target: { value: '等待时补充的草稿' } });
|
|
await act(async () => finish({ ...topic([original, { ...request(sent.requestId), response: '完整回答' }]), revision: 2 }));
|
|
expect(screen.getByText('完整回答')).toBeVisible();
|
|
expect(screen.getByRole('textbox')).toHaveValue('等待时补充的草稿');
|
|
expect(JSON.parse(localStorage.getItem(`makelore-consultation-draft:kid-${identity.id}:agent:agent-a`)!)).toMatchObject({ text: '等待时补充的草稿' });
|
|
});
|
|
|
|
it('opens one chat without topic controls or creating a topic, and preserves it across project switches', async () => {
|
|
const view = await mount();
|
|
expect(screen.queryByRole('button', { name: '智能体新话题' })).toBeNull();
|
|
expect(screen.queryByRole('combobox')).toBeNull();
|
|
expect(api.create).not.toHaveBeenCalled();
|
|
expect(screen.getByText('旧的问题')).toBeInTheDocument();
|
|
fireEvent.change(screen.getByRole('textbox'), { target: { value: '跨项目草稿' } });
|
|
view.rerender(<TeacherChatPanel {...props} projectId="project-b" projectName="游戏" sourceId="source-b" />);
|
|
expect(screen.getByRole('textbox')).toHaveValue('跨项目草稿');
|
|
expect(screen.getByText('旧的问题')).toBeInTheDocument();
|
|
expect(screen.getByText('本轮项目：游戏 · 当前操作对话')).toBeInTheDocument();
|
|
fireEvent.click(screen.getByRole('button', { name: '提问' }));
|
|
await waitFor(() => expect(api.sendConversation).toHaveBeenCalledWith('agent-a', expect.objectContaining({
|
|
projectId: 'project-b', sourceConversationId: 'source-b', text: '跨项目草稿' })));
|
|
});
|
|
it('merges older pages and streamed current turns without replacing earlier messages', async () => {
|
|
api.conversation.mockImplementation(async (_id, before) => ({ topic: before ? topic([request('0', '更早的问题')]) : topic(), before: before ? null : '1' }));
|
|
await mount();
|
|
fireEvent.click(screen.getByText('查看更早消息'));
|
|
await screen.findByText('更早的问题');
|
|
await act(async () => stream.dispatchEvent(new MessageEvent('snapshot', { data: JSON.stringify({
|
|
...topic([request('2', '新的问题')]), revision: 2,
|
|
}) })));
|
|
expect(screen.getByText('更早的问题')).toBeInTheDocument();
|
|
expect(screen.getByText('旧的问题')).toBeInTheDocument();
|
|
expect(screen.getByText('新的问题')).toBeInTheDocument();
|
|
});
|
|
it('rejects stale references and prevents carrying an old project reply into the new Pi draft', async () => {
|
|
const bring = vi.fn();
|
|
let view!: ReturnType<typeof render>;
|
|
await act(async () => { view = render(<TeacherChatPanel {...props} quote={{ kind: 'code', text: '旧项目代码' }} onBringBack={bring} />); });
|
|
fireEvent.change(screen.getByRole('textbox'), { target: { value: '看看' } });
|
|
view.rerender(<TeacherChatPanel {...props} projectId="project-b" sourceId="source-b" onBringBack={bring} />);
|
|
fireEvent.click(screen.getByRole('button', { name: '提问' }));
|
|
expect(api.sendConversation).not.toHaveBeenCalled();
|
|
expect(screen.getByRole('alert')).toHaveTextContent('引用来自另一个项目');
|
|
expect(screen.getByText('请打开来源项目和操作对话后带回')).toBeDisabled();
|
|
});
|
|
it('opens an empty chat without a model call and creates it only through the first send', async () => {
|
|
api.conversation.mockResolvedValue({ topic: null, before: null });
|
|
await mount();
|
|
expect(api.sendConversation).not.toHaveBeenCalled();
|
|
fireEvent.change(screen.getByRole('textbox'), { target: { value: '你好' } });
|
|
fireEvent.click(screen.getByRole('button', { name: '提问' }));
|
|
await waitFor(() => expect(api.sendConversation).toHaveBeenCalledTimes(1));
|
|
expect(api.create).not.toHaveBeenCalled();
|
|
});
|
|
it('merges a delayed older page without overwriting a newer streamed answer', async () => {
|
|
let resolve!: (value: {topic: TeacherTopic; before: null}) => void;
|
|
const page = new Promise<{topic: TeacherTopic; before: null}>(done => { resolve = done; });
|
|
api.conversation.mockImplementation((_id, before) => before ? page : Promise.resolve({ topic: topic(), before: '1' }));
|
|
await mount();
|
|
fireEvent.click(screen.getByText('查看更早消息'));
|
|
await act(async () => stream.dispatchEvent(new MessageEvent('snapshot', {data: JSON.stringify({
|
|
...topic([{...request('1'), response: '更新后的回答'}]), revision: 3,
|
|
})})));
|
|
await act(async () => resolve({topic: topic([request('0', '分页里较早的问题'), request('1')]), before: null}));
|
|
expect(screen.getByText('分页里较早的问题')).toBeVisible();
|
|
expect(screen.getByText('更新后的回答')).toBeVisible();
|
|
});
|
|
it('keeps legacy records read only when a live update arrives and returns to the same chat', async () => {
|
|
const old = {...topic([request('old', '旧来源会话')]), id: 'legacy', conversation: undefined};
|
|
api.legacyHistory.mockResolvedValue({items: [{id: 'legacy', title: '来源记录', legacySourceId: 'removed-source'}], lastSelectedTopicId: null});
|
|
api.legacyRead.mockResolvedValue(old);
|
|
const view = await mount();
|
|
fireEvent.click(screen.getByText('旧记录'));
|
|
fireEvent.click(await screen.findByText('来源记录（只读）'));
|
|
await screen.findByText('旧来源会话');
|
|
view.rerender(<TeacherChatPanel {...props} externalTopic={{...topic(), revision: 4}} />);
|
|
expect(screen.getByText('旧来源会话')).toBeVisible();
|
|
expect(screen.getByRole('button', {name: '提问'})).toBeDisabled();
|
|
fireEvent.click(screen.getByText('返回与智能体的聊天'));
|
|
await screen.findByText('旧的问题');
|
|
expect(api.sendConversation).not.toHaveBeenCalled();
|
|
});
|
|
});
|