} from "./pi-embedded-subscribe.e2e-harness.js";

import { subscribeEmbeddedPiSession } from "./pi-embedded-subscribe.js";

type ReplyMock = ReturnType<typeof vi.fn>;

type ReplyPayload = { text?: string };

function requireFirstReplyPayload(mock: ReplyMock): ReplyPayload {

const call = mock.mock.calls[0];

if (!call) {

throw new Error("expected first reply call");

}

const payload = call[0];

if (!payload || typeof payload !== "object") {

throw new Error("expected first reply payload");

}

return payload as ReplyPayload;

}

describe("subscribeEmbeddedPiSession", () => {

it("filters to <final> and suppresses output without a start tag", () => {

const { session, emit } = createStubSessionHarness();

emitAssistantTextDelta({ emit, delta: "<final>Hi there</final>" });

expect(onPartialReply).toHaveBeenCalledTimes(1);

const firstPayload = onPartialReply.mock.calls.at(0)?.[0];

const firstPayload = requireFirstReplyPayload(onPartialReply);

expect(firstPayload?.text).toBe("Hi there");

onPartialReply.mockClear();

emitAssistantTextDelta({ emit, delta: "<final>Hello world</final>" });

expect(onPartialReply).toHaveBeenCalledTimes(1);

expect(onPartialReply.mock.calls.at(0)?.[0]?.text).toBe("Hello world");

expect(requireFirstReplyPayload(onPartialReply).text).toBe("Hello world");

});

it("strips final tags split across streamed deltas without emitting tag remnants", () => {

await Promise.resolve();

expect(onBlockReply).toHaveBeenCalledTimes(1);

expect(onBlockReply.mock.calls.at(0)?.[0]?.text).toBe("Answer ends with <fi");

expect(requireFirstReplyPayload(onBlockReply).text).toBe("Answer ends with <fi");

});

it("keeps a trailing final-tag prefix when synchronous message_end drains chunked text_end replies", async () => {

emitAssistantTextDelta({ emit, delta: "Hello world" });

const payload = onPartialReply.mock.calls.at(0)?.[0];

const payload = requireFirstReplyPayload(onPartialReply);

expect(payload?.text).toBe("Hello world");

});

it("emits block replies on message_end", async () => {

await Promise.resolve();

expect(onBlockReply).toHaveBeenCalledTimes(1);

const payload = onBlockReply.mock.calls.at(0)?.[0];

const payload = requireFirstReplyPayload(onBlockReply);

expect(payload?.text).toBe("Hello block");

});

});