diff --git a/pkg/util/utils_test.go b/pkg/util/utils_test.go
index 26a10724a4..1be9f669fc 100644
--- 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)
+		}
+	}
+}