pkg/units/size_test.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 TestFromHumanSize(t *testing.T) {
	assertFromHumanSize(t, "32", false, 32)
	assertFromHumanSize(t, "32b", false, 32)
	assertFromHumanSize(t, "32B", false, 32)
	assertFromHumanSize(t, "32k", false, 32*1000)
	assertFromHumanSize(t, "32K", false, 32*1000)
	assertFromHumanSize(t, "32kb", false, 32*1000)
	assertFromHumanSize(t, "32Kb", false, 32*1000)
	assertFromHumanSize(t, "32Mb", false, 32*1000*1000)
	assertFromHumanSize(t, "32Gb", false, 32*1000*1000*1000)
	assertFromHumanSize(t, "32Tb", false, 32*1000*1000*1000*1000)
	assertFromHumanSize(t, "8Pb", false, 8*1000*1000*1000*1000*1000)

	assertFromHumanSize(t, "", true, -1)
	assertFromHumanSize(t, "hello", true, -1)
	assertFromHumanSize(t, "-32", true, -1)
	assertFromHumanSize(t, " 32 ", true, -1)
	assertFromHumanSize(t, "32 mb", true, -1)
	assertFromHumanSize(t, "32m b", true, -1)
	assertFromHumanSize(t, "32bm", true, -1)
}

func assertFromHumanSize(t *testing.T, size string, expectError bool, expectedBytes int64) {
	actualBytes, err := FromHumanSize(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)
	}
}

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)
	}
}