photoprism/internal/entity/photo_album_test.go

package entity

import (
	"testing"

	"github.com/stretchr/testify/assert"
)

func TestNewPhotoAlbum(t *testing.T) {
	t.Run("new album", func(t *testing.T) {
		m := NewPhotoAlbum("ABC", "EFG")
		assert.Equal(t, "ABC", m.PhotoUID)
		assert.Equal(t, "EFG", m.AlbumUID)
	})
}

func TestPhotoAlbum_TableName(t *testing.T) {
	photoAlbum := &PhotoAlbum{}
	tableName := photoAlbum.TableName()

	assert.Equal(t, "photos_albums", tableName)
}

func TestFirstOrCreatePhotoAlbum(t *testing.T) {
	model := PhotoAlbumFixtures.Get("1", "pt9jtdre2lvl0yh7", "at9lxuqxpogaaba8")
	result := FirstOrCreatePhotoAlbum(&model)

	if result == nil {
		t.Fatal("result should not be nil")
	}

	if result.AlbumUID != model.AlbumUID {
		t.Errorf("AlbumUID should be the same: %s %s", result.AlbumUID, model.AlbumUID)
	}

	if result.PhotoUID != model.PhotoUID {
		t.Errorf("PhotoUID should be the same: %s %s", result.PhotoUID, model.PhotoUID)
	}
}