1
0
Fork 0
mirror of https://github.com/moby/moby.git synced 2022-11-09 12:21:53 -05:00
moby--moby/pkg/ansiescape/split_test.go

54 lines
1.3 KiB
Go
Raw Normal View History

package ansiescape
import (
"bufio"
"strings"
"testing"
)
func TestSplit(t *testing.T) {
lines := []string{
"test line 1",
"another test line",
"some test line",
"line with non-cursor moving sequence \x1b[1T", // Scroll Down
"line with \x1b[31;1mcolor\x1b[0m then reset", // "color" in Bold Red
"cursor forward \x1b[1C and backward \x1b[1D",
"invalid sequence \x1babcd",
"",
"after empty",
}
splitSequences := []string{
"\x1b[1A", // Cursor up
"\x1b[1B", // Cursor down
"\x1b[1E", // Cursor next line
"\x1b[1F", // Cursor previous line
"\x1b[1;1H", // Move cursor to position
"\x1b[1;1h", // Move cursor to position
"\n",
"\r\n",
"\n\r",
"\x1b[1A\r",
"\r\x1b[1A",
}
for _, sequence := range splitSequences {
scanner := bufio.NewScanner(strings.NewReader(strings.Join(lines, sequence)))
scanner.Split(ScanANSILines)
i := 0
for scanner.Scan() {
if i >= len(lines) {
t.Fatalf("Too many scanned lines")
}
scanned := scanner.Text()
if scanned != lines[i] {
t.Fatalf("Wrong line scanned with sequence %q\n\tExpected: %q\n\tActual: %q", sequence, lines[i], scanned)
}
i++
}
if i < len(lines) {
t.Errorf("Wrong number of lines for sequence %q: %d, expected %d", sequence, i, len(lines))
}
}
}