From d864ac3f04dabfd2caf55b87c9bd74cb7323021a Mon Sep 17 00:00:00 2001
From: Zhou Hao <zhouhao@cn.fujitsu.com>
Date: Thu, 19 Dec 2019 17:48:02 +0800
Subject: [PATCH] pkg/util: add CalculateSizeInMB test

Signed-off-by: zhouhao <zhouhao@cn.fujitsu.com>
---
 pkg/util/utils_test.go | 19 +++++++++++++++++++
 1 file changed, 19 insertions(+)

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