Agent: components 261-280

tool_approval_dialog_test.go

@@ -0,0 +1,97 @@
+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."})
+	})
+}