mirror of https://github.com/restic/restic.git
105 lines
2.0 KiB
Go
105 lines
2.0 KiB
Go
package ui
|
||
|
||
import (
|
||
"testing"
|
||
|
||
"github.com/restic/restic/internal/test"
|
||
)
|
||
|
||
func TestFormatBytes(t *testing.T) {
|
||
for _, c := range []struct {
|
||
size uint64
|
||
want string
|
||
}{
|
||
{0, "0 B"},
|
||
{1023, "1023 B"},
|
||
{1024, "1.000 KiB"},
|
||
{5<<20 + 1<<19, "5.500 MiB"},
|
||
{1 << 30, "1.000 GiB"},
|
||
{2 << 30, "2.000 GiB"},
|
||
{1<<40 - 1<<36, "960.000 GiB"},
|
||
{1 << 40, "1.000 TiB"},
|
||
} {
|
||
if got := FormatBytes(c.size); got != c.want {
|
||
t.Errorf("want %q, got %q", c.want, got)
|
||
}
|
||
}
|
||
}
|
||
|
||
func TestFormatPercent(t *testing.T) {
|
||
for _, c := range []struct {
|
||
num, denom uint64
|
||
want string
|
||
}{
|
||
{0, 5, "0.00%"},
|
||
{3, 7, "42.86%"},
|
||
{99, 99, "100.00%"},
|
||
} {
|
||
if got := FormatPercent(c.num, c.denom); got != c.want {
|
||
t.Errorf("want %q, got %q", c.want, got)
|
||
}
|
||
}
|
||
}
|
||
|
||
func TestParseBytes(t *testing.T) {
|
||
for _, tt := range []struct {
|
||
in string
|
||
expected int64
|
||
}{
|
||
{"1024", 1024},
|
||
{"1024b", 1024},
|
||
{"1024B", 1024},
|
||
{"1k", 1024},
|
||
{"100k", 102400},
|
||
{"100K", 102400},
|
||
{"10M", 10485760},
|
||
{"100m", 104857600},
|
||
{"20G", 21474836480},
|
||
{"10g", 10737418240},
|
||
{"2T", 2199023255552},
|
||
{"2t", 2199023255552},
|
||
{"9223372036854775807", 1<<63 - 1},
|
||
} {
|
||
actual, err := ParseBytes(tt.in)
|
||
test.OK(t, err)
|
||
test.Equals(t, tt.expected, actual)
|
||
}
|
||
}
|
||
|
||
func TestParseBytesInvalid(t *testing.T) {
|
||
for _, s := range []string{
|
||
"",
|
||
" ",
|
||
"foobar",
|
||
"zzz",
|
||
"18446744073709551615", // 1<<64-1.
|
||
"9223372036854775807k", // 1<<63-1 kiB.
|
||
"9999999999999M",
|
||
"99999999999999999999",
|
||
} {
|
||
v, err := ParseBytes(s)
|
||
if err == nil {
|
||
t.Errorf("wanted error for invalid value %q, got nil", s)
|
||
}
|
||
test.Equals(t, int64(0), v)
|
||
}
|
||
}
|
||
|
||
func TestTerminalDisplayWidth(t *testing.T) {
|
||
for _, c := range []struct {
|
||
input string
|
||
want int
|
||
}{
|
||
{"foo", 3},
|
||
{"aéb", 3},
|
||
{"ab", 3},
|
||
{"a’b", 3},
|
||
{"aあb", 4},
|
||
} {
|
||
if got := TerminalDisplayWidth(c.input); got != c.want {
|
||
t.Errorf("wrong display width for '%s', want %d, got %d", c.input, c.want, got)
|
||
}
|
||
}
|
||
|
||
}
|