feat: 用自定义侧边栏替换聊天室 Conversations 组件,提取公共 SidebarGroup 和 date-group
This commit is contained in:
177
tests/web/components/ChatPage.test.tsx
Normal file
177
tests/web/components/ChatPage.test.tsx
Normal file
@@ -0,0 +1,177 @@
|
||||
import { fireEvent, screen, waitFor } from "@testing-library/react";
|
||||
import { describe, expect, test, vi } from "bun:test";
|
||||
import { createElement } from "react";
|
||||
|
||||
import type { Model, Project } from "../../../src/shared/api";
|
||||
|
||||
import { ChatPage } from "../../../src/web/features/chat/ChatPage";
|
||||
import { ProjectContext } from "../../../src/web/shared/hooks/use-current-project";
|
||||
import { installFetchMock, jsonResponse, renderWithProviders } from "../test-utils";
|
||||
|
||||
const PROJECT_ID = "proj-1";
|
||||
|
||||
const MOCK_PROJECT: Project = {
|
||||
archivedAt: null,
|
||||
createdAt: "2026-01-01T00:00:00.000Z",
|
||||
description: "",
|
||||
id: PROJECT_ID,
|
||||
name: "测试项目",
|
||||
status: "active",
|
||||
updatedAt: "2026-01-01T00:00:00.000Z",
|
||||
};
|
||||
|
||||
const TEXT_MODEL: Model = {
|
||||
capabilities: ["text"],
|
||||
contextLength: null,
|
||||
createdAt: "2024-01-01T00:00:00.000Z",
|
||||
id: "model-1",
|
||||
maxOutputTokens: null,
|
||||
modelId: "gpt-4o",
|
||||
name: "GPT-4o",
|
||||
providerId: "pv1",
|
||||
updatedAt: "2024-01-01T00:00:00.000Z",
|
||||
};
|
||||
|
||||
const CONVERSATION = {
|
||||
createdAt: "2026-06-03T00:00:00.000Z",
|
||||
id: "conv-1",
|
||||
modelId: "model-1",
|
||||
projectId: PROJECT_ID,
|
||||
title: "测试对话",
|
||||
updatedAt: "2026-06-03T00:00:00.000Z",
|
||||
};
|
||||
|
||||
function renderChatPage() {
|
||||
return renderWithProviders(
|
||||
createElement(ProjectContext.Provider, {
|
||||
children: createElement(ChatPage),
|
||||
value: MOCK_PROJECT,
|
||||
}),
|
||||
);
|
||||
}
|
||||
|
||||
function setupFetchMock() {
|
||||
return installFetchMock((call) => {
|
||||
if (call.url.includes("/models")) {
|
||||
return jsonResponse({ items: [TEXT_MODEL], total: 1 });
|
||||
}
|
||||
if (call.url.includes("/conversations") && call.method === "GET") {
|
||||
return jsonResponse({ items: [CONVERSATION], page: 1, pageSize: 200, total: 1 });
|
||||
}
|
||||
if (call.url.endsWith("/conversations") && call.method === "POST") {
|
||||
return jsonResponse({ conversation: { ...CONVERSATION, id: "conv-new", title: "新会话" } }, { status: 201 });
|
||||
}
|
||||
if (call.method === "DELETE" && call.url.includes("/conversations/")) {
|
||||
return new Response(null, { status: 204 });
|
||||
}
|
||||
if (call.url.includes("/messages")) {
|
||||
return jsonResponse({ items: [], total: 0 });
|
||||
}
|
||||
if (/\/conversations\/conv-1$/.exec(call.url)) {
|
||||
return jsonResponse({ conversation: CONVERSATION });
|
||||
}
|
||||
return jsonResponse({ error: "not found" }, { status: 404 });
|
||||
});
|
||||
}
|
||||
|
||||
void vi.mock("@ai-sdk/react", () => ({
|
||||
useChat: () => ({
|
||||
messages: [],
|
||||
regenerate: () => undefined,
|
||||
sendMessage: () => undefined,
|
||||
setMessages: (msgs: unknown) => msgs,
|
||||
status: "ready",
|
||||
stop: () => undefined,
|
||||
}),
|
||||
}));
|
||||
|
||||
void vi.mock("ai", () => ({
|
||||
DefaultChatTransport: function () {
|
||||
return undefined;
|
||||
},
|
||||
}));
|
||||
|
||||
describe("ChatPage", () => {
|
||||
test("渲染对话侧边栏和欢迎页", async () => {
|
||||
setupFetchMock();
|
||||
renderChatPage();
|
||||
|
||||
await waitFor(() => {
|
||||
expect(screen.getByText("测试对话")).not.toBeNull();
|
||||
});
|
||||
|
||||
expect(screen.getByText("你好,我是阿福")).not.toBeNull();
|
||||
});
|
||||
|
||||
test("点击新对话按钮创建并选中对话", async () => {
|
||||
const calls = setupFetchMock();
|
||||
renderChatPage();
|
||||
|
||||
await waitFor(() => {
|
||||
expect(screen.getByText("新对话")).not.toBeNull();
|
||||
});
|
||||
|
||||
fireEvent.click(screen.getByText("新对话"));
|
||||
|
||||
await waitFor(() => {
|
||||
const createCall = calls.find((c) => c.url.endsWith("/conversations") && c.method === "POST");
|
||||
expect(createCall).toBeTruthy();
|
||||
});
|
||||
});
|
||||
|
||||
test("点击对话切换选中", async () => {
|
||||
setupFetchMock();
|
||||
renderChatPage();
|
||||
|
||||
await waitFor(() => {
|
||||
expect(screen.getByText("测试对话")).not.toBeNull();
|
||||
});
|
||||
|
||||
fireEvent.click(screen.getByText("测试对话"));
|
||||
|
||||
await waitFor(() => {
|
||||
expect(screen.queryByText("你好,我是阿福")).toBeNull();
|
||||
});
|
||||
});
|
||||
|
||||
test("删除对话后列表更新", async () => {
|
||||
let deleted = false;
|
||||
installFetchMock((call) => {
|
||||
if (call.method === "DELETE" && call.url.includes("/conversations/conv-1")) {
|
||||
deleted = true;
|
||||
return new Response(null, { status: 204 });
|
||||
}
|
||||
if (call.url.includes("/models")) {
|
||||
return jsonResponse({ items: [TEXT_MODEL], total: 1 });
|
||||
}
|
||||
if (call.url.includes("/conversations") && call.method === "GET") {
|
||||
if (deleted) {
|
||||
return jsonResponse({ items: [], page: 1, pageSize: 200, total: 0 });
|
||||
}
|
||||
return jsonResponse({ items: [CONVERSATION], page: 1, pageSize: 200, total: 1 });
|
||||
}
|
||||
if (call.url.includes("/messages")) {
|
||||
return jsonResponse({ items: [], total: 0 });
|
||||
}
|
||||
return jsonResponse({ error: "not found" }, { status: 404 });
|
||||
});
|
||||
|
||||
renderChatPage();
|
||||
|
||||
await waitFor(() => {
|
||||
expect(screen.getByText("测试对话")).not.toBeNull();
|
||||
});
|
||||
|
||||
fireEvent.click(screen.getByLabelText("删除"));
|
||||
|
||||
await waitFor(() => {
|
||||
expect(screen.getByText("确认删除该对话?")).not.toBeNull();
|
||||
});
|
||||
|
||||
fireEvent.click(screen.getByText("删 除"));
|
||||
|
||||
await waitFor(() => {
|
||||
expect(screen.getByText("暂无对话")).not.toBeNull();
|
||||
});
|
||||
});
|
||||
});
|
||||
Reference in New Issue
Block a user