storj/private/memory/size_test.go

186 lines
3.5 KiB
Go
Raw Normal View History

2019-01-24 20:15:10 +00:00
// Copyright (C) 2019 Storj Labs, Inc.
2018-10-01 12:29:27 +01:00
// See LICENSE for copying information.
package memory_test
import (
"testing"
"storj.io/storj/private/memory"
)
const (
2019-01-15 22:17:37 +00:00
eib = 1 << 60
pib = 1 << 50
tib = 1 << 40
gib = 1 << 30
mib = 1 << 20
kib = 1 << 10
eb = 1e18
pb = 1e15
tb = 1e12
gb = 1e9
mb = 1e6
kb = 1e3
)
2019-01-15 22:17:37 +00:00
func TestBase2Size(t *testing.T) {
var tests = []struct {
size memory.Size
text string
}{
// basics
2019-01-15 22:17:37 +00:00
{1 * eib, "1.0 EiB"},
{1 * pib, "1.0 PiB"},
{1 * tib, "1.0 TiB"},
{1 * gib, "1.0 GiB"},
{1 * mib, "1.0 MiB"},
{1 * kib, "1.0 KiB"},
{1, "1 B"},
// complicated
{68 * tib, "68.0 TiB"},
{256 * mib, "256.0 MiB"},
{500, "500 B"},
{5, "5 B"},
{1, "1 B"},
{0, "0 B"},
2019-01-15 22:17:37 +00:00
}
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)
}
// test negative
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)
}
2019-01-15 22:17:37 +00:00
}
}
func TestBase10Size(t *testing.T) {
var tests = []struct {
size memory.Size
text string
}{
// basics
{1 * pb, "1.0 PB"},
{1 * eb, "1.0 EB"},
{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 B"},
}
for i, test := range tests {
2019-01-15 22:17:37 +00:00
if test.size.Base10String() != 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)
}
// test negative
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"},
2019-01-15 22:17:37 +00:00
{1 * kib, "1.0kib"},
{1 * pib, "1.0pib"},
{1 * eib, "1.0eib"},
{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)
}
// test negative
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)
}
}
}