Add unit tests for scanner media_type

api/scanner/media_type/media_type_test.go

@@ -0,0 +1,42 @@
+package media_type_test
+
+import (
+	"os"
+	"testing"
+
+	"github.com/photoview/photoview/api/scanner/media_type"
+	"github.com/photoview/photoview/api/test_utils"
+	"github.com/stretchr/testify/assert"
+)
+
+func TestMain(m *testing.M) {
+	os.Exit(test_utils.UnitTestRun(m))
+}
+
+func TestMediaTypeIsFunctions(t *testing.T) {
+	rawType := media_type.TypeCR2
+	pngType := media_type.TypePng
+	mp4Type := media_type.TypeMP4
+
+	assert.True(t, rawType.IsRaw())
+	assert.False(t, pngType.IsRaw())
+
+	assert.True(t, pngType.IsWebCompatible())
+	assert.False(t, rawType.IsWebCompatible())
+
+	assert.True(t, mp4Type.IsVideo())
+	assert.False(t, pngType.IsVideo())
+
+	assert.True(t, pngType.IsBasicTypeSupported())
+}
+
+func TestMediaTypeFromExtension(t *testing.T) {
+	pngType, found := media_type.GetExtensionMediaType(".PNG")
+
+	assert.True(t, found)
+	assert.Equal(t, media_type.TypePng, pngType)
+}
+
+func TestMediaTypeGetExtensions(t *testing.T) {
+	assert.ElementsMatch(t, []string{".jpg", ".JPG", ".jpeg", ".JPEG"}, media_type.TypeJpeg.FileExtensions())
+}