Files
MujicaUI/slash_command_menu_test.go
T
2026-10-05 15:02:31 -04:00

123 lines
3.3 KiB
Go

package mujica
import (
"testing"
"github.com/egoist/mygo/ui"
)
var slashCommandsSample = []SlashCommand{
{Name: "summarize", Description: "Summarize the thread"},
{Name: "search", Description: "Search the web"},
{Name: "translate", Description: "Translate to French"},
}
// slashTester is a composer with a slash menu; it records chosen commands and sends.
type slashTester struct {
text string
chosen []string
sends int
off bool
}
func (s *slashTester) view(c *ui.Context) {
Use(c, Settings{})
ui.Column(c).Fill().Justify(ui.End).Padding(16).Children(func() {
p := PromptComposer(c, &s.text, PromptComposerOptions{Label: "Prompt", Disabled: s.off})
if p.Submitted() {
s.sends++
}
if cmd, ok := SlashCommandMenu(c, p, &s.text, slashCommandsSample).Chosen(); ok {
s.chosen = append(s.chosen, cmd.Name)
}
})
}
func TestSlashCommandMenuFilterAndKeys(t *testing.T) {
s := &slashTester{}
tt := ui.NewTester(s.view, 500, 400)
_ = tt.Click("Prompt")
tt.Type("/s")
sr, _ := tt.Find("Search the web")
tr, ok := tt.Find("Translate to French")
if !ok || !tt.HasText("/summarize") || tr.Y <= sr.Y {
t.Fatalf("filter /s does not list prefix matches first: %q", tt.Texts())
}
tt.Type("e")
if tt.HasText("/summarize") || !tt.HasText("/search") {
t.Fatalf("filter /se: %q", tt.Texts())
}
tt.Key(0, ui.KeyBackspace)
tt.Key(0, ui.KeyDown)
tt.Key(0, ui.KeyEnter)
if s.text != "/search " || len(s.chosen) != 1 || s.chosen[0] != "search" || s.sends != 0 {
t.Fatalf("Enter: text %q chosen %v sends %d", s.text, s.chosen, s.sends)
}
if tt.HasText("/summarize") {
t.Fatal("menu stays open after choosing")
}
}
func TestSlashCommandMenuTabCompletesAndEscapeCloses(t *testing.T) {
s := &slashTester{}
tt := ui.NewTester(s.view, 500, 400)
_ = tt.Click("Prompt")
tt.Type("/tr")
tt.Key(0, ui.KeyTab)
if s.text != "/translate " || len(s.chosen) != 0 {
t.Fatalf("Tab: text %q chosen %v", s.text, s.chosen)
}
s.text = ""
tt.Frame()
tt.Type("/")
if !tt.HasText("/summarize") {
t.Fatal("menu did not open on /")
}
tt.Key(0, ui.KeyEscape)
if tt.HasText("/summarize") {
t.Fatal("Escape did not close the menu")
}
tt.Type("t")
if !tt.HasText("/translate") {
t.Fatal("menu did not reopen as the filter changed")
}
}
func TestSlashCommandMenuClickAndEmpty(t *testing.T) {
s := &slashTester{}
tt := ui.NewTester(s.view, 500, 400)
_ = tt.Click("Prompt")
tt.Type("/zz")
if !tt.HasText("No matching commands") {
t.Fatalf("empty state: %q", tt.Texts())
}
tt.Key(0, ui.KeyBackspace)
tt.Key(0, ui.KeyBackspace)
if err := tt.Click("/translate"); err != nil {
t.Fatal(err)
}
if len(s.chosen) != 1 || s.chosen[0] != "translate" || !tt.Focused("Prompt") {
t.Fatalf("click chose %v, composer focused %v", s.chosen, tt.Focused("Prompt"))
}
}
func TestSlashCommandMenuDisabledComposer(t *testing.T) {
s := &slashTester{text: "/s", off: true}
tt := ui.NewTester(s.view, 500, 400)
_ = tt.Click("Prompt")
tt.Key(0, ui.KeyEnter)
if tt.HasText("/search") || len(s.chosen) != 0 {
t.Fatal("disabled composer shows or uses the menu")
}
}
func TestSlashCommandMenuSnapshot(t *testing.T) {
snapshotAfter(t, "slash_command_menu", 520, 360, func(c *ui.Context) {
(*ui.Local(c.Root(), "s", func() *slashTester { return &slashTester{} })).view(c)
}, func(tt *ui.Tester) {
_ = tt.Click("Prompt")
tt.Type("/")
tt.Key(0, ui.KeyDown)
})
}