98 lines
2.4 KiB
Go
98 lines
2.4 KiB
Go
package mujica
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/egoist/mygo/ui"
|
|
)
|
|
|
|
type approvalApp struct {
|
|
open bool
|
|
allowed, denied int
|
|
}
|
|
|
|
func (a *approvalApp) view(c *ui.Context) {
|
|
Use(c, Settings{})
|
|
if Button(c, "Run", ButtonOptions{}).Clicked() {
|
|
a.open = true
|
|
}
|
|
r := ToolApprovalDialog(c, &a.open, ToolApprovalOptions{Tool: "delete_files", Args: `{"glob":"*.tmp"}`, Reason: "Clean the build folder."})
|
|
if r.Allowed {
|
|
a.allowed++
|
|
}
|
|
if r.Denied {
|
|
a.denied++
|
|
}
|
|
}
|
|
|
|
func newApprovalApp() (*approvalApp, *ui.Tester) {
|
|
a := &approvalApp{}
|
|
tt := ui.NewTester(a.view, 600, 500)
|
|
_ = tt.Click("Run")
|
|
return a, tt
|
|
}
|
|
|
|
func TestToolApprovalDialogDenyFirst(t *testing.T) {
|
|
a, tt := newApprovalApp()
|
|
for _, s := range []string{"delete_files", "Clean the build folder.", ` "glob": "*.tmp"`} {
|
|
if !tt.HasText(s) {
|
|
t.Fatalf("missing %q in %q", s, tt.Texts())
|
|
}
|
|
}
|
|
if !tt.Focused("Deny") {
|
|
t.Fatal("Deny does not have the focus")
|
|
}
|
|
tt.Key(0, ui.KeyEnter)
|
|
if a.denied != 1 || a.allowed != 0 || a.open {
|
|
t.Fatalf("Enter: denied %d allowed %d open %v", a.denied, a.allowed, a.open)
|
|
}
|
|
if !tt.Focused("Run") {
|
|
t.Fatal("focus not restored")
|
|
}
|
|
}
|
|
|
|
func TestToolApprovalDialogAllowByTabAndBackdrop(t *testing.T) {
|
|
a, tt := newApprovalApp()
|
|
tt.ClickAt(5, 495)
|
|
if !a.open || a.denied != 0 {
|
|
t.Fatal("backdrop click acted")
|
|
}
|
|
tt.Key(0, ui.KeyTab)
|
|
if !tt.Focused("Allow") {
|
|
t.Fatal("Tab did not reach Allow")
|
|
}
|
|
tt.Key(0, ui.KeyEnter)
|
|
if a.allowed != 1 || a.denied != 0 || a.open {
|
|
t.Fatalf("allowed %d denied %d", a.allowed, a.denied)
|
|
}
|
|
}
|
|
|
|
func TestToolApprovalDialogEscapeDenies(t *testing.T) {
|
|
a, tt := newApprovalApp()
|
|
tt.Key(0, ui.KeyEscape)
|
|
if a.denied != 1 || a.allowed != 0 || a.open {
|
|
t.Fatalf("Escape: denied %d allowed %d open %v", a.denied, a.allowed, a.open)
|
|
}
|
|
}
|
|
|
|
func TestToolApprovalDialogNeedsTool(t *testing.T) {
|
|
defer func() {
|
|
if recover() == nil {
|
|
t.Fatal("no panic")
|
|
}
|
|
}()
|
|
open := true
|
|
ui.NewTester(func(c *ui.Context) {
|
|
Use(c, Settings{})
|
|
ToolApprovalDialog(c, &open, ToolApprovalOptions{})
|
|
}, 200, 200)
|
|
}
|
|
|
|
func TestToolApprovalDialogSnapshot(t *testing.T) {
|
|
snapshot(t, "tool_approval_dialog", 560, 380, func(c *ui.Context) {
|
|
open := true
|
|
ToolApprovalDialog(c, &open, ToolApprovalOptions{Tool: "delete_files", Args: `{"glob":"build/*.tmp","dry_run":false}`,
|
|
Reason: "The agent wants to clean temporary files from the build folder."})
|
|
})
|
|
}
|