From 96fadd949a38a52362d79f79b3d4e85b08f98a1a Mon Sep 17 00:00:00 2001
From: "Seungbae.yu" <72970043+dbadoy@users.noreply.github.com>
Date: Wed, 29 Jun 2022 19:13:00 +0900
Subject: [PATCH] common: increase StorageSize test coverage (#25188)

---
 common/size_test.go | 21 +++++++++++++++++++++
 1 file changed, 21 insertions(+)

diff --git a/common/size_test.go b/common/size_test.go
index 0938d483c4bb0..28f053d39f5d5 100644
--- a/common/size_test.go
+++ b/common/size_test.go
@@ -25,6 +25,8 @@ func TestStorageSizeString(t *testing.T) {
 		size StorageSize
 		str  string
 	}{
+		{2839274474874, "2.58 TiB"},
+		{2458492810, "2.29 GiB"},
 		{2381273, "2.27 MiB"},
 		{2192, "2.14 KiB"},
 		{12, "12.00 B"},
@@ -36,3 +38,22 @@ func TestStorageSizeString(t *testing.T) {
 		}
 	}
 }
+
+func TestStorageSizeTerminalString(t *testing.T) {
+	tests := []struct {
+		size StorageSize
+		str  string
+	}{
+		{2839274474874, "2.58TiB"},
+		{2458492810, "2.29GiB"},
+		{2381273, "2.27MiB"},
+		{2192, "2.14KiB"},
+		{12, "12.00B"},
+	}
+
+	for _, test := range tests {
+		if test.size.TerminalString() != test.str {
+			t.Errorf("%f: got %q, want %q", float64(test.size), test.size.TerminalString(), test.str)
+		}
+	}
+}