photoprism/internal/photoprism/related_test.go

package photoprism

import (
	"testing"

	"github.com/stretchr/testify/assert"

	"github.com/photoprism/photoprism/internal/config"
	"github.com/photoprism/photoprism/pkg/fs"
)

func TestRelatedFiles_HasPreview(t *testing.T) {
	cfg := config.TestConfig()

	t.Run("JPEG", func(t *testing.T) {
		mediaFile, err := NewMediaFile(cfg.ExamplesPath() + "/telegram_2020-01-30_09-57-18.jpg")
		if err != nil {
			t.Fatal(err)
		}
		mediaFile2, err2 := NewMediaFile(cfg.ExamplesPath() + "/gopher-video.mp4")
		if err2 != nil {
			t.Fatal(err2)
		}
		mediaFile3, err3 := NewMediaFile(cfg.ExamplesPath() + "/iphone_7.heic")
		if err3 != nil {
			t.Fatal(err3)
		}
		relatedFiles := RelatedFiles{
			Files: MediaFiles{mediaFile, mediaFile2},
			Main:  mediaFile3,
		}
		assert.True(t, relatedFiles.HasPreview())
	})
	t.Run("PNG", func(t *testing.T) {
		mediaFile, err := NewMediaFile(cfg.ExamplesPath() + "/gopher-video.mp4")
		if err != nil {
			t.Fatal(err)
		}
		mediaFile2, err2 := NewMediaFile(cfg.ExamplesPath() + "/Screenshot 2019-05-21 at 10.45.52.png")
		if err2 != nil {
			t.Fatal(err2)
		}
		mediaFile3, err3 := NewMediaFile(cfg.ExamplesPath() + "/iphone_7.heic")
		if err3 != nil {
			t.Fatal(err3)
		}
		relatedFiles := RelatedFiles{
			Files: MediaFiles{mediaFile, mediaFile2},
			Main:  mediaFile3,
		}
		assert.True(t, relatedFiles.HasPreview())
	})
	t.Run("False", func(t *testing.T) {
		mediaFile3, err3 := NewMediaFile(cfg.ExamplesPath() + "/iphone_7.heic")
		if err3 != nil {
			t.Fatal(err3)
		}
		mediaFile2, err2 := NewMediaFile(cfg.ExamplesPath() + "/gopher-video.mp4")
		if err2 != nil {
			t.Fatal(err2)
		}
		relatedFiles := RelatedFiles{
			Files: MediaFiles{mediaFile3, mediaFile2},
			Main:  nil,
		}
		assert.False(t, relatedFiles.HasPreview())
	})
}

func TestRelatedFiles_String(t *testing.T) {
	cfg := config.TestConfig()

	t.Run("true", func(t *testing.T) {
		mediaFile, err := NewMediaFile(cfg.ExamplesPath() + "/telegram_2020-01-30_09-57-18.jpg")
		if err != nil {
			t.Fatal(err)
		}
		mediaFile2, err2 := NewMediaFile(cfg.ExamplesPath() + "/Screenshot 2019-05-21 at 10.45.52.png")
		if err2 != nil {
			t.Fatal(err2)
		}
		mediaFile3, err3 := NewMediaFile(cfg.ExamplesPath() + "/iphone_7.heic")
		if err3 != nil {
			t.Fatal(err3)
		}
		relatedFiles := RelatedFiles{
			Files: MediaFiles{mediaFile, mediaFile2},
			Main:  mediaFile3,
		}
		assert.Equal(t, "telegram_2020-01-30_09-57-18.jpg, Screenshot 2019-05-21 at 10.45.52.png", relatedFiles.String())
	})
}

func TestRelatedFiles_Len(t *testing.T) {
	cfg := config.TestConfig()

	t.Run("true", func(t *testing.T) {
		mediaFile, err := NewMediaFile(cfg.ExamplesPath() + "/telegram_2020-01-30_09-57-18.jpg")
		if err != nil {
			t.Fatal(err)
		}
		mediaFile2, err2 := NewMediaFile(cfg.ExamplesPath() + "/Screenshot 2019-05-21 at 10.45.52.png")
		if err2 != nil {
			t.Fatal(err2)
		}
		mediaFile3, err3 := NewMediaFile(cfg.ExamplesPath() + "/iphone_7.heic")
		if err3 != nil {
			t.Fatal(err3)
		}
		relatedFiles := RelatedFiles{
			Files: MediaFiles{mediaFile, mediaFile2},
			Main:  mediaFile3,
		}
		assert.Equal(t, 2, relatedFiles.Len())
	})
}

