90 lines
2.1 KiB
Go
90 lines
2.1 KiB
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())
|
|
} |