Function TestRunLoopExitsPromptlyOnCancel_DuringPoll

agent/agentgit/agentgit_test.go:1484–1519

TestRunLoopExitsPromptlyOnCancel_DuringPoll pins that RunLoop returns quickly when its context is cancelled while it is blocked on the fallback poll ticker. Regression guard for the fallback interval: if a future change introduces a non-cancellable wait here, this test will hang and fail.

(t *testing.T)

// interval: if a future change introduces a non-cancellable wait
// here, this test will hang and fail.
func TestRunLoopExitsPromptlyOnCancel_DuringPoll(t *testing.T) {
	t.Parallel()

	logger := slogtest.Make(t, nil)
	mClock := quartz.NewMock(t)
	h := agentgit.NewHandler(logger, agentgit.WithClock(mClock))

	ctx, cancel := context.WithCancel(context.Background())
	t.Cleanup(cancel)

	// Trap NewTicker so the test can synchronize on RunLoop's
	// ticker creation rather than racing against it with a
	// best-effort Advance.
	tickerTrap := mClock.Trap().NewTicker()
	defer tickerTrap.Close()

	done := make(chan struct{})
	go func() {
		defer close(done)
		h.RunLoop(ctx, func() {})
	}()

	// Wait until RunLoop has actually called clock.NewTicker, then
	// release the trap so the ticker is installed. At this point
	// RunLoop is deterministically inside its select, blocked on
	// <-ticker.C / <-scanTrigger / <-ctx.Done().
	tickerTrap.MustWait(ctx).MustRelease(ctx)

	cancel()

	select {
	case <-done:
	case <-time.After(testutil.WaitShort):
		t.Fatal("RunLoop did not return within WaitShort after ctx cancel")
	}
}

// TestRunLoopExitsPromptlyOnCancel_DuringCooldown pins that RunLoop
// returns quickly when its context is cancelled while a