WIP motion 2

token_usage_chart_test.go

@@ -0,0 +1,64 @@
+package mujica
+
+import (
+	"testing"
+
+	"github.com/egoist/mygo/ui"
+)
+
+func tokenUsageSample() []TokenUsage {
+	return []TokenUsage{{Label: "Plan", Input: 1200, Output: 300}, {Label: "Search", Input: 4800, Output: 650}, {Label: "Write", Input: 2100, Output: 2900}}
+}
+
+func TestTokenUsageChartBarsKeyboard(t *testing.T) {
+	usage := tokenUsageSample()
+	h := &chartHarness{build: func(c *ui.Context) ChartResult { return TokenUsageChart(c, usage, TokenUsageChartOptions{}) }}
+	tt := ui.NewTester(h.view, 520, 420)
+	tabTo(t, tt, "Token usage")
+	want := []string{"Plan · Input 1200", "Plan · Output 300", "Search · Input 4800"}
+	for i, w := range want {
+		tt.Key(0, ui.KeyRight)
+		if got := chartReadout(tt); got != w {
+			t.Fatalf("mark %d readout %q, want %q", i, got, w)
+		}
+	}
+	if !tt.HasText("Input") || !tt.HasText("Output") {
+		t.Fatal("legend missing")
+	}
+}
+
+func TestTokenUsageChartLineEmptyInvalid(t *testing.T) {
+	usage := tokenUsageSample()
+	h := &chartHarness{build: func(c *ui.Context) ChartResult {
+		return TokenUsageChart(c, usage, TokenUsageChartOptions{Line: true, Frame: ChartFrame{Label: "Tokens per step"}})
+	}}
+	tt := ui.NewTester(h.view, 520, 420)
+	tabTo(t, tt, "Tokens per step")
+	tt.Key(0, ui.KeyRight)
+	if h.res.Point < 0 {
+		t.Fatal("line chart has no keyboard path")
+	}
+	usage = nil
+	tt.Frame()
+	if !tt.HasText("No data") {
+		t.Fatalf("empty: %q", tt.Texts())
+	}
+	defer func() {
+		if recover() == nil {
+			t.Fatal("negative tokens did not panic")
+		}
+	}()
+	ui.Render(func(c *ui.Context) {
+		Use(c, Settings{})
+		TokenUsageChart(c, []TokenUsage{{Label: "x", Input: -1}}, TokenUsageChartOptions{})
+	}, 300, 200, 1)
+}
+
+func TestTokenUsageChartSnapshot(t *testing.T) {
+	usage := tokenUsageSample()
+	snapshot(t, "token_usage_chart", 520, 360, func(c *ui.Context) {
+		ui.Column(c).Padding(16).Children(func() {
+			TokenUsageChart(c, usage, TokenUsageChartOptions{})
+		})
+	})
+}