okami.mom/albedo

git clone https://git.nkp.pet/okami.mom/albedo

nekomimi.petfeat(usage): time provider calls and show tok/s on the turn signoff0a3eddd2026-10-07

// Clickable rows and selections that scroll are drawn and
// driven in the terminal client: the daemon e2e sees committed text, never
// rendered rows or mouse events, so these regressions are invisible outside
// the TUI.
package tui

import (
	"fmt"
	"slices"
	"strings"
	"testing"

	"albedo/cli/internal/daemon"
	tea "charm.land/bubbletea/v2"
	"github.com/charmbracelet/x/ansi"
)

// Dragging past the top edge scrolls the transcript and keeps selecting, so
// the copy reaches rows that were never on screen together.
func TestDraggingPastTheEdgeScrollsAndKeepsSelecting(t *testing.T) {
	m := newTestChatModel(t, &daemon.Session{ID: "s"})
	m.SetSize(80, 20)
	for i := range 40 {
		m.appendSettledEntry(HistoryEntry{Kind: EntryNote, Text: fmt.Sprintf("note %02d", i), Timestamp: int64(i + 1)})
	}
	m.refreshViewportContent()
	if !m.Follow || m.scrollLimit == 0 {
		t.Fatalf("fixture does not overflow: follow %v limit %d", m.Follow, m.scrollLimit)
	}
	first := 2 + m.Notices.ChromeRows()
	at := func(row int) tea.Mouse { return tea.Mouse{X: 10, Y: first + row, Button: tea.MouseLeft} }
	m, _ = m.Update(tea.MouseClickMsg(at(m.Viewport.Height() - 2)))
	m, cmd := m.Update(tea.MouseMotionMsg(at(0)))
	if cmd == nil || m.dragDir != -1 {
		t.Fatalf("the top edge does not start a scroll: dir %d", m.dragDir)
	}
	start := m.scrollOffset
	for range 5 {
		m, cmd = m.Update(dragScrollMsg{gen: m.dragGen})
	}
	if m.scrollOffset != start-5 || m.dragHead.Row != m.scrollOffset || cmd == nil {
		t.Fatalf("scrolled %d -> %d, head row %d", start, m.scrollOffset, m.dragHead.Row)
	}
	sel := Selection{Anchor: *m.dragAnchor, Head: m.dragHead, Gutter: railWidth}
	text := SelectedText(m.frameLines, sel)
	offscreen := strings.TrimSpace(ansi.Strip(m.frameLines[start-2]))
	if !strings.HasPrefix(offscreen, "note") || !strings.Contains(text, offscreen) {
		t.Fatalf("the copy stops at the screen, missing %q:\n%s", offscreen, text)
	}

	m, _ = m.Update(tea.MouseMotionMsg(at(m.Viewport.Height() / 2)))
	if m.dragDir != 0 {
		t.Fatalf("leaving the edge keeps scrolling: dir %d", m.dragDir)
	}
}

// The signoff copies the whole turn's prose as markdown, and the cells page lists every
// call with the one under the cursor in full.
func rowWith(t *testing.T, lines []string, text string) int {
	t.Helper()
	for i, line := range lines {
		if strings.Contains(ansi.Strip(line), text) {
			return i
		}
	}
	t.Fatalf("no row says %q", text)
	return -1
}

func TestSignoffCopiesTheTurnAsMarkdown(t *testing.T) {
	m := newTestChatModel(t, &daemon.Session{ID: "s"})
	m.SetSize(80, 40)
	m.appendSettledEntry(HistoryEntry{Kind: EntryUser, Text: "hi", Timestamp: 1})
	m.appendSettledEntry(HistoryEntry{Kind: EntryAssistant, Text: "# first\n\n```go\nx := 1\n```", Timestamp: 2})
	m.appendSettledEntry(HistoryEntry{Kind: EntryTool, ToolName: "bash", Timestamp: 3})
	m.appendSettledEntry(HistoryEntry{Kind: EntryAssistant, Text: "**second**", Timestamp: 4})
	m.appendSettledEntry(HistoryEntry{Kind: EntryTurnEnd, Mood: moodDone, Timestamp: 5})
	m.refreshViewportContent()
	row := rowWith(t, m.frameLines, "⧉")
	act, _ := actionOf(m.frameLines[row])
	if cmd := m.actAt(row); cmd == nil || m.CopyStatus != "copied" {
		t.Fatalf("the signoff did not copy: status %q", m.CopyStatus)
	}
	if got, _ := m.replyMarkdown(act.key); got != "# first\n\n```go\nx := 1\n```\n\n**second**" {
		t.Fatalf("the signoff copied %q", got)
	}
}

