111 lines
3.2 KiB
TypeScript
111 lines
3.2 KiB
TypeScript
import "@testing-library/jest-dom";
|
|
import { fireEvent, render, screen } from "@testing-library/react";
|
|
|
|
import { AgentProgressTimeline } from "./AgentProgressTimeline";
|
|
import type { ChatProgress } from "./GlobalChatbox.types";
|
|
|
|
describe("AgentProgressTimeline", () => {
|
|
beforeEach(() => {
|
|
jest.useFakeTimers();
|
|
jest.setSystemTime(new Date("2026-01-01T00:00:00.000Z"));
|
|
});
|
|
|
|
afterEach(() => {
|
|
jest.useRealTimers();
|
|
});
|
|
|
|
it("shows the running step and keeps the timeline expanded while running", () => {
|
|
const now = Date.now();
|
|
const progress: ChatProgress[] = [
|
|
{
|
|
id: "start",
|
|
phase: "start",
|
|
status: "running",
|
|
title: "收到请求",
|
|
startedAt: now - 5000,
|
|
elapsedMs: 5000,
|
|
elapsedSnapshotAt: now,
|
|
},
|
|
{
|
|
id: "tool",
|
|
phase: "tool",
|
|
status: "running",
|
|
title: "正在调用 tjwater_cli",
|
|
detail: "analysis bottlenecks",
|
|
startedAt: now - 1200,
|
|
elapsedMs: 1200,
|
|
elapsedSnapshotAt: now,
|
|
},
|
|
];
|
|
|
|
render(<AgentProgressTimeline progress={progress} />);
|
|
|
|
expect(screen.getByText(/Agent 过程:/)).toBeInTheDocument();
|
|
expect(screen.getByText(/耗时 5.0s/)).toBeInTheDocument();
|
|
expect(screen.getByText("查询后端数据")).toBeInTheDocument();
|
|
expect(screen.getByText("analysis bottlenecks")).toBeInTheDocument();
|
|
expect(screen.getByText("1.2s")).toBeInTheDocument();
|
|
});
|
|
|
|
it("summarizes completed steps and lets users expand details", async () => {
|
|
const progress: ChatProgress[] = [
|
|
{
|
|
id: "request-received",
|
|
phase: "start",
|
|
status: "completed",
|
|
title: "收到请求",
|
|
startedAt: Date.now() - 8000,
|
|
endedAt: Date.now(),
|
|
durationMs: 8000,
|
|
},
|
|
{
|
|
id: "done",
|
|
phase: "complete",
|
|
status: "completed",
|
|
title: "分析完成",
|
|
startedAt: Date.now() - 1000,
|
|
endedAt: Date.now(),
|
|
durationMs: 1000,
|
|
},
|
|
];
|
|
|
|
render(<AgentProgressTimeline progress={progress} />);
|
|
|
|
expect(screen.getByText(/已完成 \(2 步\)/)).toBeInTheDocument();
|
|
expect(screen.getByText(/耗时 8.0s/)).toBeInTheDocument();
|
|
expect(screen.queryByText("分析完成")).not.toBeVisible();
|
|
|
|
fireEvent.click(screen.getByText(/Agent 过程:/));
|
|
|
|
expect(screen.getByText("分析完成")).toBeVisible();
|
|
});
|
|
|
|
it("treats stale running steps as finished after a complete event", () => {
|
|
const progress: ChatProgress[] = [
|
|
{
|
|
id: "tool",
|
|
phase: "tool",
|
|
status: "completed",
|
|
title: "正在调用 tjwater_cli",
|
|
startedAt: Date.now() - 4000,
|
|
endedAt: Date.now(),
|
|
},
|
|
{
|
|
id: "done",
|
|
phase: "complete",
|
|
status: "completed",
|
|
title: "分析完成",
|
|
startedAt: Date.now() - 500,
|
|
endedAt: Date.now(),
|
|
durationMs: 500,
|
|
},
|
|
];
|
|
|
|
render(<AgentProgressTimeline progress={progress} />);
|
|
|
|
expect(screen.getByText(/已完成 \(2 步\)/)).toBeInTheDocument();
|
|
expect(screen.getByText("4.0s")).toBeInTheDocument();
|
|
expect(screen.queryByRole("progressbar")).not.toBeInTheDocument();
|
|
});
|
|
});
|