feat: print last output when doWaitFor times out (#117)

* feat: print last output when doWaitFor times out * fix test
charmbracelet · Jul 10, 2024 · 1a5969e · 1a5969e
1 parent d0f850a
commit 1a5969e
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/exp/teatest/teatest.go b/exp/teatest/teatest.go
@@ -99,7 +99,7 @@ func doWaitFor(r io.Reader, condition func(bts []byte) bool, options ...WaitForO
 		}
 		time.Sleep(wf.CheckInterval)
 	}
-	return fmt.Errorf("WaitFor: condition not met after %s", wf.Duration)
+	return fmt.Errorf("WaitFor: condition not met after %s. Last output:\n%s", wf.Duration, b.String())
 }
 
 // TestModel is a model that is being tested.

diff --git a/exp/teatest/teatest_test.go b/exp/teatest/teatest_test.go
@@ -29,7 +29,7 @@ func TestWaitForTimeout(t *testing.T) {
 	if err == nil {
 		t.Fatal("expected an error, got nil")
 	}
-	if err.Error() != "WaitFor: condition not met after 1ms" {
+	if err.Error() != "WaitFor: condition not met after 1ms. Last output:\nnope" {
 		t.Fatalf("unexpected error: %s", err.Error())
 	}
 }