68928c3485
Docker-DCO-1.1-Signed-off-by: Michael Crosby <michael@crosbymichael.com> (github: crosbymichael)
54 lines
1.5 KiB
Go
54 lines
1.5 KiB
Go
package units
|
|
|
|
import (
|
|
"strings"
|
|
"testing"
|
|
)
|
|
|
|
func TestHumanSize(t *testing.T) {
|
|
|
|
size := strings.Trim(HumanSize(1000), " \t")
|
|
expect := "1 kB"
|
|
if size != expect {
|
|
t.Errorf("1000 -> expected '%s', got '%s'", expect, size)
|
|
}
|
|
|
|
size = strings.Trim(HumanSize(1024), " \t")
|
|
expect = "1.024 kB"
|
|
if size != expect {
|
|
t.Errorf("1024 -> expected '%s', got '%s'", expect, size)
|
|
}
|
|
}
|
|
|
|
func TestRAMInBytes(t *testing.T) {
|
|
assertRAMInBytes(t, "32", false, 32)
|
|
assertRAMInBytes(t, "32b", false, 32)
|
|
assertRAMInBytes(t, "32B", false, 32)
|
|
assertRAMInBytes(t, "32k", false, 32*1024)
|
|
assertRAMInBytes(t, "32K", false, 32*1024)
|
|
assertRAMInBytes(t, "32kb", false, 32*1024)
|
|
assertRAMInBytes(t, "32Kb", false, 32*1024)
|
|
assertRAMInBytes(t, "32Mb", false, 32*1024*1024)
|
|
assertRAMInBytes(t, "32Gb", false, 32*1024*1024*1024)
|
|
|
|
assertRAMInBytes(t, "", true, -1)
|
|
assertRAMInBytes(t, "hello", true, -1)
|
|
assertRAMInBytes(t, "-32", true, -1)
|
|
assertRAMInBytes(t, " 32 ", true, -1)
|
|
assertRAMInBytes(t, "32 mb", true, -1)
|
|
assertRAMInBytes(t, "32m b", true, -1)
|
|
assertRAMInBytes(t, "32bm", true, -1)
|
|
}
|
|
|
|
func assertRAMInBytes(t *testing.T, size string, expectError bool, expectedBytes int64) {
|
|
actualBytes, err := RAMInBytes(size)
|
|
if (err != nil) && !expectError {
|
|
t.Errorf("Unexpected error parsing '%s': %s", size, err)
|
|
}
|
|
if (err == nil) && expectError {
|
|
t.Errorf("Expected to get an error parsing '%s', but got none (bytes=%d)", size, actualBytes)
|
|
}
|
|
if actualBytes != expectedBytes {
|
|
t.Errorf("Expected '%s' to parse as %d bytes, got %d", size, expectedBytes, actualBytes)
|
|
}
|
|
}
|