describe("tui-event-handlers: streaming watchdog", () => {

const expectedTimeoutMessage =

"This response is taking longer than expected. Send another message to continue.";

"This response is taking longer than expected. Still waiting for the current run.";

beforeEach(() => {

vi.useFakeTimers();

return { state, chatLog, tui, setActivityStatus, loadHistory, noteLocalRunId, handlers };

};

it("resets activityStatus to idle when no stream delta arrives for the watchdog window", () => {

it("keeps the active run busy when no stream delta arrives for the watchdog window", () => {

const { state, chatLog, setActivityStatus, handlers } = createHarness({

streamingWatchdogMs: 5_000,

});

vi.advanceTimersByTime(5_001);

expect(setActivityStatus).toHaveBeenLastCalledWith("idle");

expect(state.activeChatRunId).toBeNull();

expect(setActivityStatus).not.toHaveBeenCalledWith("idle");

expect(state.activeChatRunId).toBe("run-stuck");

expect(chatLog.addPendingSystem).toHaveBeenCalledWith("run-stuck", expectedTimeoutMessage);

handlers.dispose?.();

});

it("flushes a deferred history reload when the watchdog clears the active run", () => {

it("keeps deferred history reload pending while the watchdog waits on the active run", () => {

const { state, loadHistory, noteLocalRunId, setActivityStatus, handlers } = createHarness({

streamingWatchdogMs: 5_000,

});

vi.advanceTimersByTime(5_001);

expect(state.activeChatRunId).toBeNull();

expect(state.activityStatus).toBe("idle");

expect(setActivityStatus).toHaveBeenLastCalledWith("idle");

expect(loadHistory).toHaveBeenCalledTimes(1);

expect(state.activeChatRunId).toBe("run-stuck");

expect(setActivityStatus).not.toHaveBeenCalledWith("idle");

expect(loadHistory).not.toHaveBeenCalled();

handlers.dispose?.();

});

vi.advanceTimersByTime(2_500);

expect(setActivityStatus).toHaveBeenLastCalledWith("idle");

expect(state.activeChatRunId).toBeNull();

expect(setActivityStatus).not.toHaveBeenCalledWith("idle");

expect(state.activeChatRunId).toBe("run-flow");

handlers.dispose?.();

});

vi.advanceTimersByTime(2_001);

expect(setActivityStatus).toHaveBeenLastCalledWith("idle");

expect(state.activeChatRunId).toBeNull();

expect(setActivityStatus).not.toHaveBeenCalledWith("idle");

expect(state.activeChatRunId).toBe("run-tools");

handlers.dispose?.();

});

handlers.dispose?.();

});

it("does not let an older run steal the active run watchdog", () => {

it("does not let another run replace a watchdog-noticed active run", () => {

const { state, chatLog, setActivityStatus, handlers } = createHarness({

streamingWatchdogMs: 5_000,

});

} satisfies ChatEvent);

vi.advanceTimersByTime(5_001);

expect(state.activeChatRunId).toBeNull();

expect(state.activeChatRunId).toBe("run-old");

expect(chatLog.addPendingSystem).toHaveBeenCalledWith("run-old", expectedTimeoutMessage);

handlers.handleChatEvent({

runId: "run-new",

sessionKey: state.currentSessionKey,

state: "delta",

message: { content: "new" },

} satisfies ChatEvent);

expect(state.activeChatRunId).toBe("run-new");

expect(state.activeChatRunId).toBe("run-old");

vi.advanceTimersByTime(3_000);

vi.advanceTimersByTime(2_001);

expect(setActivityStatus).toHaveBeenLastCalledWith("idle");

expect(state.activeChatRunId).toBeNull();

expect(chatLog.addPendingSystem).toHaveBeenCalledTimes(2);

expect(setActivityStatus).not.toHaveBeenCalledWith("idle");

expect(state.activeChatRunId).toBe("run-old");

expect(chatLog.addPendingSystem).toHaveBeenCalledTimes(1);

handlers.dispose?.();

});