forked from TrueCloudLab/restic
89 lines
1.8 KiB
Go
89 lines
1.8 KiB
Go
package termstatus
|
|
|
|
import (
|
|
"strconv"
|
|
"testing"
|
|
|
|
rtest "github.com/restic/restic/internal/test"
|
|
)
|
|
|
|
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 _, r := range s {
|
|
w++
|
|
if wideRune(r) {
|
|
w++
|
|
}
|
|
}
|
|
benchmarkTruncate(b, s, w-1)
|
|
}
|