forked from Mirrors/bubbletea
93 lines
1.9 KiB
Go
93 lines
1.9 KiB
Go
package main
|
|
|
|
import (
|
|
"bytes"
|
|
"io"
|
|
"regexp"
|
|
"testing"
|
|
"time"
|
|
|
|
tea "github.com/charmbracelet/bubbletea"
|
|
"github.com/charmbracelet/lipgloss"
|
|
"github.com/charmbracelet/x/exp/teatest"
|
|
"github.com/muesli/termenv"
|
|
)
|
|
|
|
func init() {
|
|
lipgloss.SetColorProfile(termenv.Ascii)
|
|
}
|
|
|
|
func TestApp(t *testing.T) {
|
|
m := model(10)
|
|
tm := teatest.NewTestModel(
|
|
t, m,
|
|
teatest.WithInitialTermSize(70, 30),
|
|
)
|
|
t.Cleanup(func() {
|
|
if err := tm.Quit(); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
})
|
|
|
|
time.Sleep(time.Second + time.Millisecond*200)
|
|
tm.Type("I'm typing things, but it'll be ignored by my program")
|
|
tm.Send("ignored msg")
|
|
tm.Send(tea.KeyMsg{
|
|
Type: tea.KeyEnter,
|
|
})
|
|
|
|
if err := tm.Quit(); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
out := readBts(t, tm.FinalOutput(t))
|
|
if !regexp.MustCompile(`This program will exit in \d+ seconds`).Match(out) {
|
|
t.Fatalf("output does not match the given regular expression: %s", string(out))
|
|
}
|
|
teatest.RequireEqualOutput(t, out)
|
|
|
|
if tm.FinalModel(t).(model) != 9 {
|
|
t.Errorf("expected model to be 10, was %d", m)
|
|
}
|
|
}
|
|
|
|
func TestAppInteractive(t *testing.T) {
|
|
m := model(10)
|
|
tm := teatest.NewTestModel(
|
|
t, m,
|
|
teatest.WithInitialTermSize(70, 30),
|
|
)
|
|
|
|
time.Sleep(time.Second + time.Millisecond*200)
|
|
tm.Send("ignored msg")
|
|
|
|
if bts := readBts(t, tm.Output()); !bytes.Contains(bts, []byte("This program will exit in 9 seconds")) {
|
|
t.Fatalf("output does not match: expected %q", string(bts))
|
|
}
|
|
|
|
teatest.WaitFor(t, tm.Output(), func(out []byte) bool {
|
|
return bytes.Contains(out, []byte("This program will exit in 7 seconds"))
|
|
}, teatest.WithDuration(5*time.Second))
|
|
|
|
tm.Send(tea.KeyMsg{
|
|
Type: tea.KeyEnter,
|
|
})
|
|
|
|
if err := tm.Quit(); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
if tm.FinalModel(t).(model) != 7 {
|
|
t.Errorf("expected model to be 7, was %d", m)
|
|
}
|
|
}
|
|
|
|
func readBts(tb testing.TB, r io.Reader) []byte {
|
|
tb.Helper()
|
|
bts, err := io.ReadAll(r)
|
|
if err != nil {
|
|
tb.Fatal(err)
|
|
}
|
|
return bts
|
|
}
|