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