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

func TestGetMimeType(t *testing.T) {
	var fileMime = map[string]string{
		"media_for_test/640x480.jpg":   "image/jpeg",
		"media_for_test/3640x3480.jpg": "image/jpeg",
		"media_for_test/video.mp4":     "video/mp4",
	}
	for k, v := range fileMime {
		mime, _ := getFileMime(k)
		msg := fmt.Sprintf("%s should be %s, not %s", k, v, mime)
		AssertEqual(t, mime, v, msg)
	}
}