chore: update View method signature in tests

charmbracelet · Jan 30, 2025 · ff6833c · ff6833c
1 parent 67f8ede
commit ff6833c
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 5 deletions.
diff --git a/exec_test.go b/exec_test.go
@@ -2,6 +2,7 @@ package tea
 
 import (
 	"bytes"
+	"fmt"
 	"os/exec"
 	"runtime"
 	"testing"
@@ -34,7 +35,7 @@ func (m *testExecModel) Update(msg Msg) (*testExecModel, Cmd) {
 	return m, nil
 }
 
-func (m *testExecModel) View() Frame {
+func (m *testExecModel) View() fmt.Stringer {
 	return NewFrame("\n")
 }
 

diff --git a/tea_test.go b/tea_test.go
@@ -4,6 +4,7 @@ import (
 	"bytes"
 	"context"
 	"errors"
+	"fmt"
 	"io"
 	"sync/atomic"
 	"testing"
@@ -50,7 +51,7 @@ func (m *testModel) Update(msg Msg) (*testModel, Cmd) {
 	return m, nil
 }
 
-func (m *testModel) View() Frame {
+func (m *testModel) View() fmt.Stringer {
 	m.executed.Store(true)
 	return NewFrame("success\n")
 }
@@ -85,7 +86,7 @@ func TestTeaQuit(t *testing.T) {
 	go func() {
 		for {
 			time.Sleep(time.Millisecond)
-			if p.Model.executed.Load() != nil {
+			if p.Model != nil && p.Model.executed.Load() != nil {
 				p.Quit()
 				return
 			}
@@ -143,7 +144,7 @@ func TestTeaKill(t *testing.T) {
 	go func() {
 		for {
 			time.Sleep(time.Millisecond)
-			if p.Model.executed.Load() != nil {
+			if p.Model != nil && p.Model.executed.Load() != nil {
 				p.Kill()
 				return
 			}
@@ -168,7 +169,7 @@ func TestTeaContext(t *testing.T) {
 	go func() {
 		for {
 			time.Sleep(time.Millisecond)
-			if p.Model.executed.Load() != nil {
+			if p.Model != nil && p.Model.executed.Load() != nil {
 				cancel()
 				return
 			}