package util

import "testing"

func TestByteSizeString(t *testing.T) {
	tests := []struct {
		b    ByteSize
		want string
	}{
		{0, "0B"},
		{1500, "1500B"},
		{1024, "1KiB"},
		{1000, "1KB"},
		{32768, "32KiB"},
		{1000000, "1MB"},
		{1024000, "1000KiB"},
		{3 << 30, "3GiB"},
	}
	for _, tt := range tests {
		if got := tt.b.String(); got != tt.want {
			t.Errorf("ByteSize(%d).String() = %q; want %q", int64(tt.b), got, tt.want)
		}
		if back, err := ParseByteSize(tt.want); err != nil || back != tt.b {
			t.Errorf("ParseByteSize(%q) = %d, %v; want %d", tt.want, back, err, tt.b)
		}
	}
}