// The rate sits right before the copy mark, after the turn's time and tools.
func TestSignoffShowsItsRateBeforeTheCopyMark(t *testing.T) {
	m := newTestChatModel(t, &daemon.Session{ID: "s"})
	m.SetSize(80, 40)
	m.appendSettledEntry(HistoryEntry{Kind: EntryAssistant, Text: "done", Timestamp: 1})
	m.appendSettledEntry(HistoryEntry{Kind: EntryTurnEnd, Mood: moodDone, ElapsedMs: 5000, Tools: 1, TokensPerSecond: 84.2, Timestamp: 2})
	m.refreshViewportContent()
	row := ansi.Strip(m.frameLines[rowWith(t, m.frameLines, "⧉")])
	if !strings.Contains(row, "1 tool · 84.2 tok/s · ⧉") {
		t.Fatalf("signoff %q does not put the rate before the copy mark", row)
	}
}

// Clicking a burst's summary lists every step in full, the commands the
// summary only names included, and clicking it again folds them away.
func TestClickingABurstListsItsSteps(t *testing.T) {
	m := newTestChatModel(t, &daemon.Session{ID: "s", Workspace: "/w/albedo"})
	m.SetSize(120, 40)
	m.appendSettledEntry(HistoryEntry{Kind: EntryUser, Text: "go", Timestamp: 1})
	for _, entry := range burstFixture() {
		m.appendSettledEntry(entry)
	}
	m.appendSettledEntry(HistoryEntry{Kind: EntryAssistant, Text: "done", Timestamp: 9})
	m.refreshViewportContent()
	full := "cd cli && go test ./..."
	row := rowWith(t, m.frameLines, "▸ thought")
	if text := ansi.Strip(strings.Join(m.frameLines, "\n")); strings.Contains(text, full) {
		t.Fatalf("a closed burst lists its steps:\n%s", text)
	}
	m.actAt(row)
	spliced := slices.Clone(m.settledLines)
	if m.rebuildSettledLines(); !slices.Equal(spliced, m.settledLines) {
		t.Fatal("opening the burst drew rows a rebuild does not")
	}
	m.refreshViewportContent()
	text := ansi.Strip(strings.Join(m.frameLines, "\n"))
	for _, step := range []string{"▾ thought", "run    " + full, "read   cli/internal/tui/chat.go", "edit   priv/python/albedo_trace.py", "python · 1/0"} {
		if !strings.Contains(text, step) {
			t.Fatalf("the open burst does not show %q:\n%s", step, text)
		}
	}
	m.actAt(rowWith(t, m.frameLines, "▾ thought"))
	if text := ansi.Strip(strings.Join(m.frameLines, "\n")); strings.Contains(text, full) || !strings.Contains(text, "▸ thought") {
		t.Fatalf("a second click does not close the burst:\n%s", text)
	}
}

// A long message of yours shows its first rows and a click target; clicking
// it shows the rest, and the row that closes it folds the message again.
func TestClickingALongMessageShowsTheRestAndFoldsIt(t *testing.T) {
	m := newTestChatModel(t, &daemon.Session{ID: "s"})
	m.SetSize(100, 30)
	var lines []string
	for i := range 40 {
		lines = append(lines, fmt.Sprintf("line %02d", i))
	}
	m.appendSettledEntry(HistoryEntry{Kind: EntryUser, Text: strings.Join(lines, "\n\n"), Timestamp: 1})
	m.appendSettledEntry(HistoryEntry{Kind: EntryAssistant, Text: "done", Timestamp: 2})
	m.refreshViewportContent()
	shown := func() string { return ansi.Strip(strings.Join(m.frameLines, "\n")) }
	if text := shown(); strings.Contains(text, "line 39") || !strings.Contains(text, "more lines · click to expand") {
		t.Fatalf("a long message is not folded:\n%s", text)
	}
	spliced := func() {
		t.Helper()
		got := slices.Clone(m.settledLines)
		if m.rebuildSettledLines(); !slices.Equal(got, m.settledLines) {
			t.Fatal("the splice drew rows a rebuild does not")
		}
		m.refreshViewportContent()
	}
	click := rowWith(t, m.frameLines, "click to expand")
	m.actAt(click)
	spliced()
	if text := shown(); !strings.Contains(text, "line 39") || !strings.Contains(text, "show less") {
		t.Fatalf("clicking does not show the rest:\n%s", text)
	}
	m.actAt(rowWith(t, m.frameLines, "show less"))
	spliced()
	if text := shown(); strings.Contains(text, "line 39") || !strings.Contains(text, "click to expand") {
		t.Fatalf("clicking again does not fold the message:\n%s", text)
	}
}

