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

devtools/schema_tree_test.go

@@ -0,0 +1,150 @@
+package devtools
+
+import (
+	"fmt"
+	"testing"
+
+	"github.com/egoist/mygo/ui"
+
+	"github.com/ZacharyZhang-NY/MujicaUI/core"
+)
+
+var schemaSample = []SchemaNode{{Name: "court", Kind: SchemaDatabase, Children: []SchemaNode{
+	{Name: "guests", Kind: SchemaTable, Children: []SchemaNode{
+		{Name: "id", Kind: SchemaColumn, Detail: "bigint primary key"},
+		{Name: "name", Kind: SchemaColumn, Detail: "text not null"},
+		{Name: "guests_name_idx", Kind: SchemaIndex, Detail: "(name)"},
+	}},
+	{Name: "seating", Kind: SchemaView},
+}}}
+
+func schemaTester(s *SchemaTreeState, roots []SchemaNode, changes *int) *ui.Tester {
+	return ui.NewTester(func(c *ui.Context) {
+		core.Use(c, core.Settings{})
+		if SchemaTree(c, s, roots, SchemaTreeOptions{}).Changed() {
+			*changes++
+		}
+	}, 520, 420)
+}
+
+func TestSchemaTreeKeyboardOpenAndChoose(t *testing.T) {
+	var s SchemaTreeState
+	n := 0
+	tt := schemaTester(&s, schemaSample, &n)
+	if _, ok := s.Selected(); ok || tt.HasText("guests") {
+		t.Fatalf("start: %q", tt.Texts())
+	}
+	tt.Key(0, ui.KeyTab)
+	tt.Key(0, ui.KeyDown) // court
+	tt.Key(0, ui.KeyRight)
+	tt.Key(0, ui.KeyDown) // guests
+	tt.Key(0, ui.KeyRight)
+	for _, s := range []string{"guests", "seating", "bigint primary key", "(name)"} {
+		if !tt.HasText(s) {
+			t.Fatalf("missing %q in %q", s, tt.Texts())
+		}
+	}
+	tt.Key(0, ui.KeyDown)
+	if p, _ := s.Selected(); p != "court/guests/id" || n == 0 {
+		t.Fatalf("selected %q changes %d", p, n)
+	}
+}
+
+func TestSchemaTreeEmptyRulesMany(t *testing.T) {
+	var s SchemaTreeState
+	n := 0
+	tt := schemaTester(&s, nil, &n)
+	if !tt.HasText("No databases") {
+		t.Fatal("empty text")
+	}
+	cols := make([]SchemaNode, 100000)
+	for i := range cols {
+		cols[i] = SchemaNode{Name: fmt.Sprint("c", i), Kind: SchemaColumn}
+	}
+	big := []SchemaNode{{Name: "wide", Kind: SchemaTable, Children: cols}}
+	var bs SchemaTreeState
+	tt = schemaTester(&bs, big, &n)
+	tt.Key(0, ui.KeyTab)
+	tt.Key(0, ui.KeyDown)
+	tt.Key(0, ui.KeyRight)
+	if got := len(tt.Texts()); got > 400 || !tt.HasText("c0") {
+		t.Fatalf("%d texts for 100000 columns", got)
+	}
+	defer func() {
+		if recover() == nil {
+			t.Fatal("duplicate path did not panic")
+		}
+	}()
+	var ds SchemaTreeState
+	schemaTester(&ds, []SchemaNode{{Name: "a"}, {Name: "a"}}, &n)
+}
+
+func TestSchemaTreeSnapshot(t *testing.T) {
+	var s SchemaTreeState
+	snapshotAfter(t, "schema_tree", 500, 300, func(c *ui.Context) {
+		ui.Column(c).Padding(16).Children(func() { SchemaTree(c, &s, schemaSample, SchemaTreeOptions{Height: 260}) })
+	}, func(tt *ui.Tester) {
+		s = SchemaTreeState{}
+		for _, k := range []ui.Key{ui.KeyTab, ui.KeyDown, ui.KeyRight, ui.KeyDown, ui.KeyRight, ui.KeyDown} {
+			tt.Key(0, k)
+		}
+	})
+}
+
+func TestSchemaTreeAncestorDisabled(t *testing.T) {
+	var s SchemaTreeState
+	disabled := false
+	changes := 0
+	tt := ui.NewTester(func(c *ui.Context) {
+		core.Use(c, core.Settings{})
+		ui.Column(c).Disabled(disabled).Children(func() {
+			if SchemaTree(c, &s, schemaSample, SchemaTreeOptions{}).Changed() {
+				changes++
+			}
+		})
+	}, 520, 420)
+	tt.Key(0, ui.KeyTab)
+	tt.Key(0, ui.KeyDown)
+	before, _ := s.Selected()
+	old := changes
+	disabled = true
+	tt.Key(0, ui.KeyRight)
+	tt.Key(0, ui.KeyDown)
+	if after, _ := s.Selected(); after != before || changes != old || tt.HasText("guests") {
+		t.Fatalf("disabled schema: before=%q after=%q events=%d old=%d texts=%q", before, after, changes, old, tt.Texts())
+	}
+}
+
+func TestSchemaTreeDisclosureHitArea(t *testing.T) {
+	var state SchemaTreeState
+	changes := 0
+	tt := schemaTester(&state, schemaSample, &changes)
+	row, ok := tt.Find("court")
+	if !ok {
+		t.Fatal("database row missing")
+	}
+	tt.ClickAt(row.X+31, row.Y+8)
+	if !tt.HasText("guests") {
+		t.Fatal("disclosure missed 28 DIP hit area")
+	}
+}
+
+func TestSchemaTreeClearsSelectionWhenEmpty(t *testing.T) {
+	var state SchemaTreeState
+	roots := schemaSample
+	tt := ui.NewTester(func(c *ui.Context) {
+		core.Use(c, core.Settings{})
+		SchemaTree(c, &state, roots, SchemaTreeOptions{})
+	}, 520, 420)
+	if err := tt.Click("court"); err != nil {
+		t.Fatal(err)
+	}
+	if _, ok := state.Selected(); !ok {
+		t.Fatal("database selection missing")
+	}
+	roots = nil
+	tt.Frame()
+	if _, ok := state.Selected(); ok || !tt.HasText("No databases") {
+		t.Fatal("empty schema retained selection")
+	}
+}