Split into category packages; module github.com/ZacharyZhang-NY/MujicaUI

input/tree_select_test.go

@@ -0,0 +1,144 @@
+package input
+
+import (
+	"slices"
+	"testing"
+
+	"github.com/egoist/mygo/ui"
+
+	"github.com/ZacharyZhang-NY/MujicaUI/core"
+)
+
+var treeSelectTestNodes = []TreeSelectNode{
+	{ID: "court", Label: "Court", Children: []TreeSelectNode{
+		{ID: "king", Label: "King"},
+		{ID: "queen", Label: "Queen"},
+		{ID: "jester", Label: "Jester", Disabled: true},
+	}},
+	{ID: "church", Label: "Church", Children: []TreeSelectNode{
+		{ID: "bishop", Label: "Bishop"},
+	}},
+	{ID: "guard", Label: "Guard"},
+}
+
+func treeSelectTester(sel *[]string, opts TreeSelectOptions, changes *int) *ui.Tester {
+	return ui.NewTester(func(c *ui.Context) {
+		core.Use(c, core.Settings{})
+		ui.Column(c).Padding(10).Children(func() {
+			if TreeSelect(c, sel, treeSelectTestNodes, opts).Changed() {
+				*changes++
+			}
+		})
+	}, 500, 500)
+}
+
+func TestTreeSelectSingle(t *testing.T) {
+	var sel []string
+	changes := 0
+	tt := treeSelectTester(&sel, TreeSelectOptions{}, &changes)
+	_ = tt.Click("Select…")
+	if tt.HasText("King") {
+		t.Fatal("collapsed children shown")
+	}
+	if err := tt.Click("Expand Court"); err != nil {
+		t.Fatal(err)
+	}
+	_ = tt.Click("Queen")
+	if !slices.Equal(sel, []string{"queen"}) || changes != 1 || tt.HasText("King") {
+		t.Fatalf("sel %v changes %d open %v", sel, changes, tt.HasText("King"))
+	}
+}
+
+func TestTreeSelectCascade(t *testing.T) {
+	var sel []string
+	changes := 0
+	tt := treeSelectTester(&sel, TreeSelectOptions{Multiple: true}, &changes)
+	_ = tt.Click("Select…")
+	_ = tt.Click("Expand Court")
+	_ = tt.Click("Court")
+	// The disabled Jester stays out.
+	if !slices.Equal(sel, []string{"king", "queen"}) {
+		t.Fatalf("cascade checked %v", sel)
+	}
+	// The chosen labels also show as tags: reach King with the keys.
+	tt.Key(0, ui.KeyRight)
+	tt.Key(0, ui.KeySpace)
+	if !slices.Equal(sel, []string{"queen"}) {
+		t.Fatalf("after unchecking King: %v", sel)
+	}
+	if got := treeSelectCheck(&treeSelectTestNodes[0], sel, TreeSelectOptions{Multiple: true}); got != 2 {
+		t.Fatalf("Court state %v, want mixed", got)
+	}
+	_ = tt.Click("Court")
+	if !slices.Equal(sel, []string{"king", "queen"}) || changes != 3 {
+		t.Fatalf("mixed parent click: %v changes %d", sel, changes)
+	}
+}
+
+func TestTreeSelectIndependent(t *testing.T) {
+	var sel []string
+	changes := 0
+	tt := treeSelectTester(&sel, TreeSelectOptions{Multiple: true, Rule: TreeSelectIndependent}, &changes)
+	_ = tt.Click("Select…")
+	_ = tt.Click("Court")
+	if !slices.Equal(sel, []string{"court"}) {
+		t.Fatalf("independent %v", sel)
+	}
+}
+
+func TestTreeSelectSearch(t *testing.T) {
+	var sel []string
+	changes := 0
+	tt := treeSelectTester(&sel, TreeSelectOptions{Multiple: true}, &changes)
+	_ = tt.Click("Select…")
+	tt.Type("bish")
+	if !tt.HasText("Church") || !tt.HasText("Bishop") || tt.HasText("Court") || tt.HasText("Guard") {
+		t.Fatalf("search shows %v", tt.Texts())
+	}
+	tt.Type("zz")
+	if !tt.HasText("No results") {
+		t.Fatal("no-results missing")
+	}
+}
+
+func TestTreeSelectKeyboard(t *testing.T) {
+	var sel []string
+	changes := 0
+	tt := treeSelectTester(&sel, TreeSelectOptions{Multiple: true, Label: "Roles"}, &changes)
+	tt.Key(0, ui.KeyTab)
+	tt.Key(0, ui.KeyEnter)
+	tt.Key(0, ui.KeyDown)
+	if !tt.Focused("Court") {
+		t.Fatal("Down from search did not reach the tree")
+	}
+	tt.Key(0, ui.KeyRight)
+	if !tt.HasText("King") {
+		t.Fatal("Right did not expand")
+	}
+	tt.Key(0, ui.KeyRight)
+	if !tt.Focused("King") {
+		t.Fatal("second Right did not enter the children")
+	}
+	tt.Key(0, ui.KeySpace)
+	tt.Key(0, ui.KeyLeft)
+	if !tt.Focused("Court") {
+		t.Fatal("Left did not go to the parent")
+	}
+	tt.Key(0, ui.KeyLeft)
+	if tt.HasText("Queen") {
+		t.Fatal("Left did not collapse")
+	}
+	if !slices.Equal(sel, []string{"king"}) {
+		t.Fatalf("sel %v", sel)
+	}
+}
+
+func TestTreeSelectSnapshot(t *testing.T) {
+	snapshot(t, "tree_select", 460, 110, func(c *ui.Context) {
+		single, multi := []string{"bishop"}, []string{"king", "queen", "bishop", "guard"}
+		ui.Row(c).Padding(20).Gap(16).Children(func() {
+			TreeSelect(c, &single, treeSelectTestNodes, TreeSelectOptions{})
+			TreeSelect(c, &multi, treeSelectTestNodes, TreeSelectOptions{Multiple: true, MaxTags: 2})
+		})
+	})
+}