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

2026-10-05 16:18:48 -04:00
1 parent 5a7e921c7e
commit 22b53e9e9e
@@ -0,0 +1,163 @@
+package feedback
+
+import (
+	"strconv"
+	"testing"
+	"time"
+
+	"github.com/egoist/mygo/ui"
+
+	"github.com/ZacharyZhang-NY/MujicaUI/core"
+)
+
+func traceSample() []TraceSpan {
+	ms := time.Millisecond
+	return []TraceSpan{
+		{ID: "root", Name: "agent.run", Start: 0, End: 900 * ms, Attrs: []TraceAttr{{"model", "opus"}}},
+		{ID: "plan", Parent: "root", Name: "llm.plan", Start: 10 * ms, End: 210 * ms, Attrs: []TraceAttr{{"tokens", "1200"}}},
+		{ID: "search", Parent: "root", Name: "tool.search", Start: 220 * ms, End: 520 * ms, Failed: true},
+		{ID: "fetch", Parent: "search", Name: "http.get", Start: 230 * ms, End: 500 * ms},
+		{ID: "write", Parent: "root", Name: "llm.write", Start: 530 * ms, End: 880 * ms},
+	}
+}
+
+func TestTraceBuild(t *testing.T) {
+	ix := traceBuild(traceSample())
+	if len(ix.roots) != 1 || ix.roots[0] != "root" || len(ix.kids["root"]) != 3 || ix.end != 900*time.Millisecond {
+		t.Fatalf("index %+v", ix)
+	}
+	for name, spans := range map[string][]TraceSpan{
+		"dup":   {{ID: "a"}, {ID: "a"}},
+		"time":  {{ID: "a", Start: 2, End: 1}},
+		"cycle": {{ID: "a", Parent: "b"}, {ID: "b", Parent: "a"}},
+		"self":  {{ID: "a", Parent: "a"}},
+	} {
+		func() {
+			defer func() {
+				if recover() == nil {
+					t.Errorf("%s did not panic", name)
+				}
+			}()
+			traceBuild(spans)
+		}()
+	}
+}
+
+func TestTraceViewerSelectShowsAttributes(t *testing.T) {
+	var s TraceViewerState
+	spans := traceSample()
+	tt := ui.NewTester(func(c *ui.Context) {
+		core.Use(c, core.Settings{})
+		ui.Column(c).Height(600).Children(func() {
+			TraceViewer(c, &s, spans, TraceViewerOptions{}).Element.Grow(1)
+		})
+	}, 900, 620)
+	if !tt.HasText("llm.plan") || !tt.HasText("200.0 ms") {
+		t.Fatalf("tree: %q", tt.Texts())
+	}
+	if _, ok := tt.Find("tool.search, failed"); !ok {
+		t.Fatal("failed span not named")
+	}
+	if tt.HasText("http.get") {
+		t.Fatal("nested span shown before opening its parent")
+	}
+	if err := tt.Click("llm.plan"); err != nil {
+		t.Fatal(err)
+	}
+	if id, _ := s.Tree.Selected(); id != "plan" || !tt.HasText("tokens") || !tt.HasText("1200") {
+		t.Fatalf("attributes: %q", tt.Texts())
+	}
+	tt.Key(0, ui.KeyDown)
+	tt.Key(0, ui.KeyRight)
+	if id, _ := s.Tree.Selected(); id != "search" || !tt.HasText("http.get") {
+		t.Fatalf("keyboard open: %q", tt.Texts())
+	}
+}
+
+func TestTraceViewerSnapshot(t *testing.T) {
+	spans := traceSample()
+	snapshot(t, "trace_viewer", 820, 420, func(c *ui.Context) {
+		s := ui.Local(c.Root(), "s", func() *TraceViewerState {
+			st := &TraceViewerState{}
+			st.Tree.SetOpen("search", true)
+			return st
+		})
+		ui.Column(c).Padding(16).Height(420).Children(func() {
+			TraceViewer(c, *s, spans, TraceViewerOptions{}).Element.Grow(1)
+		})
+	})
+}
+
+func TestTraceViewerRefreshKeepsCollapsedRoot(t *testing.T) {
+	var s TraceViewerState
+	spans := traceSample()
+	tt, _ := motionFrames(func(c *ui.Context) {
+		TraceViewer(c, &s, spans, TraceViewerOptions{}).Element.Height(300)
+	}, 700, 400)
+	s.Tree.SetOpen("root", false)
+	tt.Frame()
+	spans = append([]TraceSpan(nil), spans...)
+	tt.Frame()
+	if s.Tree.IsOpen("root") || tt.HasText("llm.plan") {
+		t.Fatal("data refresh reopened a collapsed root")
+	}
+}
+
+func TestTraceViewerLargeDataAndVersion(t *testing.T) {
+	spans := make([]TraceSpan, 100000)
+	for i := range spans {
+		spans[i] = TraceSpan{ID: strconv.Itoa(i), Name: "Span " + strconv.Itoa(i), End: time.Millisecond}
+	}
+	var s TraceViewerState
+	version := 0
+	tt := ui.NewTester(func(c *ui.Context) {
+		core.Use(c, core.Settings{})
+		TraceViewer(c, &s, spans, TraceViewerOptions{Version: version}).Element.Height(300)
+	}, 700, 400)
+	if len(s.index.byID) != 100000 || len(tt.Texts()) > 150 {
+		t.Fatalf("indexed %d texts %d", len(s.index.byID), len(tt.Texts()))
+	}
+	spans[0].End = 2 * time.Millisecond
+	version++
+	tt.Frame()
+	if !tt.HasText("2.0 ms") {
+		t.Fatal("stale duration")
+	}
+}
+
+func TestTraceViewerLargeAttributePreview(t *testing.T) {
+	spans := []TraceSpan{{ID: "root", Name: "Root", Attrs: make([]TraceAttr, 100000)}}
+	for i := range spans[0].Attrs {
+		spans[0].Attrs[i] = TraceAttr{Key: "Key " + strconv.Itoa(i), Value: "Value"}
+	}
+	var s TraceViewerState
+	tt := ui.NewTester(func(c *ui.Context) {
+		core.Use(c, core.Settings{})
+		TraceViewer(c, &s, spans, TraceViewerOptions{}).Element.Height(500)
+	}, 700, 600)
+	if err := tt.Click("Root"); err != nil {
+		t.Fatal(err)
+	}
+	if !tt.HasText("Key 0") || len(tt.Texts()) > 150 {
+		t.Fatalf("attribute preview built %d texts", len(tt.Texts()))
+	}
+}
+
+func TestTraceViewerAttributesKeepTreeVisible(t *testing.T) {
+	spans := []TraceSpan{{ID: "a", Name: "First span", Attrs: make([]TraceAttr, 20)}, {ID: "b", Name: "Second span"}}
+	for i := range spans[0].Attrs {
+		spans[0].Attrs[i] = TraceAttr{Key: strconv.Itoa(i), Value: "Value"}
+	}
+	var s TraceViewerState
+	tt := ui.NewTester(func(c *ui.Context) {
+		core.Use(c, core.Settings{})
+		TraceViewer(c, &s, spans, TraceViewerOptions{}).Element.Height(260)
+	}, 700, 600)
+	if err := tt.Click("First span"); err != nil {
+		t.Fatal(err)
+	}
+	bounds, ok := tt.Find("Second span")
+	if !ok || bounds.H <= 0 || bounds.W <= 0 {
+		t.Fatalf("attributes hid tree row: %+v", bounds)
+	}
+}