wip-devtools

schema_tree_test.go

@@ -0,0 +1,90 @@
+package mujica
+
+import (
+	"fmt"
+	"testing"
+
+	"github.com/egoist/mygo/ui"
+)
+
+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) {
+		Use(c, 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)
+		}
+	})
+}