From 3af76d129675c0d67334648e661243649bf905e6 Mon Sep 17 00:00:00 2001
From: Patrick Peng Sun <patrick@lawipac.com>
Date: Wed, 10 May 2017 16:38:15 +1000
Subject: [PATCH] file size test case passed using map loop

---
 fileinfo.go      |  4 ++++
 fileinfo_test.go | 20 ++++++++++++++++++++
 2 files changed, 24 insertions(+)
 create mode 100644 fileinfo_test.go

diff --git a/fileinfo.go b/fileinfo.go
index 31fc920..c520147 100644
--- a/fileinfo.go
+++ b/fileinfo.go
@@ -18,3 +18,7 @@ func getFileSize(s string) (size int64, err error) {
 	}
 	return
 }
+
+func getFileMime(s string) (mime string) {
+	return ""
+}
diff --git a/fileinfo_test.go b/fileinfo_test.go
new file mode 100644
index 0000000..81f2985
--- /dev/null
+++ b/fileinfo_test.go
@@ -0,0 +1,20 @@
+package main
+
+import (
+	"fmt"
+	"testing"
+)
+
+func TestGetFileSize(t *testing.T) {
+
+	var fileSize = map[string]int{
+		"media_for_test/640x480.jpg":   50668,
+		"media_for_test/3640x3480.jpg": 785190,
+		"media_for_test/video.mp4":     75520,
+	}
+	for k, v := range fileSize {
+		size, _ := getFileSize(k)
+		msg := fmt.Sprintf("%s should have size %d, not %d", k, v, size)
+		AssertEqual(t, size, int64(v), msg)
+	}
+}