vultr/openclaw
1
0
Fork 0
mirror of https://github.com/openclaw/openclaw.git synced 2026-05-18 18:04:45 +00:00
Code Issues Packages Projects Releases Wiki Activity

test: dedupe embedded subscribe flush mock reads

Browse Source
This commit is contained in:
Peter Steinberger
2026-05-13 04:25:31 +01:00
parent 9831b28dd7
commit f6d093e75e
1 changed files with 10 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
src/agents/pi-embedded-subscribe.subscribe-embedded-pi-session.calls-onblockreplyflush-before-tool-execution-start-preserve.test.ts
View File

@@ -5,6 +5,14 @@ import {
} from "./pi-embedded-subscribe.e2e-harness.js";
import { subscribeEmbeddedPiSession } from "./pi-embedded-subscribe.js";
function firstBlockReplyText(onBlockReply: ReturnType<typeof vi.fn>): string | undefined {
const firstCall = onBlockReply.mock.calls[0];
if (!firstCall) {
throw new Error("expected onBlockReply to be called");
}
return firstCall[0]?.text;
}
describe("subscribeEmbeddedPiSession", () => {
it("calls onBlockReplyFlush before tool_execution_start to preserve message boundaries", () => {
const { session, emit } = createStubSessionHarness();
@@ -83,7 +91,7 @@ describe("subscribeEmbeddedPiSession", () => {
await Promise.resolve();
expect(onBlockReply).toHaveBeenCalledTimes(1);
expect(onBlockReply.mock.calls.at(0)?.[0]?.text).toBe("Short chunk.");
expect(firstBlockReplyText(onBlockReply)).toBe("Short chunk.");
expect(onBlockReplyFlush).toHaveBeenCalledTimes(1);
});
@@ -157,7 +165,7 @@ describe("subscribeEmbeddedPiSession", () => {
await Promise.resolve();
expect(onBlockReply).toHaveBeenCalledTimes(1);
expect(onBlockReply.mock.calls.at(0)?.[0]?.text).toBe("Final reply before lifecycle end.");
expect(firstBlockReplyText(onBlockReply)).toBe("Final reply before lifecycle end.");
expect(onBlockReplyFlush).toHaveBeenCalledTimes(1);
});