func TestRelatedFiles_Count(t *testing.T) {
	cfg := config.TestConfig()
	t.Run("NoMainFile", func(t *testing.T) {
		relatedFiles := RelatedFiles{
			Files: MediaFiles{},
			Main:  nil,
		}
		assert.Equal(t, 0, relatedFiles.Count())
	})
	t.Run("None", func(t *testing.T) {
		mediaFile, err := NewMediaFile(cfg.ExamplesPath() + "/telegram_2020-01-30_09-57-18.jpg")
		if err != nil {
			t.Fatal(err)
		}
		relatedFiles := RelatedFiles{
			Files: MediaFiles{},
			Main:  mediaFile,
		}
		assert.Equal(t, 0, relatedFiles.Count())
	})
	t.Run("One", func(t *testing.T) {
		mediaFile, err := NewMediaFile(cfg.ExamplesPath() + "/telegram_2020-01-30_09-57-18.jpg")
		if err != nil {
			t.Fatal(err)
		}
		mediaFile2, err2 := NewMediaFile(cfg.ExamplesPath() + "/Screenshot 2019-05-21 at 10.45.52.png")
		if err2 != nil {
			t.Fatal(err2)
		}
		mediaFile3, err3 := NewMediaFile(cfg.ExamplesPath() + "/iphone_7.heic")
		if err3 != nil {
			t.Fatal(err3)
		}
		relatedFiles := RelatedFiles{
			Files: MediaFiles{mediaFile, mediaFile2},
			Main:  mediaFile3,
		}
		assert.Equal(t, 1, relatedFiles.Count())
	})
}

func TestRelatedFiles_MainFileType(t *testing.T) {
	cfg := config.TestConfig()
	t.Run("None", func(t *testing.T) {
		relatedFiles := RelatedFiles{
			Files: MediaFiles{},
			Main:  nil,
		}
		assert.Equal(t, "", relatedFiles.MainFileType())
	})
	t.Run("Primary", func(t *testing.T) {
		mediaFile, err := NewMediaFile(cfg.ExamplesPath() + "/telegram_2020-01-30_09-57-18.jpg")
		if err != nil {
			t.Fatal(err)
		}
		relatedFiles := RelatedFiles{
			Files: MediaFiles{},
			Main:  mediaFile,
		}
		assert.Equal(t, string(fs.ImageJPEG), relatedFiles.MainFileType())
	})
	t.Run("Heif", func(t *testing.T) {
		mediaFile, err := NewMediaFile(cfg.ExamplesPath() + "/telegram_2020-01-30_09-57-18.jpg")
		if err != nil {
			t.Fatal(err)
		}
		mediaFile2, err2 := NewMediaFile(cfg.ExamplesPath() + "/Screenshot 2019-05-21 at 10.45.52.png")
		if err2 != nil {
			t.Fatal(err2)
		}
		mediaFile3, err3 := NewMediaFile(cfg.ExamplesPath() + "/iphone_7.heic")
		if err3 != nil {
			t.Fatal(err3)
		}
		relatedFiles := RelatedFiles{
			Files: MediaFiles{mediaFile, mediaFile2},
			Main:  mediaFile3,
		}
		assert.Equal(t, string(fs.ImageHEIC), relatedFiles.MainFileType())
	})
}

func TestRelatedFiles_MainLogName(t *testing.T) {
	cfg := config.TestConfig()
	t.Run("None", func(t *testing.T) {
		relatedFiles := RelatedFiles{
			Files: MediaFiles{},
			Main:  nil,
		}
		assert.Equal(t, "", relatedFiles.MainFileType())
	})
	t.Run("Telegram", func(t *testing.T) {
		mediaFile, err := NewMediaFile(cfg.ExamplesPath() + "/telegram_2020-01-30_09-57-18.jpg")
		if err != nil {
			t.Fatal(err)
		}
		relatedFiles := RelatedFiles{
			Files: MediaFiles{},
			Main:  mediaFile,
		}
		assert.Equal(t, "telegram_2020-01-30_09-57-18.jpg", relatedFiles.MainLogName())
	})
	t.Run("iPhone7", func(t *testing.T) {
		mediaFile, err := NewMediaFile(cfg.ExamplesPath() + "/telegram_2020-01-30_09-57-18.jpg")
		if err != nil {
			t.Fatal(err)
		}
		mediaFile2, err2 := NewMediaFile(cfg.ExamplesPath() + "/Screenshot 2019-05-21 at 10.45.52.png")
		if err2 != nil {
			t.Fatal(err2)
		}
		mediaFile3, err3 := NewMediaFile(cfg.ExamplesPath() + "/iphone_7.heic")
		if err3 != nil {
			t.Fatal(err3)
		}
		relatedFiles := RelatedFiles{
			Files: MediaFiles{mediaFile, mediaFile2, mediaFile3},
			Main:  mediaFile3,
		}
		assert.Equal(t, "iphone_7.heic", relatedFiles.MainLogName())
	})
}