photoprism/internal/entity/file_share_test.go

package entity

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

func TestFileShare_TableName(t *testing.T) {
	fileShare := &FileShare{}
	assert.Equal(t, "files_share", fileShare.TableName())
}

func TestNewFileShare(t *testing.T) {
	r := NewFileShare(123, 123, "test")
	assert.IsType(t, &FileShare{}, r)
	assert.Equal(t, uint(0x7b), r.FileID)
	assert.Equal(t, uint(0x7b), r.AccountID)
	assert.Equal(t, "test", r.RemoteName)
	assert.Equal(t, "new", r.Status)
}

func TestFirstOrCreateFileShare(t *testing.T) {
	fileShare := &FileShare{FileID: 123, AccountID: 888, RemoteName: "test888"}
	result := FirstOrCreateFileShare(fileShare)

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

	if result.FileID != fileShare.FileID {
		t.Errorf("FileID should be the same: %d %d", result.FileID, fileShare.FileID)
	}

	if result.AccountID != fileShare.AccountID {
		t.Errorf("AccountID should be the same: %d %d", result.AccountID, fileShare.AccountID)
	}
}