mirror of
https://github.com/antonmedv/fx
synced 2024-11-11 07:10:28 +00:00
104 lines
2.3 KiB
Go
104 lines
2.3 KiB
Go
package main
|
|
|
|
import (
|
|
"bytes"
|
|
"io"
|
|
"os"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/charmbracelet/bubbles/textinput"
|
|
tea "github.com/charmbracelet/bubbletea"
|
|
"github.com/charmbracelet/lipgloss"
|
|
"github.com/charmbracelet/x/exp/teatest"
|
|
"github.com/muesli/termenv"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func init() {
|
|
lipgloss.SetColorProfile(termenv.ANSI)
|
|
}
|
|
|
|
func prepare(t *testing.T) *teatest.TestModel {
|
|
file, err := os.Open("testdata/example.json")
|
|
require.NoError(t, err)
|
|
|
|
json, err := io.ReadAll(file)
|
|
require.NoError(t, err)
|
|
|
|
head, err := parse(json)
|
|
require.NoError(t, err)
|
|
|
|
m := &model{
|
|
top: head,
|
|
head: head,
|
|
wrap: true,
|
|
showCursor: true,
|
|
digInput: textinput.New(),
|
|
searchInput: textinput.New(),
|
|
search: newSearch(),
|
|
}
|
|
tm := teatest.NewTestModel(
|
|
t, m,
|
|
teatest.WithInitialTermSize(80, 40),
|
|
)
|
|
return tm
|
|
}
|
|
|
|
func read(t *testing.T, tm *teatest.TestModel) []byte {
|
|
var out []byte
|
|
teatest.WaitFor(t,
|
|
tm.Output(),
|
|
func(b []byte) bool {
|
|
out = b
|
|
return bytes.Contains(b, []byte("{"))
|
|
},
|
|
teatest.WithCheckInterval(time.Millisecond*100),
|
|
teatest.WithDuration(time.Second),
|
|
)
|
|
return out
|
|
}
|
|
|
|
func TestOutput(t *testing.T) {
|
|
tm := prepare(t)
|
|
|
|
teatest.RequireEqualOutput(t, read(t, tm))
|
|
|
|
tm.Send(tea.KeyMsg{Type: tea.KeyRunes, Runes: []rune("q")})
|
|
tm.WaitFinished(t, teatest.WithFinalTimeout(time.Second))
|
|
}
|
|
|
|
func TestNavigation(t *testing.T) {
|
|
tm := prepare(t)
|
|
|
|
tm.Send(tea.KeyMsg{Type: tea.KeyDown})
|
|
tm.Send(tea.KeyMsg{Type: tea.KeyDown})
|
|
tm.Send(tea.KeyMsg{Type: tea.KeyDown})
|
|
teatest.RequireEqualOutput(t, read(t, tm))
|
|
|
|
tm.Send(tea.KeyMsg{Type: tea.KeyRunes, Runes: []rune("q")})
|
|
tm.WaitFinished(t, teatest.WithFinalTimeout(time.Second))
|
|
}
|
|
|
|
func TestDig(t *testing.T) {
|
|
tm := prepare(t)
|
|
|
|
tm.Send(tea.KeyMsg{Type: tea.KeyRunes, Runes: []rune(".")})
|
|
tm.Send(tea.KeyMsg{Type: tea.KeyRunes, Runes: []rune(".year")})
|
|
tm.Send(tea.KeyMsg{Type: tea.KeyEnter})
|
|
teatest.RequireEqualOutput(t, read(t, tm))
|
|
|
|
tm.Send(tea.KeyMsg{Type: tea.KeyRunes, Runes: []rune("q")})
|
|
tm.WaitFinished(t, teatest.WithFinalTimeout(time.Second))
|
|
}
|
|
|
|
func TestCollapseRecursive(t *testing.T) {
|
|
tm := prepare(t)
|
|
|
|
tm.Send(tea.KeyMsg{Type: tea.KeyShiftLeft})
|
|
teatest.RequireEqualOutput(t, read(t, tm))
|
|
|
|
tm.Send(tea.KeyMsg{Type: tea.KeyRunes, Runes: []rune("q")})
|
|
tm.WaitFinished(t, teatest.WithFinalTimeout(time.Second))
|
|
}
|