diff --git a/packages/cli/src/acp-integration/session/Session.test.ts b/packages/cli/src/acp-integration/session/Session.test.ts index f720b4cbba..354c01699b 100644 --- a/packages/cli/src/acp-integration/session/Session.test.ts +++ b/packages/cli/src/acp-integration/session/Session.test.ts @@ -3259,6 +3259,7 @@ describe('Session', () => { callback({ prompt: 'scheduled prompt' }); }), stop: vi.fn(), + disable: vi.fn(), getExitSummary: vi.fn().mockReturnValue(undefined), }; mockConfig.isCronEnabled = vi.fn().mockReturnValue(true); @@ -3308,7 +3309,9 @@ describe('Session', () => { }, }, }); - expect(scheduler.stop).toHaveBeenCalledTimes(1); + // Token limit disables the scheduler (permanent for the session, so + // a later LoopWakeup is rejected), not just stops it. + expect(scheduler.disable).toHaveBeenCalledTimes(1); await vi.waitFor(() => { expect(mockClient.sessionUpdate).toHaveBeenCalledWith({ sessionId: 'test-session-id',