Merge pull request #6125 from q384566678/proxy

add unit test for CalculateSizeInMB
kubernetes · Dec 19, 2019 · 56c72e2 · 56c72e2
2 parents cf50577 + d864ac3
commit 56c72e2
Showing 1 changed file with 19 additions and 0 deletions.
diff --git a/pkg/util/utils_test.go b/pkg/util/utils_test.go
@@ -80,3 +80,22 @@ func TestConcatStrings(t *testing.T) {
 		}
 	}
 }
+
+func TestCalculateSizeInMB(t *testing.T) {
+	testData := []struct {
+		size           string
+		expectedNumber int
+	}{
+		{"1024kb", 1},
+		{"1024KB", 1},
+		{"1024mb", 1024},
+		{"1024b", 0},
+	}
+
+	for _, tt := range testData {
+		number := CalculateSizeInMB(tt.size)
+		if number != tt.expectedNumber {
+			t.Fatalf("Expected '%d'' but got '%d'", tt.expectedNumber, number)
+		}
+	}
+}