Function waitForCapability

go/session_test.go:916–930

waitForCapability polls Session.Capabilities() until predicate matches or timeout. Returns the last observed capabilities. Avoids time.Sleep in tests.

(t *testing.T, session *Session, predicate func(SessionCapabilities) bool, timeout time.Duration)

// waitForCapability polls Session.Capabilities() until predicate matches or timeout.
// Returns the last observed capabilities. Avoids time.Sleep in tests.
func waitForCapability(t *testing.T, session *Session, predicate func(SessionCapabilities) bool, timeout time.Duration) SessionCapabilities {
	t.Helper()
	deadline := time.Now().Add(timeout)
	var last SessionCapabilities
	for {
		last = session.Capabilities()
		if predicate(last) {
			return last
		}
		if time.Now().After(deadline) {
			return last
		}
		time.Sleep(5 * time.Millisecond)
	}
}

func TestSession_ElicitationCapabilityGating(t *testing.T) {
	t.Run("elicitation errors when capability is missing", func(t *testing.T) {