github/openclaw
1
1
Fork 0
mirror of https://github.com/openclaw/openclaw.git synced 2026-02-19 18:39:20 -05:00
Code Issues Packages Projects Releases Wiki Activity

test(gateway): dedupe assistant chat event assertions

Browse Source
This commit is contained in:
Peter Steinberger
2026-02-19 08:14:21 +00:00
parent 644d037969
commit 8bb1747ad9
1 changed files with 34 additions and 30 deletions
Download Patch File Download Diff File Expand all files Collapse all files

64
src/gateway/server-chat.agent-events.test.ts
View File

@@ -43,21 +43,38 @@ describe("agent event handler", () => {
};
}
it("emits chat delta for assistant text-only events", () => {
const { broadcast, nodeSendToSession, chatRunState, handler, nowSpy } = createHarness({
now: 1_000,
function emitRun1AssistantText(
harness: ReturnType<typeof createHarness>,
text: string,
): ReturnType<typeof createHarness> {
harness.chatRunState.registry.add("run-1", {
sessionKey: "session-1",
clientRunId: "client-1",
});
chatRunState.registry.add("run-1", { sessionKey: "session-1", clientRunId: "client-1" });
handler({
harness.handler({
runId: "run-1",
seq: 1,
stream: "assistant",
ts: Date.now(),
data: { text: "Hello world" },
data: { text },
});
return harness;
}
const chatCalls = broadcast.mock.calls.filter(([event]) => event === "chat");
function chatBroadcastCalls(broadcast: ReturnType<typeof vi.fn>) {
return broadcast.mock.calls.filter(([event]) => event === "chat");
}
function sessionChatCalls(nodeSendToSession: ReturnType<typeof vi.fn>) {
return nodeSendToSession.mock.calls.filter(([, event]) => event === "chat");
}
it("emits chat delta for assistant text-only events", () => {
const { broadcast, nodeSendToSession, nowSpy } = emitRun1AssistantText(
createHarness({ now: 1_000 }),
"Hello world",
);
const chatCalls = chatBroadcastCalls(broadcast);
expect(chatCalls).toHaveLength(1);
const payload = chatCalls[0]?.[1] as {
state?: string;
@@ -65,29 +82,17 @@ describe("agent event handler", () => {
};
expect(payload.state).toBe("delta");
expect(payload.message?.content?.[0]?.text).toBe("Hello world");
const sessionChatCalls = nodeSendToSession.mock.calls.filter(([, event]) => event === "chat");
expect(sessionChatCalls).toHaveLength(1);
expect(sessionChatCalls(nodeSendToSession)).toHaveLength(1);
nowSpy?.mockRestore();
});
it("does not emit chat delta for NO_REPLY streaming text", () => {
const { broadcast, nodeSendToSession, chatRunState, handler, nowSpy } = createHarness({
now: 1_000,
});
chatRunState.registry.add("run-1", { sessionKey: "session-1", clientRunId: "client-1" });
handler({
runId: "run-1",
seq: 1,
stream: "assistant",
ts: Date.now(),
data: { text: " NO_REPLY " },
});
const chatCalls = broadcast.mock.calls.filter(([event]) => event === "chat");
expect(chatCalls).toHaveLength(0);
const sessionChatCalls = nodeSendToSession.mock.calls.filter(([, event]) => event === "chat");
expect(sessionChatCalls).toHaveLength(0);
const { broadcast, nodeSendToSession, nowSpy } = emitRun1AssistantText(
createHarness({ now: 1_000 }),
" NO_REPLY ",
);
expect(chatBroadcastCalls(broadcast)).toHaveLength(0);
expect(sessionChatCalls(nodeSendToSession)).toHaveLength(0);
nowSpy?.mockRestore();
});
@@ -112,13 +117,12 @@ describe("agent event handler", () => {
data: { phase: "end" },
});
const chatCalls = broadcast.mock.calls.filter(([event]) => event === "chat");
const chatCalls = chatBroadcastCalls(broadcast);
expect(chatCalls).toHaveLength(1);
const payload = chatCalls[0]?.[1] as { state?: string; message?: unknown };
expect(payload.state).toBe("final");
expect(payload.message).toBeUndefined();
const sessionChatCalls = nodeSendToSession.mock.calls.filter(([, event]) => event === "chat");
expect(sessionChatCalls).toHaveLength(1);
expect(sessionChatCalls(nodeSendToSession)).toHaveLength(1);
nowSpy?.mockRestore();
});