test condition exits

Author: Uwe Jugel

assert/assertions_test.go

@@ -3450,6 +3450,24 @@ func TestEventuallyTrue(t *testing.T) {
 	True(t, Eventually(t, condition, 100*time.Millisecond, 20*time.Millisecond))
 }
 
+func TestEventuallyFailNow(t *testing.T) {
+	t.Parallel()
+
+	mockT := new(CollectT)
+	state := 0
+	condition := func() bool {
+		state += 1
+		mockT.FailNow()
+		panic("unreachable")
+	}
+
+	False(t, Eventually(mockT, condition, 100*time.Millisecond, 20*time.Millisecond))
+	True(t, mockT.failed())
+	Len(t, mockT.errors, 1)
+	ErrorContains(t, mockT.errors[0], "Condition exited unexpectedly")
+	Equal(t, 1, state, "Expected condition to be called exactly once")
+}
+
 // errorsCapturingT is a mock implementation of TestingT that captures errors reported with Errorf.
 type errorsCapturingT struct {
 	errors []error
@@ -3543,6 +3561,24 @@ func TestEventuallyWithTFailNow(t *testing.T) {
 	Len(t, mockT.errors, 1)
 }
 
+func TestEventuallyWithTOuterFailNow(t *testing.T) {
+	t.Parallel()
+
+	mockT := new(CollectT)
+	state := 0
+	condition := func(collect *CollectT) {
+		state += 1
+		mockT.FailNow()
+		panic("unreachable")
+	}
+
+	False(t, EventuallyWithT(mockT, condition, 100*time.Millisecond, 20*time.Millisecond))
+	True(t, mockT.failed())
+	Len(t, mockT.errors, 1)
+	ErrorContains(t, mockT.errors[0], "Condition exited unexpectedly")
+	Equal(t, 1, state, "Expected condition to be called exactly once")
+}
+
 // Check that a long running condition doesn't block Eventually.
 // See issue 805 (and its long tail of following issues)
 func TestEventuallyTimeout(t *testing.T) {
@@ -3634,6 +3670,24 @@ func TestNeverFailQuickly(t *testing.T) {
 	False(t, Never(mockT, condition, 100*time.Millisecond, time.Second))
 }
 
+func TestNeverFailNow(t *testing.T) {
+	t.Parallel()
+
+	mockT := new(CollectT)
+	counter := 0
+	condition := func() bool {
+		counter += 1
+		mockT.FailNow()
+		panic("unreachable")
+	}
+
+	False(t, Never(mockT, condition, 100*time.Millisecond, 20*time.Millisecond))
+	True(t, mockT.failed(), "Expected Never to mark TestingT as failed")
+	Len(t, mockT.errors, 1)
+	ErrorContains(t, mockT.errors[0], "Condition exited unexpectedly")
+	Equal(t, 1, counter, "Expected condition to be called exactly once")
+}
+
 func Test_validateEqualArgs(t *testing.T) {
 	t.Parallel()