81 lines
1.8 KiB
Go
81 lines
1.8 KiB
Go
|
package pb
|
||
|
|
|
||
|
|
import (
|
||
|
|
"github.com/fatih/color"
|
||
|
|
"testing"
|
||
|
|
)
|
||
|
|
|
||
|
|
var testColorString = color.RedString("red") +
|
||
|
|
color.GreenString("hello") +
|
||
|
|
"simple" +
|
||
|
|
color.WhiteString("進捗")
|
||
|
|
|
||
|
|
func TestUtilCellCount(t *testing.T) {
|
||
|
|
if e, l := 18, CellCount(testColorString); l != e {
|
||
|
|
t.Errorf("Invalid length %d, expected %d", l, e)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestUtilStripString(t *testing.T) {
|
||
|
|
if r, e := StripString("12345", 4), "1234"; r != e {
|
||
|
|
t.Errorf("Invalid result '%s', expected '%s'", r, e)
|
||
|
|
}
|
||
|
|
|
||
|
|
if r, e := StripString("12345", 5), "12345"; r != e {
|
||
|
|
t.Errorf("Invalid result '%s', expected '%s'", r, e)
|
||
|
|
}
|
||
|
|
if r, e := StripString("12345", 10), "12345"; r != e {
|
||
|
|
t.Errorf("Invalid result '%s', expected '%s'", r, e)
|
||
|
|
}
|
||
|
|
|
||
|
|
s := color.RedString("1") + "23"
|
||
|
|
e := color.RedString("1") + "2"
|
||
|
|
if r := StripString(s, 2); r != e {
|
||
|
|
t.Errorf("Invalid result '%s', expected '%s'", r, e)
|
||
|
|
}
|
||
|
|
return
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestUtilRound(t *testing.T) {
|
||
|
|
if v := round(4.4); v != 4 {
|
||
|
|
t.Errorf("Unexpected result: %v", v)
|
||
|
|
}
|
||
|
|
if v := round(4.501); v != 5 {
|
||
|
|
t.Errorf("Unexpected result: %v", v)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestUtilFormatBytes(t *testing.T) {
|
||
|
|
inputs := []struct {
|
||
|
|
v int64
|
||
|
|
s bool
|
||
|
|
e string
|
||
|
|
}{
|
||
|
|
{v: 1000, s: false, e: "1000 B"},
|
||
|
|
{v: 1024, s: false, e: "1.00 KiB"},
|
||
|
|
{v: 3*_MiB + 140*_KiB, s: false, e: "3.14 MiB"},
|
||
|
|
{v: 2 * _GiB, s: false, e: "2.00 GiB"},
|
||
|
|
{v: 2048 * _GiB, s: false, e: "2.00 TiB"},
|
||
|
|
|
||
|
|
{v: 999, s: true, e: "999 B"},
|
||
|
|
{v: 1024, s: true, e: "1.02 kB"},
|
||
|
|
{v: 3*_MB + 140*_kB, s: true, e: "3.14 MB"},
|
||
|
|
{v: 2 * _GB, s: true, e: "2.00 GB"},
|
||
|
|
{v: 2048 * _GB, s: true, e: "2.05 TB"},
|
||
|
|
}
|
||
|
|
|
||
|
|
for _, input := range inputs {
|
||
|
|
actual := formatBytes(input.v, input.s)
|
||
|
|
if actual != input.e {
|
||
|
|
t.Errorf("Expected {%s} was {%s}", input.e, actual)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func BenchmarkUtilsCellCount(b *testing.B) {
|
||
|
|
b.ReportAllocs()
|
||
|
|
for i := 0; i < b.N; i++ {
|
||
|
|
CellCount(testColorString)
|
||
|
|
}
|
||
|
|
}
|