226 lines
13 KiB
TypeScript
226 lines
13 KiB
TypeScript
import { act, cleanup, fireEvent, render, screen, within } from '@testing-library/react';
|
|
import { afterEach, beforeEach, expect, it, vi } from 'vitest';
|
|
import { CloudKnowledgePanel } from '@/pages/CloudAgents/CloudKnowledge';
|
|
import type { CloudKnowledgeFile } from '../../shared/cloud-agents';
|
|
|
|
const api = vi.hoisted(() => ({ call: vi.fn(), uploadKnowledge: vi.fn() }));
|
|
vi.mock('@/lib/cloud-agents-api', () => ({ cloudAgentsApi: api }));
|
|
const file: CloudKnowledgeFile = { file_id: 'file', name: '学习路线图.pdf', size: 3000, status: 'uploaded', chunk_count: 0, error: null, available: false };
|
|
let items: CloudKnowledgeFile[];
|
|
beforeEach(() => {
|
|
vi.resetAllMocks(); items = [];
|
|
api.call.mockImplementation(async operation => {
|
|
if (operation === 'knowledge') return { databases: [{ kb_id: 'kb', name: '资料' }, { kb_id: 'other', name: '其他库' }], models: [] };
|
|
if (operation === 'knowledgeFiles') return { files: items, next_offset: null };
|
|
if (operation === 'processKnowledge') {
|
|
items = items.map(value => ({ ...value, processing_task: { task_id: 'task', status: 'pending', error: null } }));
|
|
return { task_id: 'task' };
|
|
}
|
|
throw new Error('unexpected ' + operation);
|
|
});
|
|
});
|
|
afterEach(() => { cleanup(); vi.useRealTimers(); });
|
|
async function open() {
|
|
render(<CloudKnowledgePanel slug="ml-agent" onCreated={() => undefined} />);
|
|
fireEvent.click(screen.getByText('管理我的知识文档'));
|
|
await screen.findByRole('option', { name: '资料' });
|
|
fireEvent.change(screen.getByLabelText('管理的知识库'), { target: { value: 'kb' } });
|
|
}
|
|
async function tick() { await act(async () => { await vi.advanceTimersByTimeAsync(5000); }); }
|
|
|
|
it('uploads then processes automatically and refreshes through queue, parse, index and readiness', async () => {
|
|
await open();
|
|
await screen.findByText('上传第一份文档，让智能体用你的资料回答。');
|
|
vi.useFakeTimers();
|
|
api.uploadKnowledge.mockImplementation(async () => { items = [file]; return file; });
|
|
await act(async () => { fireEvent.click(screen.getByRole('button', { name: '上传文档', exact: true })); });
|
|
expect(screen.getByText('等待处理')).toBeVisible();
|
|
expect(screen.queryByText(/0 个分块/)).not.toBeInTheDocument();
|
|
expect(api.call).toHaveBeenCalledWith('processKnowledge', expect.objectContaining({ kb_id: 'kb', file_id: 'file' }));
|
|
items = [{ ...file, status: 'parsing', processing_task: { task_id: 'task', status: 'running', error: null } }];
|
|
await tick(); expect(screen.getByText('正在解析文档')).toBeVisible();
|
|
items = [{ ...items[0], status: 'indexing' }];
|
|
await tick(); expect(screen.getByText('正在建立索引')).toBeVisible();
|
|
items = [{ ...file, status: 'indexed', available: true, chunk_count: 12, processing_task: { task_id: 'task', status: 'success', error: null } }];
|
|
await tick();
|
|
expect(screen.getByText('可用于回答')).toBeVisible();
|
|
expect(screen.getByText(/12 段可检索内容/)).toBeVisible();
|
|
expect(api.uploadKnowledge).toHaveBeenCalledTimes(1);
|
|
});
|
|
|
|
it('does not call zero chunks searchable and distinguishes loading from an empty library', async () => {
|
|
let resolve!: (value: unknown) => void;
|
|
const original = api.call.getMockImplementation()!;
|
|
api.call.mockImplementation((operation, input) => operation === 'knowledgeFiles'
|
|
? new Promise(value => { resolve = value; }) : original(operation, input));
|
|
await open();
|
|
expect(screen.getByText('正在读取文档状态…')).toBeVisible();
|
|
expect(screen.queryByText(/上传第一份/)).not.toBeInTheDocument();
|
|
await act(async () => resolve({ files: [{ ...file, status: 'indexed' }], next_offset: null }));
|
|
expect(screen.getByText('没有可用内容')).toBeVisible();
|
|
expect(screen.queryByText('可用于回答')).not.toBeInTheDocument();
|
|
expect(screen.getByText(/扫描 PDF 需先识别文字/)).toBeVisible();
|
|
});
|
|
|
|
it('retains the saved upload and processing identity after an uncertain submission', async () => {
|
|
const original = api.call.getMockImplementation()!;
|
|
const attempts: unknown[] = [];
|
|
api.call.mockImplementation(async (operation, input) => {
|
|
if (operation === 'processKnowledge') {
|
|
attempts.push(input);
|
|
if (attempts.length === 1) throw new Error('连接超时');
|
|
}
|
|
return original(operation, input);
|
|
});
|
|
api.uploadKnowledge.mockImplementation(async () => { items = [file]; return file; });
|
|
await open();
|
|
fireEvent.click(screen.getByRole('button', { name: '上传文档', exact: true }));
|
|
expect(await screen.findByRole('alert')).toHaveTextContent('文件已保存，处理请求尚未确认');
|
|
expect(screen.getByRole('article', { name: file.name })).toBeVisible();
|
|
fireEvent.click(screen.getByRole('button', { name: '重试处理' }));
|
|
await screen.findByText('等待处理');
|
|
expect(attempts).toHaveLength(2);
|
|
expect(attempts[0]).toEqual(attempts[1]);
|
|
expect(api.uploadKnowledge).toHaveBeenCalledTimes(1);
|
|
});
|
|
|
|
it('restores terminal task errors on entry and clears stale read errors after recovery', async () => {
|
|
items = [{ ...file, processing_task: { task_id: 'task', status: 'failed', error: '文档读取失败' } }];
|
|
await open();
|
|
expect(await screen.findByText('处理失败')).toBeVisible();
|
|
expect(screen.getByText('文档读取失败')).toBeVisible();
|
|
vi.useFakeTimers();
|
|
api.call.mockRejectedValueOnce(new Error('读取超时'));
|
|
await act(async () => fireEvent.click(screen.getByRole('button', { name: '刷新状态' })));
|
|
expect(screen.getByRole('alert')).toHaveTextContent('以下保留上次结果');
|
|
expect(screen.getByRole('article')).toBeVisible();
|
|
await tick();
|
|
expect(screen.queryByRole('alert')).not.toBeInTheDocument();
|
|
});
|
|
|
|
it.each(['success', 'failed'] as const)('uses a newly observed %s Task to resolve an uncertain submission', async status => {
|
|
items = [{ ...file, processing_task: { task_id: 'old-task', status: 'failed', error: '上次失败' } }];
|
|
const attempts: { operation_id: string }[] = [];
|
|
const original = api.call.getMockImplementation()!;
|
|
api.call.mockImplementation(async (operation, input) => {
|
|
if (operation === 'processKnowledge') {
|
|
attempts.push(input);
|
|
if (attempts.length < 3) throw new Error('连接超时');
|
|
}
|
|
return original(operation, input);
|
|
});
|
|
await open();
|
|
await screen.findByText('处理失败');
|
|
vi.useFakeTimers();
|
|
await act(async () => fireEvent.click(screen.getByRole('button', { name: '重试处理' })));
|
|
await tick(); // 旧 Task 仍可见时，保留同一请求身份。
|
|
expect(screen.getByRole('alert')).toHaveTextContent('处理请求尚未确认');
|
|
await act(async () => fireEvent.click(screen.getByRole('button', { name: '重试处理' })));
|
|
expect(attempts[1].operation_id).toBe(attempts[0].operation_id);
|
|
items = [{ ...file, status: status === 'success' ? 'indexed' : 'uploaded',
|
|
available: status === 'success', chunk_count: status === 'success' ? 3 : 0,
|
|
processing_task: { task_id: 'new-task', status, error: status === 'failed' ? '新处理失败' : null } }];
|
|
await tick();
|
|
expect(screen.queryByRole('alert')).not.toBeInTheDocument();
|
|
if (status === 'success') {
|
|
expect(screen.getByText('可用于回答')).toBeVisible();
|
|
expect(screen.queryByRole('button', { name: '重试处理' })).not.toBeInTheDocument();
|
|
} else {
|
|
expect(screen.getByText('新处理失败')).toBeVisible();
|
|
await act(async () => fireEvent.click(screen.getByRole('button', { name: '重试处理' })));
|
|
expect(attempts[2].operation_id).not.toBe(attempts[0].operation_id);
|
|
expect(screen.getByText('等待处理')).toBeVisible();
|
|
}
|
|
});
|
|
|
|
it('keeps all loaded pages refreshing and ignores a late response from the previous knowledge base', async () => {
|
|
let late!: (value: unknown) => void;
|
|
let useLate = false;
|
|
const original = api.call.getMockImplementation()!;
|
|
const second = { ...file, file_id: 'second', name: '第二页.txt' };
|
|
api.call.mockImplementation(async (operation, input) => {
|
|
if (operation !== 'knowledgeFiles') return original(operation, input);
|
|
if (input.kb_id === 'other') return { files: [], next_offset: null };
|
|
if (useLate) return new Promise(resolve => { late = resolve; });
|
|
return input.offset ? { files: [second], next_offset: null } : { files: [file], next_offset: 100 };
|
|
});
|
|
await open();
|
|
const more = await screen.findByRole('button', { name: '加载更多文档' });
|
|
vi.useFakeTimers();
|
|
await act(async () => fireEvent.click(more));
|
|
expect(screen.getByRole('article', { name: '第二页.txt' })).toBeVisible();
|
|
second.status = 'indexed'; second.available = true; second.chunk_count = 6;
|
|
await tick();
|
|
expect(within(screen.getByRole('article', { name: '第二页.txt' })).getByText('可用于回答')).toBeVisible();
|
|
useLate = true;
|
|
await tick();
|
|
await act(async () => fireEvent.change(screen.getByLabelText('管理的知识库'), { target: { value: 'other' } }));
|
|
await act(async () => late({ files: [file], next_offset: null }));
|
|
expect(screen.queryByRole('article')).not.toBeInTheDocument();
|
|
expect(screen.getByText('上传第一份文档，让智能体用你的资料回答。')).toBeVisible();
|
|
});
|
|
|
|
it('imports an attachment and processes it without claiming immediate readiness', async () => {
|
|
const original = api.call.getMockImplementation()!;
|
|
api.call.mockImplementation(async (operation, input) => {
|
|
if (operation === 'threads') return { threads: [{ thread_id: 'thread', title: '我的对话' }], next_offset: null };
|
|
if (operation === 'attachments') return { attachments: [{ file_id: 'attachment', file_name: '附件.pdf' }] };
|
|
if (operation === 'importKnowledgeAttachment') { items = [file]; return file; }
|
|
return original(operation, input);
|
|
});
|
|
await open();
|
|
fireEvent.click(screen.getByRole('button', { name: '从对话附件导入' }));
|
|
await screen.findByRole('option', { name: '我的对话' });
|
|
fireEvent.change(screen.getByLabelText('附件来源对话'), { target: { value: 'thread' } });
|
|
await screen.findByRole('option', { name: '附件.pdf' });
|
|
fireEvent.change(screen.getByLabelText('入库附件'), { target: { value: 'attachment' } });
|
|
fireEvent.click(screen.getByRole('button', { name: '确认导入知识库' }));
|
|
await screen.findByText('等待处理');
|
|
expect(screen.queryByText('可用于回答')).not.toBeInTheDocument();
|
|
});
|
|
|
|
it('keeps indexed content available when replacement cleanup fails and permits retry', async () => {
|
|
items = [{ ...file, status: 'indexed', chunk_count: 4, available: true, replaces_file_id: 'old',
|
|
processing_task: { task_id: 'task', status: 'failed', error: '旧文档清理失败' } }];
|
|
await open();
|
|
expect(await screen.findByText('内容已可用，处理未全部完成')).toBeVisible();
|
|
expect(screen.getByText(/已显示 1 份文档 · 1 份可用/)).toBeVisible();
|
|
expect(screen.getByText(/4 段可检索内容/)).toBeVisible();
|
|
expect(screen.getByText(/旧文档清理失败/)).toBeVisible();
|
|
vi.useFakeTimers();
|
|
await act(async () => fireEvent.click(screen.getByRole('button', { name: '重试处理' })));
|
|
expect(screen.getByText('内容已可用，正在完成处理')).toBeVisible();
|
|
expect(screen.getByText(/已显示 1 份文档 · 1 份可用/)).toBeVisible();
|
|
expect(screen.getByText(/4 段可检索内容/)).toBeVisible();
|
|
items = [{ ...items[0], processing_task: { task_id: 'task', status: 'running', error: null } }];
|
|
await tick();
|
|
expect(screen.getByText('内容已可用，正在完成处理')).toBeVisible();
|
|
expect(screen.getByText(/已显示 1 份文档 · 1 份可用/)).toBeVisible();
|
|
expect(screen.getByText(/4 段可检索内容/)).toBeVisible();
|
|
items = [{ ...items[0], processing_task: { task_id: 'task', status: 'success', error: null } }];
|
|
await tick();
|
|
expect(screen.getByText('可用于回答')).toBeVisible();
|
|
expect(api.call).toHaveBeenCalledWith('processKnowledge', expect.objectContaining({ file_id: 'file' }));
|
|
expect(api.uploadKnowledge).not.toHaveBeenCalled();
|
|
});
|
|
|
|
it('automatically processes a replacement while retaining the usable old document on failure', async () => {
|
|
items = [{ ...file, status: 'indexed', chunk_count: 4, available: true }];
|
|
const replacement = { ...file, file_id: 'new', name: '新版.pdf', replaces_file_id: 'file' };
|
|
api.uploadKnowledge.mockImplementation(async () => { items = [...items, replacement]; return replacement; });
|
|
const original = api.call.getMockImplementation()!;
|
|
api.call.mockImplementation(async (operation, input) => {
|
|
if (operation === 'processKnowledge') {
|
|
items = items.map(value => value.file_id === 'new' ? { ...replacement, status: 'error_parsing', processing_task: { task_id: 'task', status: 'failed', error: '文字读取失败' } } : value);
|
|
return { task_id: 'task' };
|
|
}
|
|
return original(operation, input);
|
|
});
|
|
await open();
|
|
fireEvent.click(await screen.findByRole('button', { name: '替换文档' }));
|
|
await screen.findByText('处理失败');
|
|
expect(within(screen.getByRole('article', { name: file.name })).getByText('可用于回答')).toBeVisible();
|
|
expect(screen.getByText(/这是替换文档/)).toBeVisible();
|
|
expect(api.uploadKnowledge).toHaveBeenCalledWith('ml-agent', 'kb', expect.any(String), 'file');
|
|
});
|