test: count final tag reply callbacks

This commit is contained in:
Shakker
2026-05-12 04:10:35 +01:00
parent ba152d4b63
commit df87c2fc56

View File

@@ -27,7 +27,7 @@ describe("subscribeEmbeddedPiSession", () => {
emit({ type: "message_start", message: { role: "assistant" } });
emitAssistantTextDelta({ emit, delta: "<final>Hi there</final>" });
expect(onPartialReply).toHaveBeenCalled();
expect(onPartialReply).toHaveBeenCalledTimes(1);
const firstPayload = onPartialReply.mock.calls[0][0];
expect(firstPayload.text).toBe("Hi there");
@@ -76,7 +76,7 @@ describe("subscribeEmbeddedPiSession", () => {
emit({ type: "message_start", message: { role: "assistant" } });
emitAssistantTextDelta({ emit, delta: "<final>Hello world</final>" });
expect(onPartialReply).toHaveBeenCalled();
expect(onPartialReply).toHaveBeenCalledTimes(1);
expect(onPartialReply.mock.calls[0][0].text).toBe("Hello world");
});
@@ -269,7 +269,7 @@ describe("subscribeEmbeddedPiSession", () => {
emit({ type: "message_end", message: assistantMessage });
await Promise.resolve();
expect(onBlockReply).toHaveBeenCalled();
expect(onBlockReply).toHaveBeenCalledTimes(1);
const payload = onBlockReply.mock.calls[0][0];
expect(payload.text).toBe("Hello block");
});