From d874a9bf6b57f3d211ab1e0a07a47b7cad18a62f Mon Sep 17 00:00:00 2001
From: Matthias Loibl <mail@matthiasloibl.com>
Date: Mon, 7 Nov 2016 00:32:32 +0100
Subject: [PATCH] Add unit test for base.FileSize()
---
modules/base/tool_test.go | 25 +++++++++++++++++++++++--
1 file changed, 23 insertions(+), 2 deletions(-)
diff --git a/modules/base/tool_test.go b/modules/base/tool_test.go
index c8ed85fdeb..e92786b35b 100644
--- a/modules/base/tool_test.go
+++ b/modules/base/tool_test.go
@@ -1,6 +1,9 @@
package base
-import "testing"
+import (
+ "github.com/stretchr/testify/assert"
+ "testing"
+)
func TestEncodeMD5(t *testing.T) {
if checksum := EncodeMD5("foobar"); checksum != "3858f62230ac3c915f300c664312c63f" {
@@ -70,7 +73,25 @@ func TestHashEmail(t *testing.T) {
// TODO: TimeSince()
// TODO: logn()
// TODO: humanateBytes()
-// TODO: FileSize()
+
+func TestFileSize(t *testing.T) {
+ var size int64
+ size = 512
+ assert.Equal(t, "512B", FileSize(size))
+ size = size * 1024
+ assert.Equal(t, "512KB", FileSize(size))
+ size = size * 1024
+ assert.Equal(t, "512MB", FileSize(size))
+ size = size * 1024
+ assert.Equal(t, "512GB", FileSize(size))
+ size = size * 1024
+ assert.Equal(t, "512TB", FileSize(size))
+ size = size * 1024
+ assert.Equal(t, "512PB", FileSize(size))
+ //size = size * 1024 TODO: Fix bug for EB
+ //assert.Equal(t, "512EB", FileSize(size))
+}
+
// TODO: Subtract()
// TODO: EllipsisString()
// TODO: TruncateString()