From eff06c9fe32504888aa9571c5220da45221319f6 Mon Sep 17 00:00:00 2001
From: Zhou Hao <zhouhao@fujitsu.com>
Date: Tue, 14 Mar 2023 11:55:30 +0800
Subject: [PATCH] test: add UT for byte String function (#2170)

Signed-off-by: Zhou Hao <zhouhao@fujitsu.com>
---
 pkg/unit/byte_test.go | 37 +++++++++++++++++++++++++++++++++++++
 1 file changed, 37 insertions(+)

diff --git a/pkg/unit/byte_test.go b/pkg/unit/byte_test.go
index bf6fc3f48..5169cba0f 100644
--- a/pkg/unit/byte_test.go
+++ b/pkg/unit/byte_test.go
@@ -162,3 +162,40 @@ size: 1Mix
 		assert.Equal(tc.size, data.Size.ToNumber())
 	}
 }
+
+func Test_String(t *testing.T) {
+	assert := testifyassert.New(t)
+	testCases := []struct {
+		data string
+		b    Bytes
+	}{
+		{
+			data: "1.0B",
+			b:    1,
+		},
+		{
+			data: "2.0KB",
+			b:    2 * 1024,
+		},
+		{
+			data: "3.0MB",
+			b:    3 * 1024 * 1024,
+		},
+		{
+			data: "4.0GB",
+			b:    4 * 1024 * 1024 * 1024,
+		},
+		{
+			data: "5.0TB",
+			b:    5 * 1024 * 1024 * 1024 * 1024,
+		},
+		{
+			data: "6.0PB",
+			b:    6 * 1024 * 1024 * 1024 * 1024 * 1024,
+		},
+	}
+
+	for _, tc := range testCases {
+		assert.Equal(tc.b.String(), tc.data)
+	}
+}