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()