// A web search's answer shows under its burst, cut to its first rows; the
// click target shows the rest and folds it again, splicing what a rebuild
// would draw.
func TestClickingAWebSearchShowsItsWholeAnswer(t *testing.T) {
	m := newTestChatModel(t, &daemon.Session{ID: "s"})
	m.SetSize(100, 30)
	var lines []string
	for i := range 12 {
		lines = append(lines, fmt.Sprintf("finding %02d", i))
	}
	search := daemon.ToolActivity{Kind: "web", Target: "latest gleam release", Detail: strings.Join(lines, "\n\n")}
	m.appendSettledEntry(HistoryEntry{Kind: EntryUser, Text: "look it up", Timestamp: 1})
	m.appendSettledEntry(traced([]daemon.ToolActivity{search}))
	m.appendSettledEntry(HistoryEntry{Kind: EntryAssistant, Text: "done", Timestamp: 3})
	m.refreshViewportContent()
	shown := func() string { return ansi.Strip(strings.Join(m.frameLines, "\n")) }
	if text := shown(); !strings.Contains(text, "looked up “latest gleam release”") || !strings.Contains(text, "finding 00") || strings.Contains(text, "finding 11") {
		t.Fatalf("the search is not summarized and folded:\n%s", text)
	}
	spliced := func() {
		t.Helper()
		got := slices.Clone(m.settledLines)
		if m.rebuildSettledLines(); !slices.Equal(got, m.settledLines) {
			t.Fatal("the splice drew rows a rebuild does not")
		}
		m.refreshViewportContent()
	}
	m.actAt(rowWith(t, m.frameLines, "click to expand"))
	spliced()
	if text := shown(); !strings.Contains(text, "finding 11") || !strings.Contains(text, "show less") {
		t.Fatalf("clicking does not show the whole answer:\n%s", text)
	}
	m.actAt(rowWith(t, m.frameLines, "show less"))
	spliced()
	if text := shown(); strings.Contains(text, "finding 11") || !strings.Contains(text, "click to expand") {
		t.Fatalf("clicking again does not fold the answer:\n%s", text)
	}
}

// Capability notes must not pad short rows to thousands of columns.
func TestALongNoteWrapsRowByRowAndFolds(t *testing.T) {
	m := newTestChatModel(t, &daemon.Session{ID: "s"})
	m.SetSize(100, 30)
	lines := []string{"Session capabilities changed.", strings.Repeat("instructions ", 200)}
	for i := range 20 {
		lines = append(lines, fmt.Sprintf("- **`dir%02d/`** — a short bullet", i))
	}
	m.appendSettledEntry(HistoryEntry{Kind: EntryNote, Source: "capabilities changed", Text: strings.Join(lines, "\n"), Timestamp: 1})
	m.appendSettledEntry(HistoryEntry{Kind: EntryAssistant, Text: "done", Timestamp: 2})
	m.refreshViewportContent()
	bounded := func() {
		t.Helper()
		for _, row := range m.settledLines {
			if ansi.StringWidth(row) > 100 {
				t.Fatalf("a note row is wider than the screen: %d columns", ansi.StringWidth(row))
			}
			if plain := ansi.Strip(row); strings.TrimSpace(plain) != "" && strings.HasSuffix(plain, "  ") {
				t.Fatalf("a note row is padded to a wider row: %q", plain)
			}
		}
	}
	spliced := func() {
		t.Helper()
		got := slices.Clone(m.settledLines)
		m.rebuildSettledLines()
		if !slices.Equal(got, m.settledLines) {
			t.Fatal("expanding or folding the note disagrees with a rebuild")
		}
		m.refreshViewportContent()
		bounded()
	}
	shown := func() string { return ansi.Strip(strings.Join(m.frameLines, "\n")) }
	bounded()
	if text := shown(); strings.Contains(text, "dir19/") || !strings.Contains(text, "more lines · click to expand") {
		t.Fatalf("a long note is not folded:\n%s", text)
	}
	m.actAt(rowWith(t, m.frameLines, "click to expand"))
	spliced()
	if text := shown(); !strings.Contains(text, "**`dir19/`**") || !strings.Contains(text, "show less") {
		t.Fatalf("clicking does not show the rest of the plain-text note:\n%s", text)
	}
	m.actAt(rowWith(t, m.frameLines, "show less"))
	spliced()
	if text := shown(); strings.Contains(text, "dir19/") || !strings.Contains(text, "click to expand") {
		t.Fatalf("clicking again does not fold the note:\n%s", text)
	}
}