forked from TrueCloudLab/restic
155 lines
3.4 KiB
Go
155 lines
3.4 KiB
Go
package termstatus
|
|
|
|
import (
|
|
"bytes"
|
|
"context"
|
|
"fmt"
|
|
"io"
|
|
"strconv"
|
|
"testing"
|
|
|
|
rtest "github.com/restic/restic/internal/test"
|
|
)
|
|
|
|
func TestSetStatus(t *testing.T) {
|
|
var buf bytes.Buffer
|
|
term := New(&buf, io.Discard, false)
|
|
|
|
term.canUpdateStatus = true
|
|
term.fd = ^uintptr(0)
|
|
term.clearCurrentLine = posixClearCurrentLine
|
|
term.moveCursorUp = posixMoveCursorUp
|
|
|
|
ctx, cancel := context.WithCancel(context.Background())
|
|
go term.Run(ctx)
|
|
|
|
const (
|
|
clear = posixControlClearLine
|
|
home = posixControlMoveCursorHome
|
|
up = posixControlMoveCursorUp
|
|
)
|
|
|
|
term.SetStatus([]string{"first"})
|
|
exp := home + clear + "first" + home
|
|
|
|
term.SetStatus([]string{"foo", "bar", "baz"})
|
|
exp += home + clear + "foo\n" + home + clear + "bar\n" +
|
|
home + clear + "baz" + home + up + up
|
|
|
|
term.SetStatus([]string{"quux", "needs\nquote"})
|
|
exp += home + clear + "quux\n" +
|
|
home + clear + "\"needs\\nquote\"\n" +
|
|
home + clear + home + up + up // Third line implicit.
|
|
|
|
cancel()
|
|
exp += home + clear + "\n" + home + clear + "\n" +
|
|
home + up + up // Status cleared.
|
|
|
|
<-term.closed
|
|
rtest.Equals(t, exp, buf.String())
|
|
}
|
|
|
|
func TestQuote(t *testing.T) {
|
|
for _, c := range []struct {
|
|
in string
|
|
needQuote bool
|
|
}{
|
|
{"foo.bar/baz", false},
|
|
{"föó_bàŕ-bãẑ", false},
|
|
{" foo ", false},
|
|
{"foo bar", false},
|
|
{"foo\nbar", true},
|
|
{"foo\rbar", true},
|
|
{"foo\abar", true},
|
|
{"\xff", true},
|
|
{`c:\foo\bar`, false},
|
|
// Issue #2260: terminal control characters.
|
|
{"\x1bm_red_is_beautiful", true},
|
|
} {
|
|
if c.needQuote {
|
|
rtest.Equals(t, strconv.Quote(c.in), Quote(c.in))
|
|
} else {
|
|
rtest.Equals(t, c.in, Quote(c.in))
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestTruncate(t *testing.T) {
|
|
var tests = []struct {
|
|
input string
|
|
width int
|
|
output string
|
|
}{
|
|
{"", 80, ""},
|
|
{"", 0, ""},
|
|
{"", -1, ""},
|
|
{"foo", 80, "foo"},
|
|
{"foo", 4, "foo"},
|
|
{"foo", 3, "foo"},
|
|
{"foo", 2, "fo"},
|
|
{"foo", 1, "f"},
|
|
{"foo", 0, ""},
|
|
{"foo", -1, ""},
|
|
{"Löwen", 4, "Löwe"},
|
|
{"あああああ/data", 7, "あああ"},
|
|
{"あああああ/data", 10, "あああああ"},
|
|
{"あああああ/data", 11, "あああああ/"},
|
|
}
|
|
|
|
for _, test := range tests {
|
|
t.Run("", func(t *testing.T) {
|
|
out := Truncate(test.input, test.width)
|
|
if out != test.output {
|
|
t.Fatalf("wrong output for input %v, width %d: want %q, got %q",
|
|
test.input, test.width, test.output, out)
|
|
}
|
|
})
|
|
}
|
|
}
|
|
|
|
func benchmarkTruncate(b *testing.B, s string, w int) {
|
|
for i := 0; i < b.N; i++ {
|
|
Truncate(s, w)
|
|
}
|
|
}
|
|
|
|
func BenchmarkTruncateASCII(b *testing.B) {
|
|
s := "This is an ASCII-only status message...\r\n"
|
|
benchmarkTruncate(b, s, len(s)-1)
|
|
}
|
|
|
|
func BenchmarkTruncateUnicode(b *testing.B) {
|
|
s := "Hello World or Καλημέρα κόσμε or こんにちは 世界"
|
|
w := 0
|
|
for i := 0; i < len(s); {
|
|
w++
|
|
wide, utfsize := wideRune(s[i:])
|
|
if wide {
|
|
w++
|
|
}
|
|
i += int(utfsize)
|
|
}
|
|
b.ResetTimer()
|
|
|
|
benchmarkTruncate(b, s, w-1)
|
|
}
|
|
|
|
func TestSanitizeLines(t *testing.T) {
|
|
var tests = []struct {
|
|
input []string
|
|
width int
|
|
output []string
|
|
}{
|
|
{[]string{""}, 80, []string{""}},
|
|
{[]string{"too long test line"}, 10, []string{"too long"}},
|
|
{[]string{"too long test line", "text"}, 10, []string{"too long\n", "text"}},
|
|
{[]string{"too long test line", "second long test line"}, 10, []string{"too long\n", "second l"}},
|
|
}
|
|
|
|
for _, test := range tests {
|
|
t.Run(fmt.Sprintf("%s %d", test.input, test.width), func(t *testing.T) {
|
|
out := sanitizeLines(test.input, test.width)
|
|
rtest.Equals(t, test.output, out)
|
|
})
|
|
}
|
|
}
|