Files
2026-07-13 13:00:08 +08:00

51 lines
1.5 KiB
Go

package agent
import (
"strings"
"testing"
)
// TestStreamedRowsBasic covers the four cursor-position cases the markdown
// redraw uses to decide how far up to move before clearing.
func TestStreamedRowsBasic(t *testing.T) {
cases := []struct {
name string
input string
width int
want int
}{
{"no newline fits", "hello", 80, 0},
{"newline only", "hello\n", 80, 1},
{"two lines no trailing", "hello\nworld", 80, 1},
{"two lines trailing", "hello\nworld\n", 80, 2},
{"empty", "", 80, 0},
{"single wrap", "abcdefghij", 5, 1}, // 10 cols / width 5 → 1 wrap
{"two wraps no nl", "abcdefghijklmno", 5, 2},
{"line exactly width", strings.Repeat("a", 80), 80, 0}, // lazy wrap, no extra row
}
for _, tc := range cases {
t.Run(tc.name, func(t *testing.T) {
if got := streamedRows(tc.input, tc.width); got != tc.want {
t.Errorf("streamedRows(%q, %d) = %d, want %d", tc.input, tc.width, got, tc.want)
}
})
}
}
// TestStreamedRowsCJK proves CJK doubles the column footprint so a Chinese
// line wraps at half the column count.
func TestStreamedRowsCJK(t *testing.T) {
in := strings.Repeat("中", 6) // 12 cols at width 10 → 1 wrap
if got := streamedRows(in, 10); got != 1 {
t.Errorf("streamedRows(6×中, 10) = %d, want 1", got)
}
}
// TestStreamedRowsIgnoresAnsi: ANSI SGR codes must not inflate the row count.
func TestStreamedRowsIgnoresAnsi(t *testing.T) {
in := "\x1b[1mhello\x1b[0m world"
if got := streamedRows(in, 80); got != 0 {
t.Errorf("ANSI in 11-char line at width 80 should be 0 rows, got %d", got)
}
}