diff --git a/api/scanner/media_type/media_type_test.go b/api/scanner/media_type/media_type_test.go new file mode 100644 index 00000000..4f723053 --- /dev/null +++ b/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()) +}