105 lines
1.8 KiB
Go
105 lines
1.8 KiB
Go
package memory_test
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"storj.io/storj/internal/memory"
|
|
)
|
|
|
|
const (
|
|
tb = 1 << 40
|
|
gb = 1 << 30
|
|
mb = 1 << 20
|
|
kb = 1 << 10
|
|
)
|
|
|
|
func TestSize(t *testing.T) {
|
|
var tests = []struct {
|
|
size memory.Size
|
|
text string
|
|
}{
|
|
// basics
|
|
{1 * tb, "1.0 TB"},
|
|
{1 * gb, "1.0 GB"},
|
|
{1 * mb, "1.0 MB"},
|
|
{1 * kb, "1.0 KB"},
|
|
{1, "1 B"},
|
|
// complicated
|
|
{68 * tb, "68.0 TB"},
|
|
{256 * mb, "256.0 MB"},
|
|
{500, "500 B"},
|
|
{5, "5 B"},
|
|
{1, "1 B"},
|
|
{0, "0"},
|
|
}
|
|
|
|
for i, test := range tests {
|
|
if test.size.String() != test.text {
|
|
t.Errorf("%d. invalid text got %v expected %v", i, test.size.String(), test.text)
|
|
}
|
|
|
|
var size memory.Size
|
|
err := size.Set(test.text)
|
|
if err != nil {
|
|
t.Errorf("%d. got error %v", i, err)
|
|
}
|
|
if test.size != size {
|
|
t.Errorf("%d. invalid size got %d expected %d", i, size, test.size)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestParse(t *testing.T) {
|
|
var tests = []struct {
|
|
size memory.Size
|
|
text string
|
|
}{
|
|
// case insensitivity
|
|
{1 * tb, "1.00TB"},
|
|
{1 * gb, "1.00gB"},
|
|
{1 * mb, "1.00Mb"},
|
|
{1 * kb, "1.00kb"},
|
|
{1, "1.00"},
|
|
{1 * tb, "1.0 TB"},
|
|
{1 * gb, "1.0 gB"},
|
|
{1 * mb, "1.0 Mb"},
|
|
{1 * kb, "1.0 kb"},
|
|
{1, "1.00"},
|
|
// without B suffix
|
|
{1 * tb, "1.00T"},
|
|
{1 * gb, "1.00g"},
|
|
{1 * mb, "1.00M"},
|
|
{1 * kb, "1.00k"},
|
|
}
|
|
|
|
for i, test := range tests {
|
|
var size memory.Size
|
|
err := size.Set(test.text)
|
|
if err != nil {
|
|
t.Errorf("%d. got error %v", i, err)
|
|
}
|
|
if test.size != size {
|
|
t.Errorf("%d. invalid size got %d expected %d", i, size, test.size)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestInvalidParse(t *testing.T) {
|
|
var tests = []string{
|
|
"1.0Q",
|
|
"1.0QB",
|
|
"z1.0KB",
|
|
"z1.0Q",
|
|
"1.0zQ",
|
|
"1.0zQB",
|
|
}
|
|
|
|
for i, test := range tests {
|
|
var size memory.Size
|
|
err := size.Set(test)
|
|
if err == nil {
|
|
t.Errorf("%d. didn't get error", i)
|
|
}
|
|
}
|
|
}
|