112 lines
3.1 KiB
Go
112 lines
3.1 KiB
Go
package mujica
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/egoist/mygo/ui"
|
|
|
|
"mujicaui/icons"
|
|
)
|
|
|
|
func promptTester(text *string, sends *int, opts PromptComposerOptions) *ui.Tester {
|
|
return ui.NewTester(func(c *ui.Context) {
|
|
Use(c, Settings{})
|
|
if PromptComposer(c, text, opts).Submitted() {
|
|
*sends++
|
|
}
|
|
}, 500, 400)
|
|
}
|
|
|
|
func TestPromptComposerEnterSendsShiftEnterBreaks(t *testing.T) {
|
|
text, sends := "", 0
|
|
tt := promptTester(&text, &sends, PromptComposerOptions{Label: "Prompt"})
|
|
_ = tt.Click("Prompt")
|
|
tt.Key(0, ui.KeyEnter)
|
|
if sends != 0 {
|
|
t.Fatal("blank prompt sent")
|
|
}
|
|
tt.Type("Bonjour")
|
|
tt.Key(ui.Shift, ui.KeyEnter)
|
|
tt.Type("Madame")
|
|
tt.Key(0, ui.KeyEnter)
|
|
if sends != 1 || text != "Bonjour\nMadame" {
|
|
t.Fatalf("sends %d text %q", sends, text)
|
|
}
|
|
}
|
|
|
|
func TestPromptComposerCmdEnterMode(t *testing.T) {
|
|
text, sends := "", 0
|
|
tt := promptTester(&text, &sends, PromptComposerOptions{Label: "Prompt", CmdEnterSends: true})
|
|
_ = tt.Click("Prompt")
|
|
tt.Type("a")
|
|
tt.Key(0, ui.KeyEnter)
|
|
tt.Type("b")
|
|
tt.Key(ui.Cmd, ui.KeyEnter)
|
|
if sends != 1 || text != "a\nb" {
|
|
t.Fatalf("sends %d text %q", sends, text)
|
|
}
|
|
}
|
|
|
|
func TestPromptComposerSlotsAndDisabled(t *testing.T) {
|
|
text, sends := "hello", 0
|
|
tt := promptTester(&text, &sends, PromptComposerOptions{Label: "Prompt", Disabled: true})
|
|
_ = tt.Click("Prompt")
|
|
tt.Type("x")
|
|
tt.Key(0, ui.KeyEnter)
|
|
if sends != 0 || text != "hello" {
|
|
t.Fatalf("disabled composer: sends %d text %q", sends, text)
|
|
}
|
|
tt = ui.NewTester(func(c *ui.Context) {
|
|
Use(c, Settings{})
|
|
PromptComposer(c, &text, PromptComposerOptions{
|
|
Attachments: func() { ui.Text(c, "report.pdf") },
|
|
Tools: func() { ui.Text(c, "Web search") },
|
|
Actions: func() { ui.Text(c, "Send it") },
|
|
})
|
|
}, 500, 400)
|
|
for _, s := range []string{"report.pdf", "Web search", "Send it", "Message"} {
|
|
if _, ok := tt.Find(s); !ok {
|
|
t.Errorf("missing %q", s)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestPromptComposerSnapshot(t *testing.T) {
|
|
text := "Draft the invitation for the spring ball,\nin the style of the court."
|
|
snapshot(t, "prompt_composer", 520, 220, func(c *ui.Context) {
|
|
ui.Column(c).Padding(16).Children(func() {
|
|
PromptComposer(c, &text, PromptComposerOptions{
|
|
Attachments: func() { AttachmentChip(c, Attachment{Name: "guests.csv", Size: 4096}) },
|
|
Tools: func() { iconAction(c, icons.Must("paperclip"), "Attach") },
|
|
Actions: func() { SendButton(c, SendButtonOptions{Shortcut: "Enter"}) },
|
|
})
|
|
})
|
|
})
|
|
}
|
|
|
|
func TestPromptComposerCompositionDoesNotSend(t *testing.T) {
|
|
text, sends := "draft", 0
|
|
composing := false
|
|
tt := ui.NewTester(func(c *ui.Context) {
|
|
Use(c, Settings{})
|
|
if PromptComposer(c, &text, PromptComposerOptions{Label: "Prompt", Composing: composing}).Submitted() {
|
|
sends++
|
|
}
|
|
}, 500, 400)
|
|
_ = tt.Click("Prompt")
|
|
composing = true
|
|
tt.Frame()
|
|
tt.Compose("邀请", 2)
|
|
tt.Key(0, ui.KeyEnter)
|
|
if sends != 0 || text != "draft" {
|
|
t.Fatalf("composition submitted: sends %d text %q", sends, text)
|
|
}
|
|
tt.Type("邀请")
|
|
composing = false
|
|
tt.Frame()
|
|
tt.Key(0, ui.KeyEnter)
|
|
if sends != 1 || text != "draft邀请" {
|
|
t.Fatalf("committed input: sends %d text %q", sends, text)
|
|
}
|
|
}
|