feat(typing): add TTL safety-net for stuck indicators (land #27428, thanks @Crpdim)

Co-authored-by: Crpdim <crpdim@users.noreply.github.com>
This commit is contained in:
Peter Steinberger
2026-02-26 11:48:35 +00:00
parent 3d30ba18a2
commit 0231cac957
3 changed files with 186 additions and 3 deletions

View File

@@ -109,4 +109,156 @@ describe("createTypingCallbacks", () => {
vi.useRealTimers();
}
});
// ========== TTL Safety Tests ==========
describe("TTL safety", () => {
it("auto-stops typing after maxDurationMs", async () => {
vi.useFakeTimers();
try {
const consoleWarn = vi.spyOn(console, "warn").mockImplementation(() => {});
const start = vi.fn().mockResolvedValue(undefined);
const stop = vi.fn().mockResolvedValue(undefined);
const onStartError = vi.fn();
const callbacks = createTypingCallbacks({
start,
stop,
onStartError,
maxDurationMs: 10_000,
});
await callbacks.onReplyStart();
expect(start).toHaveBeenCalledTimes(1);
expect(stop).not.toHaveBeenCalled();
// Advance past TTL
await vi.advanceTimersByTimeAsync(10_000);
// Should auto-stop
expect(stop).toHaveBeenCalledTimes(1);
expect(consoleWarn).toHaveBeenCalledWith(expect.stringContaining("TTL exceeded"));
consoleWarn.mockRestore();
} finally {
vi.useRealTimers();
}
});
it("does not auto-stop if idle is called before TTL", async () => {
vi.useFakeTimers();
try {
const consoleWarn = vi.spyOn(console, "warn").mockImplementation(() => {});
const start = vi.fn().mockResolvedValue(undefined);
const stop = vi.fn().mockResolvedValue(undefined);
const onStartError = vi.fn();
const callbacks = createTypingCallbacks({
start,
stop,
onStartError,
maxDurationMs: 10_000,
});
await callbacks.onReplyStart();
// Stop before TTL
await vi.advanceTimersByTimeAsync(5_000);
callbacks.onIdle?.();
await flushMicrotasks();
expect(stop).toHaveBeenCalledTimes(1);
// Advance past original TTL
await vi.advanceTimersByTimeAsync(10_000);
// Should not have triggered TTL warning
expect(consoleWarn).not.toHaveBeenCalled();
// Stop should still be called only once
expect(stop).toHaveBeenCalledTimes(1);
consoleWarn.mockRestore();
} finally {
vi.useRealTimers();
}
});
it("uses default 60s TTL when not specified", async () => {
vi.useFakeTimers();
try {
const start = vi.fn().mockResolvedValue(undefined);
const stop = vi.fn().mockResolvedValue(undefined);
const onStartError = vi.fn();
const callbacks = createTypingCallbacks({ start, stop, onStartError });
await callbacks.onReplyStart();
// Should not stop at 59s
await vi.advanceTimersByTimeAsync(59_000);
expect(stop).not.toHaveBeenCalled();
// Should stop at 60s
await vi.advanceTimersByTimeAsync(1_000);
expect(stop).toHaveBeenCalledTimes(1);
} finally {
vi.useRealTimers();
}
});
it("disables TTL when maxDurationMs is 0", async () => {
vi.useFakeTimers();
try {
const start = vi.fn().mockResolvedValue(undefined);
const stop = vi.fn().mockResolvedValue(undefined);
const onStartError = vi.fn();
const callbacks = createTypingCallbacks({
start,
stop,
onStartError,
maxDurationMs: 0,
});
await callbacks.onReplyStart();
// Should not auto-stop even after long time
await vi.advanceTimersByTimeAsync(300_000);
expect(stop).not.toHaveBeenCalled();
} finally {
vi.useRealTimers();
}
});
it("resets TTL timer on restart after idle", async () => {
vi.useFakeTimers();
try {
const start = vi.fn().mockResolvedValue(undefined);
const stop = vi.fn().mockResolvedValue(undefined);
const onStartError = vi.fn();
const callbacks = createTypingCallbacks({
start,
stop,
onStartError,
maxDurationMs: 10_000,
});
// First start
await callbacks.onReplyStart();
await vi.advanceTimersByTimeAsync(5_000);
// Idle and restart
callbacks.onIdle?.();
await flushMicrotasks();
expect(stop).toHaveBeenCalledTimes(1);
// Reset mock to track second start
stop.mockClear();
// After stop, callbacks are closed, so new onReplyStart should be no-op
await callbacks.onReplyStart();
await vi.advanceTimersByTimeAsync(15_000);
// Should not trigger stop again since it's closed
expect(stop).not.toHaveBeenCalled();
} finally {
vi.useRealTimers();
}
});
});
});