From df87c2fc56f18c58ff7e01cd3e5edecaa7535a6d Mon Sep 17 00:00:00 2001
From: Shakker <shakkerdroid@gmail.com>
Date: Tue, 12 May 2026 04:10:35 +0100
Subject: [PATCH] test: count final tag reply callbacks

---
 ...ilters-final-suppresses-output-without-start-tag.test.ts | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/src/agents/pi-embedded-subscribe.subscribe-embedded-pi-session.filters-final-suppresses-output-without-start-tag.test.ts b/src/agents/pi-embedded-subscribe.subscribe-embedded-pi-session.filters-final-suppresses-output-without-start-tag.test.ts
index 1fe7b2c596d..af18698e5be 100644
--- a/src/agents/pi-embedded-subscribe.subscribe-embedded-pi-session.filters-final-suppresses-output-without-start-tag.test.ts
+++ b/src/agents/pi-embedded-subscribe.subscribe-embedded-pi-session.filters-final-suppresses-output-without-start-tag.test.ts
@@ -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");
   });