| @@ -18,3 +18,7 @@ func getFileSize(s string) (size int64, err error) { | |||
| } | |||
| return | |||
| } | |||
| func getFileMime(s string) (mime string) { | |||
| return "" | |||
| } | |||
| @@ -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) | |||
| } | |||
| } | |||