fix: do not swallow context errors

Fixes #1212 Closes #1217
charmbracelet · Feb 4, 2025 · 6e49d3a · 6e49d3a
1 parent 7f68efb
commit 6e49d3a
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/tea.go b/tea.go
@@ -622,7 +622,7 @@ func (p *Program) Run() (Model, error) {
 	model, err := p.eventLoop(model, cmds)
 	killed := p.ctx.Err() != nil || err != nil
 	if killed && err == nil {
-		err = fmt.Errorf("%w: %s", ErrProgramKilled, p.ctx.Err())
+		err = fmt.Errorf("%w: %w", ErrProgramKilled, p.ctx.Err())
 	}
 	if err == nil {
 		// Ensure we rendered the final state of the model.

diff --git a/tea_test.go b/tea_test.go
@@ -141,6 +141,10 @@ func TestTeaKill(t *testing.T) {
 	if _, err := p.Run(); !errors.Is(err, ErrProgramKilled) {
 		t.Fatalf("Expected %v, got %v", ErrProgramKilled, err)
 	}
+
+	if _, err := p.Run(); !errors.Is(err, context.Canceled) {
+		t.Fatalf("Expected %v, got %v", context.Canceled, err)
+	}
 }
 
 func TestTeaContext(t *testing.T) {