photoprism/mediafile_test.go

package photoprism

import (
	"testing"
	"github.com/stretchr/testify/assert"
)

func TestMediaFile_FindRelatedImages(t *testing.T) {
	conf := NewTestConfig()

	conf.InitializeTestData(t)

	mediaFile := NewMediaFile(conf.ImportPath + "/raw/20140717_154212_1EC48F8489.cr2")

	expectedBaseFilename := conf.ImportPath + "/raw/20140717_154212_1EC48F8489"

	related, _, err := mediaFile.GetRelatedFiles()

	assert.Empty(t, err)

	assert.Len(t, related, 3)

	for _, result := range related {
		filename := result.GetFilename()

		extension := result.GetExtension()

		baseFilename := filename[0:len(filename)-len(extension)]

		assert.Equal(t, expectedBaseFilename, baseFilename)
	}
}

func TestMediaFile_GetPerceptiveHash(t *testing.T) {
	conf := NewTestConfig()

	conf.InitializeTestData(t)

	mediaFile1 := NewMediaFile(conf.ImportPath + "/20130203_193332_0AE340D280.jpg")

	hash1, _ := mediaFile1.GetPerceptualHash()

	assert.Equal(t, "66debc383325d3bd", hash1)

	mediaFile2 := NewMediaFile(conf.ImportPath + "/20130203_193332_0AE340D280_V2.jpg")

	hash2, _ := mediaFile2.GetPerceptualHash()

	assert.Equal(t, "e6debc393325c3b9", hash2)

	distance, _ := mediaFile1.GetPerceptualDistance(hash2)

	assert.Equal(t, 4, distance)

	mediaFile3 := NewMediaFile(conf.ImportPath + "/iphone/IMG_6788.JPG")
	hash3, _ := mediaFile3.GetPerceptualHash()

	assert.Equal(t, "f1e2858b171d3e78", hash3)

	distance, _ = mediaFile1.GetPerceptualDistance(hash3)

	assert.Equal(t, 33, distance)
}


func TestMediaFile_GetMimeType(t *testing.T) {
	conf := NewTestConfig()

	conf.InitializeTestData(t)

	image1 := NewMediaFile(conf.ImportPath + "/iphone/IMG_6788.JPG")

	assert.Equal(t, "image/jpeg", image1.GetMimeType())

	image2 := NewMediaFile(conf.ImportPath + "/raw/20140717_154212_1EC48F8489.cr2")

	assert.Equal(t, "application/octet-stream", image2.GetMimeType())
}

func TestMediaFile_Exists(t *testing.T) {
	conf := NewTestConfig()

	mediaFile := NewMediaFile(conf.ImportPath + "/iphone/IMG_6788.JPG")

	assert.True(t, mediaFile.Exists())

	mediaFile = NewMediaFile(conf.ImportPath + "/iphone/IMG_6788_XYZ.JPG")

	assert.False(t, mediaFile.Exists())
}