Agent: components 261-280

sub_agent_tree_test.go

@@ -0,0 +1,81 @@
+package mujica
+
+import (
+	"testing"
+
+	"github.com/egoist/mygo/ui"
+)
+
+var subAgentSample = []SubAgent{
+	{ID: "lead", Name: "Steward", Task: "Prepare the ball", State: AgentRunning, Children: []SubAgent{
+		{ID: "music", Name: "Musician", Task: "Hire the quartet", State: AgentDone},
+		{ID: "kitchen", Name: "Cook", Task: "Plan the menu", State: AgentWaiting, Children: []SubAgent{
+			{ID: "wine", Name: "Sommelier", Task: "Choose the wine", State: AgentFailed},
+		}},
+	}},
+}
+
+func subAgentTester(opts SubAgentTreeOptions, changes *int) *ui.Tester {
+	opts.Still = true
+	return ui.NewTester(func(c *ui.Context) {
+		Use(c, Settings{})
+		if SubAgentTree(c, subAgentSample, opts).Changed() {
+			*changes++
+		}
+	}, 560, 300)
+}
+
+func TestSubAgentTreeExpandByPointerAndKeys(t *testing.T) {
+	changes := 0
+	tt := subAgentTester(SubAgentTreeOptions{}, &changes)
+	if tt.HasText("Musician") || !tt.HasText("Running") {
+		t.Fatalf("texts %q", tt.Texts())
+	}
+	_ = tt.Click("Steward")
+	if changes != 1 || !tt.HasText("Musician") || !tt.HasText("Waiting") || tt.HasText("Sommelier") {
+		t.Fatalf("click: changes %d texts %q", changes, tt.Texts())
+	}
+	tt.Key(0, ui.KeyDown)
+	tt.Key(0, ui.KeyDown) // Cook
+	tt.Key(0, ui.KeyRight)
+	if !tt.HasText("Sommelier") || !tt.HasText("Failed") {
+		t.Fatal("Right did not expand Cook")
+	}
+	tt.Key(0, ui.KeyRight) // already open: nothing
+	tt.Key(0, ui.KeyLeft)
+	if tt.HasText("Sommelier") || changes != 3 {
+		t.Fatalf("Left: changes %d", changes)
+	}
+}
+
+func TestSubAgentTreeDisabled(t *testing.T) {
+	changes := 0
+	tt := subAgentTester(SubAgentTreeOptions{Disabled: true}, &changes)
+	_ = tt.Click("Steward")
+	tt.Key(0, ui.KeyTab)
+	tt.Key(0, ui.KeyRight)
+	if changes != 0 || tt.HasText("Musician") {
+		t.Fatal("disabled tree expanded")
+	}
+}
+
+func TestSubAgentTreeRejectsDuplicateIDs(t *testing.T) {
+	defer func() {
+		if recover() == nil {
+			t.Fatal("no panic")
+		}
+	}()
+	ui.NewTester(func(c *ui.Context) {
+		Use(c, Settings{})
+		SubAgentTree(c, []SubAgent{{ID: "a", Name: "x", Children: []SubAgent{{ID: "a", Name: "y"}}}}, SubAgentTreeOptions{})
+	}, 200, 200)
+}
+
+func TestSubAgentTreeSnapshot(t *testing.T) {
+	snapshotAfter(t, "sub_agent_tree", 540, 200, func(c *ui.Context) {
+		ui.Column(c).Padding(12).Children(func() { SubAgentTree(c, subAgentSample, SubAgentTreeOptions{Still: true}) })
+	}, func(tt *ui.Tester) {
+		_ = tt.Click("Steward")
+		_ = tt.Click("Cook")
+	})
+}