247 lines
7.9 KiB
TypeScript
247 lines
7.9 KiB
TypeScript
/**
|
||
* 单元测试: ComputerServer — SSE 事件缓冲（ACP 引擎性能优化）
|
||
*
|
||
* 覆盖内容：
|
||
* - pushSseEvent 在无客户端时写入缓冲
|
||
* - 缓冲条数上限 SSE_EVENT_BUFFER_MAX
|
||
* - getSseEventBufferSize 查询
|
||
*/
|
||
|
||
import { describe, it, expect, vi, beforeEach } from "vitest";
|
||
import {
|
||
pushSseEvent,
|
||
getSseEventBufferSize,
|
||
clearSseEventBuffer,
|
||
clearAllSseEventBuffers,
|
||
setSessionFirstTokenContextForTest,
|
||
hasSessionFirstTokenContext,
|
||
} from "./computerServer";
|
||
|
||
const mockPerfLogger = vi.hoisted(() => ({
|
||
info: vi.fn(),
|
||
}));
|
||
const mockFirstTokenTrace = vi.hoisted(() => ({
|
||
trace: vi.fn(),
|
||
}));
|
||
|
||
// 避免拉起 unifiedAgent 与 Electron 等重模块
|
||
vi.mock("./engines/unifiedAgent", () => ({
|
||
agentService: {
|
||
isReady: true,
|
||
getEngineType: () => "qimingcode",
|
||
getAgentConfig: () => ({ workspaceDir: "/tmp" }),
|
||
getAcpEngine: () => null,
|
||
hasRunningEngines: false,
|
||
getEngineForProject: () => null,
|
||
},
|
||
}));
|
||
vi.mock("electron-log", () => ({
|
||
default: {
|
||
info: vi.fn(),
|
||
warn: vi.fn(),
|
||
error: vi.fn(),
|
||
debug: vi.fn(),
|
||
},
|
||
}));
|
||
vi.mock("./constants", () => ({
|
||
LOCALHOST_HOSTNAME: "127.0.0.1",
|
||
APP_DATA_DIR_NAME: ".qimingclaw",
|
||
LOGS_DIR_NAME: "logs",
|
||
}));
|
||
vi.mock("./startupPorts", () => ({
|
||
getConfiguredPorts: () => ({ fileServer: 0 }),
|
||
}));
|
||
vi.mock("./utils/logRedact", () => ({
|
||
redactForLog: (x: unknown) => x,
|
||
redactStringForLog: (s: string) => s,
|
||
}));
|
||
vi.mock("../bootstrap/logConfig", () => ({
|
||
getPerfLogger: () => mockPerfLogger,
|
||
}));
|
||
vi.mock("./engines/perf/firstTokenTrace", () => ({
|
||
firstTokenTrace: mockFirstTokenTrace,
|
||
}));
|
||
|
||
describe("ComputerServer — SSE 事件缓冲", () => {
|
||
const sessionId = "ses-test-buffer-001";
|
||
|
||
beforeEach(() => {
|
||
vi.clearAllMocks();
|
||
clearAllSseEventBuffers();
|
||
// 清空该 session 的缓冲（通过多次 push 触发内部逻辑会累积，这里依赖模块状态）
|
||
// 注意：pushSseEvent 依赖 computerServer 模块内的 sseClients/sseEventBuffers
|
||
// 无客户端时会写入 buffer；测试间若复用 sessionId 会累积，故用唯一 sessionId 或单测内只测一次
|
||
});
|
||
|
||
it("getSseEventBufferSize 在无缓冲时返回 0", () => {
|
||
expect(getSseEventBufferSize("nonexistent-session")).toBe(0);
|
||
});
|
||
|
||
it("无客户端时 pushSseEvent 将事件写入缓冲", () => {
|
||
pushSseEvent(sessionId, "prompt_start", {
|
||
sessionId,
|
||
messageType: "promptStart",
|
||
});
|
||
expect(getSseEventBufferSize(sessionId)).toBe(1);
|
||
pushSseEvent(sessionId, "message_part", { type: "text", text: "hi" });
|
||
expect(getSseEventBufferSize(sessionId)).toBe(2);
|
||
});
|
||
|
||
it("缓冲最多保留 SSE_EVENT_BUFFER_MAX 条", () => {
|
||
const SSE_EVENT_BUFFER_MAX = 50;
|
||
for (let i = 0; i < SSE_EVENT_BUFFER_MAX + 10; i++) {
|
||
pushSseEvent(sessionId, "ev", { i });
|
||
}
|
||
expect(getSseEventBufferSize(sessionId)).toBe(SSE_EVENT_BUFFER_MAX);
|
||
});
|
||
|
||
it("TTL 过期后同 session 再次 push 会 prune 掉旧 buffer 并新建", () => {
|
||
const ttlSessionId = "ses-ttl-only";
|
||
vi.useFakeTimers({ now: 0 });
|
||
pushSseEvent(ttlSessionId, "ev1", {});
|
||
expect(getSseEventBufferSize(ttlSessionId)).toBe(1);
|
||
vi.advanceTimersByTime(31000); // 超过 30s TTL
|
||
pushSseEvent(ttlSessionId, "ev2", {});
|
||
// 旧 buffer 被 prune，当前 push 写入新 buffer，应只有 1 条
|
||
expect(getSseEventBufferSize(ttlSessionId)).toBe(1);
|
||
vi.useRealTimers();
|
||
});
|
||
|
||
it("clearSseEventBuffer 清除指定 session 的缓冲", () => {
|
||
const ses1 = "ses-clear-001";
|
||
const ses2 = "ses-clear-002";
|
||
pushSseEvent(ses1, "ev", { data: 1 });
|
||
pushSseEvent(ses1, "ev", { data: 2 });
|
||
pushSseEvent(ses2, "ev", { data: 3 });
|
||
expect(getSseEventBufferSize(ses1)).toBe(2);
|
||
expect(getSseEventBufferSize(ses2)).toBe(1);
|
||
|
||
clearSseEventBuffer(ses1);
|
||
expect(getSseEventBufferSize(ses1)).toBe(0);
|
||
expect(getSseEventBufferSize(ses2)).toBe(1); // ses2 不受影响
|
||
});
|
||
|
||
it("clearSseEventBuffer 对不存在的 session 是幂等的", () => {
|
||
expect(() => clearSseEventBuffer("nonexistent")).not.toThrow();
|
||
expect(getSseEventBufferSize("nonexistent")).toBe(0);
|
||
});
|
||
|
||
it("clearSseEventBuffer 传入空字符串/null/undefined 不报错", () => {
|
||
expect(() => clearSseEventBuffer("")).not.toThrow();
|
||
expect(() => clearSseEventBuffer(null as any)).not.toThrow();
|
||
expect(() => clearSseEventBuffer(undefined as any)).not.toThrow();
|
||
});
|
||
|
||
it("clearAllSseEventBuffers 清除所有 session 的缓冲", () => {
|
||
const ses1 = "ses-clearall-001";
|
||
const ses2 = "ses-clearall-002";
|
||
const ses3 = "ses-clearall-003";
|
||
pushSseEvent(ses1, "ev", {});
|
||
pushSseEvent(ses2, "ev", {});
|
||
pushSseEvent(ses3, "ev", {});
|
||
expect(getSseEventBufferSize(ses1)).toBe(1);
|
||
expect(getSseEventBufferSize(ses2)).toBe(1);
|
||
expect(getSseEventBufferSize(ses3)).toBe(1);
|
||
|
||
clearAllSseEventBuffers();
|
||
expect(getSseEventBufferSize(ses1)).toBe(0);
|
||
expect(getSseEventBufferSize(ses2)).toBe(0);
|
||
expect(getSseEventBufferSize(ses3)).toBe(0);
|
||
});
|
||
|
||
it("clearAllSseEventBuffers 对空缓冲也是幂等的", () => {
|
||
clearAllSseEventBuffers(); // 清空
|
||
expect(() => clearAllSseEventBuffers()).not.toThrow(); // 再次清空
|
||
});
|
||
|
||
it("首个真实 token 输出 /chat.firstToken 与 /chat.newSession.firstToken", () => {
|
||
vi.useFakeTimers({ now: 10_000 });
|
||
const session = "ses-ttft-new-001";
|
||
setSessionFirstTokenContextForTest(session, {
|
||
requestId: "rid-ttft-new-001",
|
||
projectId: "project-ttft",
|
||
engine: "qimingcode",
|
||
chatReceivedAt: 8_500,
|
||
isNewSession: true,
|
||
});
|
||
|
||
pushSseEvent(session, "agent_message_chunk", {
|
||
data: { content: { text: "hello" } },
|
||
});
|
||
|
||
const perfLogs = mockPerfLogger.info.mock.calls.map((call) => call[0]);
|
||
expect(
|
||
perfLogs.some((msg: string) =>
|
||
msg.includes("[PERF] /chat.firstToken: 1500ms"),
|
||
),
|
||
).toBe(true);
|
||
expect(
|
||
perfLogs.some((msg: string) =>
|
||
msg.includes("[PERF] /chat.newSession.firstToken: 1500ms"),
|
||
),
|
||
).toBe(true);
|
||
expect(mockFirstTokenTrace.trace).toHaveBeenCalledWith(
|
||
"chat.first_token.returned",
|
||
expect.objectContaining({
|
||
requestId: "rid-ttft-new-001",
|
||
sessionId: session,
|
||
projectId: "project-ttft",
|
||
engine: "qimingcode",
|
||
}),
|
||
expect.objectContaining({
|
||
ttftMs: 1500,
|
||
isNewSession: true,
|
||
}),
|
||
);
|
||
|
||
// 在 end_turn 前上下文仍保留，便于统计后续阶段
|
||
expect(hasSessionFirstTokenContext(session)).toBe(true);
|
||
pushSseEvent(session, "end_turn", {});
|
||
expect(hasSessionFirstTokenContext(session)).toBe(false);
|
||
vi.useRealTimers();
|
||
});
|
||
|
||
it("复用会话只输出 /chat.firstToken，不输出 newSession 专属日志", () => {
|
||
vi.useFakeTimers({ now: 30_000 });
|
||
const session = "ses-ttft-reuse-001";
|
||
setSessionFirstTokenContextForTest(session, {
|
||
requestId: "rid-ttft-reuse-001",
|
||
projectId: "project-reuse",
|
||
engine: "qimingcode",
|
||
chatReceivedAt: 29_200,
|
||
isNewSession: false,
|
||
});
|
||
|
||
pushSseEvent(session, "agent_message_chunk", {
|
||
data: { content: { text: "reuse" } },
|
||
});
|
||
|
||
const perfLogs = mockPerfLogger.info.mock.calls.map((call) => call[0]);
|
||
expect(
|
||
perfLogs.some((msg: string) =>
|
||
msg.includes("[PERF] /chat.firstToken: 800ms"),
|
||
),
|
||
).toBe(true);
|
||
expect(
|
||
perfLogs.some((msg: string) =>
|
||
msg.includes("/chat.newSession.firstToken"),
|
||
),
|
||
).toBe(false);
|
||
vi.useRealTimers();
|
||
});
|
||
|
||
it("clearSseEventBuffer 会清理对应 session 的首字上下文", () => {
|
||
const session = "ses-clear-context-001";
|
||
setSessionFirstTokenContextForTest(session, {
|
||
requestId: "rid-clear-ctx-001",
|
||
projectId: "project-clear",
|
||
engine: "qimingcode",
|
||
chatReceivedAt: Date.now(),
|
||
isNewSession: true,
|
||
});
|
||
expect(hasSessionFirstTokenContext(session)).toBe(true);
|
||
clearSseEventBuffer(session);
|
||
expect(hasSessionFirstTokenContext(session)).toBe(false);
|
||
});
|
||
});
|