size_test: add more tests

Co-authored-by: Kir Kolyshkin <kolyshkin@gmail.com>
docker · May 16, 2022 · cec4960 · cec4960
1 parent 54c3e55
commit cec4960
Showing 1 changed file with 50 additions and 0 deletions.
diff --git a/size_test.go b/size_test.go
@@ -83,6 +83,10 @@ func TestHumanSize(t *testing.T) {
 }
 
 func TestFromHumanSize(t *testing.T) {
+	assertSuccessEquals(t, 0, FromHumanSize, "0")
+	assertSuccessEquals(t, 0, FromHumanSize, "0b")
+	assertSuccessEquals(t, 0, FromHumanSize, "0B")
+	assertSuccessEquals(t, 0, FromHumanSize, "0 B")
 	assertSuccessEquals(t, 32, FromHumanSize, "32")
 	assertSuccessEquals(t, 32, FromHumanSize, "32b")
 	assertSuccessEquals(t, 32, FromHumanSize, "32B")
@@ -100,10 +104,56 @@ func TestFromHumanSize(t *testing.T) {
 	assertSuccessEquals(t, 32, FromHumanSize, "32.5 B")
 	assertSuccessEquals(t, 300, FromHumanSize, "0.3 K")
 
+	// We do not tolerate extra leading or trailing spaces
+	// (except for a space after the number and a missing suffix).
+	assertSuccessEquals(t, 0, FromHumanSize, "0 ")
+
+	assertError(t, FromHumanSize, " 0")
+	assertError(t, FromHumanSize, " 0b")
+	assertError(t, FromHumanSize, " 0B")
+	assertError(t, FromHumanSize, " 0 B")
+	assertError(t, FromHumanSize, "0b ")
+	assertError(t, FromHumanSize, "0B ")
+	assertError(t, FromHumanSize, "0 B ")
+
 	assertError(t, FromHumanSize, "")
 	assertError(t, FromHumanSize, "hello")
+	assertError(t, FromHumanSize, ".")
+	assertError(t, FromHumanSize, ". ")
+	assertError(t, FromHumanSize, " ")
+	assertError(t, FromHumanSize, "  ")
+	assertError(t, FromHumanSize, " .")
+	assertError(t, FromHumanSize, " . ")
+	assertError(t, FromHumanSize, "0.")
+	assertError(t, FromHumanSize, "0. ")
+	assertError(t, FromHumanSize, "0.b")
+	assertError(t, FromHumanSize, "0.B")
+	assertError(t, FromHumanSize, "-0")
+	assertError(t, FromHumanSize, "-0b")
+	assertError(t, FromHumanSize, "-0B")
+	assertError(t, FromHumanSize, "-0 b")
+	assertError(t, FromHumanSize, "-0 B")
 	assertError(t, FromHumanSize, "-32")
 	assertError(t, FromHumanSize, ".3kB")
+	assertError(t, FromHumanSize, "-32b")
+	assertError(t, FromHumanSize, "-32B")
+	assertError(t, FromHumanSize, "-32 b")
+	assertError(t, FromHumanSize, "-32 B")
+	assertError(t, FromHumanSize, "32.")
+	assertError(t, FromHumanSize, "32.b")
+	assertError(t, FromHumanSize, "32.B")
+	assertError(t, FromHumanSize, "32. b")
+	assertError(t, FromHumanSize, "32. B")
+	assertError(t, FromHumanSize, "32b.")
+	assertError(t, FromHumanSize, "32B.")
+	assertError(t, FromHumanSize, "32 b.")
+	assertError(t, FromHumanSize, "32 B.")
+	assertError(t, FromHumanSize, "32 bb")
+	assertError(t, FromHumanSize, "32 BB")
+	assertError(t, FromHumanSize, "32 b b")
+	assertError(t, FromHumanSize, "32 B B")
+	assertError(t, FromHumanSize, "32  b")
+	assertError(t, FromHumanSize, "32  B")
 	assertError(t, FromHumanSize, " 32 ")
 	assertError(t, FromHumanSize, "32m b")
 	assertError(t, FromHumanSize, "32bm")