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

expect(state.activeChatRunId).toBeNull();

expect(chatLog.addSystem).toHaveBeenCalledWith(

expect.stringContaining("streaming watchdog"),

);

expect(chatLog.addSystem).toHaveBeenCalledWith(expect.stringContaining("streaming watchdog"));

handlers.dispose?.();

});

vi.advanceTimersByTime(3_000);

// 6s total, but the latest delta was only 3s ago, so the watchdog must not

// have fired yet.

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

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

vi.advanceTimersByTime(10_000);

// After a normal final, the watchdog timer must have been cancelled and

// cannot later re-overwrite the status or emit the warning banner.

const statusCalls = setActivityStatus.mock.calls.map((c) => c[0]);

expect(statusCalls.filter((s) => s === "idle").length).toBe(1);

expect(chatLog.addSystem).not.toHaveBeenCalledWith(

handlers.dispose?.();

});

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

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

streamingWatchdogMs: 5_000,

});

handlers.handleChatEvent({

runId: "run-old",

sessionKey: state.currentSessionKey,

state: "delta",

message: { content: "old" },

} satisfies ChatEvent);

vi.advanceTimersByTime(5_001);

expect(state.activeChatRunId).toBeNull();

handlers.handleChatEvent({

runId: "run-new",

sessionKey: state.currentSessionKey,

state: "delta",

message: { content: "new" },

} satisfies ChatEvent);

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

vi.advanceTimersByTime(3_000);

handlers.handleChatEvent({

runId: "run-old",

sessionKey: state.currentSessionKey,

state: "delta",

message: { content: "old again" },

} satisfies ChatEvent);

vi.advanceTimersByTime(2_001);

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

expect(state.activeChatRunId).toBeNull();

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

handlers.dispose?.();

});

it("dispose clears a pending watchdog without firing it", () => {

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

streamingWatchdogMs: 5_000,