package entity import ( "testing" "time" "github.com/photoprism/photoprism/pkg/fs" "github.com/stretchr/testify/assert" ) func TestFirstFileByHash(t *testing.T) { t.Run("not existing file", func(t *testing.T) { f, err := FirstFileByHash("xxx") assert.Equal(t, "record not found", err.Error()) assert.Equal(t, uint(0), f.ID) }) t.Run("existing file", func(t *testing.T) { f, err := FirstFileByHash("2cad9168fa6acc5c5c2965ddf6ec465ca42fd818") if err != nil { t.Fatal(err) } assert.Equal(t, uint(0xf4240), f.ID) }) } func TestFile_ShareFileName(t *testing.T) { t.Run("photo with title", func(t *testing.T) { photo := &Photo{TakenAtLocal: time.Date(2019, 01, 15, 0, 0, 0, 0, time.UTC), PhotoTitle: "Berlin / Morning Mood"} file := &File{Photo: photo, FileType: "jpg", FileUID: "foobar345678765", FileHash: "e98eb86480a72bd585d228a709f0622f90e86cbc"} filename := file.ShareFileName() assert.Contains(t, filename, "20190115-000000-Berlin-Morning-Mood") assert.Contains(t, filename, fs.JpegExt) }) t.Run("photo without title", func(t *testing.T) { photo := &Photo{TakenAtLocal: time.Date(2019, 01, 15, 0, 0, 0, 0, time.UTC), PhotoTitle: ""} file := &File{Photo: photo, FileType: "jpg", PhotoUID: "123", FileUID: "foobar345678765", FileHash: "e98eb86480a72bd585d228a709f0622f90e86cbc"} filename := file.ShareFileName() assert.Equal(t, filename, "e98eb86480a72bd585d228a709f0622f90e86cbc.jpg") }) t.Run("photo without photo", func(t *testing.T) { file := &File{Photo: nil, FileType: "jpg", FileUID: "foobar345678765", FileHash: "e98eb86480a72bd585d228a709f0622f90e86cbc"} filename := file.ShareFileName() assert.Equal(t, "e98eb86480a72bd585d228a709f0622f90e86cbc.jpg", filename) }) t.Run("file hash < 8", func(t *testing.T) { photo := &Photo{TakenAtLocal: time.Date(2019, 01, 15, 0, 0, 0, 0, time.UTC), PhotoTitle: "Berlin / Morning Mood"} file := &File{Photo: photo, FileType: "jpg", FileUID: "foobar345678765", FileHash: "e98"} filename := file.ShareFileName() assert.NotContains(t, filename, "20190115-000000-Berlin-Morning-Mood") }) t.Run("no file uid", func(t *testing.T) { file := &File{Photo: nil, FileType: "jpg", FileHash: "e98ijhyt"} filename := file.ShareFileName() assert.Equal(t, filename, "e98ijhyt.jpg") }) } func TestFile_Changed(t *testing.T) { var deletedAt = time.Date(2019, 01, 15, 0, 0, 0, 0, time.UTC) t.Run("different modified times", func(t *testing.T) { file := &File{Photo: nil, FileType: "jpg", FileSize: 500, ModTime: time.Date(2019, 01, 15, 0, 0, 0, 0, time.UTC).Unix()} time := time.Date(2020, 01, 15, 0, 0, 0, 0, time.UTC) assert.Equal(t, true, file.Changed(500, time)) }) t.Run("different sizes", func(t *testing.T) { file := &File{Photo: nil, FileType: "jpg", FileSize: 600, ModTime: time.Date(2019, 01, 15, 0, 0, 0, 0, time.UTC).Unix()} time := time.Date(2019, 01, 15, 0, 0, 0, 0, time.UTC) assert.Equal(t, true, file.Changed(500, time)) }) t.Run("no change", func(t *testing.T) { file := &File{Photo: nil, FileType: "jpg", FileSize: 500, ModTime: time.Date(2019, 01, 15, 0, 0, 0, 0, time.UTC).Unix()} time := time.Date(2019, 01, 15, 0, 0, 0, 0, time.UTC) assert.Equal(t, false, file.Changed(500, time)) }) t.Run("deleted", func(t *testing.T) { file := &File{Photo: nil, FileType: "jpg", FileSize: 500, ModTime: time.Date(2019, 01, 15, 0, 0, 0, 0, time.UTC).Unix(), DeletedAt: &deletedAt} time := time.Date(2019, 01, 15, 0, 0, 0, 0, time.UTC) assert.Equal(t, true, file.Changed(500, time)) }) } func TestFile_Create(t *testing.T) { t.Run("photo id == 0", func(t *testing.T) { file := File{PhotoID: 0} assert.Error(t, file.Create()) }) t.Run("file already exists", func(t *testing.T) { file := &File{PhotoID: 123, FileType: "jpg", FileSize: 500, ModTime: time.Date(2019, 01, 15, 0, 0, 0, 0, time.UTC).Unix()} file.Create() assert.Error(t, file.Create()) }) t.Run("success", func(t *testing.T) { photo := &Photo{TakenAtLocal: time.Date(2019, 01, 15, 0, 0, 0, 0, time.UTC), PhotoTitle: "Berlin / Morning Mood"} file := &File{Photo: photo, FileType: "jpg", FileSize: 500, PhotoID: 766, FileName: "testname", FileRoot: "xyz"} err := file.Create() if err != nil { t.Fatal(err) } }) } func TestFile_Purge(t *testing.T) { t.Run("success", func(t *testing.T) { file := &File{Photo: nil, FileType: "jpg", FileSize: 500} assert.Equal(t, nil, file.Purge()) }) } func TestFile_AllFilesMissing(t *testing.T) { t.Run("true", func(t *testing.T) { photo := &Photo{TakenAtLocal: time.Date(2019, 01, 15, 0, 0, 0, 0, time.UTC), PhotoTitle: ""} file := &File{Photo: photo, FileType: "jpg", PhotoUID: "123", FileUID: "123", FileMissing: true} file2 := &File{Photo: photo, FileType: "jpg", PhotoUID: "123", FileUID: "456", FileMissing: true} assert.True(t, file.AllFilesMissing()) assert.NotEmpty(t, file2) }) //TODO test false /*t.Run("false", func(t *testing.T) { file := FileFixturesExampleJPG assert.False(t, file.AllFilesMissing()) assert.NotEmpty(t, file) })*/ } func TestFile_Save(t *testing.T) { t.Run("save without photo", func(t *testing.T) { file := &File{Photo: nil, FileType: "jpg", PhotoUID: "123", FileUID: "123"} err := file.Save() if err == nil { t.Fatal("error should not be nil") } if file.ID != 0 { t.Fatalf("file id should be 0: %d", file.ID) } assert.Equal(t, "file: photo id must not be empty (save 123)", err.Error()) }) t.Run("success", func(t *testing.T) { photo := &Photo{TakenAtLocal: time.Date(2019, 01, 15, 0, 0, 0, 0, time.UTC), PhotoTitle: "Berlin / Morning Mood"} file := &File{Photo: photo, FileType: "jpg", FileSize: 500, PhotoID: 766, FileName: "Food", FileRoot: "", UpdatedAt: time.Date(2019, 01, 15, 0, 0, 0, 0, time.UTC)} err := file.Save() if err != nil { t.Fatal(err) } }) } func TestFile_UpdateVideoInfos(t *testing.T) { t.Run("success", func(t *testing.T) { file := &File{FileType: "jpg", FileWidth: 600, FileName: "VideoUpdate", PhotoID: 1000003} assert.Equal(t, "bridge.mp4", FileFixturesExampleBridgeVideo.FileName) assert.Equal(t, int(1200), FileFixturesExampleBridgeVideo.FileWidth) err := file.UpdateVideoInfos() if err != nil { t.Fatal(err) } var files Files if err := Db().Where("photo_id = ? AND file_video = 1", file.PhotoID).Find(&files).Error; err != nil { t.Fatal(err) } assert.Len(t, files, 1) for _, f := range files { assert.Equal(t, "bridge.mp4", f.FileName) assert.Equal(t, int(600), f.FileWidth) } }) } func TestFile_Update(t *testing.T) { t.Run("success", func(t *testing.T) { file := &File{FileType: "jpg", FileSize: 500, FileName: "ToBeUpdated", FileRoot: "", PhotoID: 5678} err := file.Save() if err != nil { t.Fatal(err) } assert.Equal(t, "ToBeUpdated", file.FileName) err2 := file.Update("FileName", "Happy") if err2 != nil { t.Fatal(err2) } assert.Equal(t, "Happy", file.FileName) }) } func TestFile_Links(t *testing.T) { t.Run("result", func(t *testing.T) { file := FileFixturesExampleBridge links := file.Links() assert.Equal(t, "5jxf3jfn2k", links[0].LinkToken) }) } func TestFile_NoJPEG(t *testing.T) { t.Run("true", func(t *testing.T) { file := &File{Photo: nil, FileType: "xmp", FileSize: 500} assert.True(t, file.NoJPEG()) }) t.Run("false", func(t *testing.T) { file := &File{Photo: nil, FileType: "jpg", FileSize: 500} assert.False(t, file.NoJPEG()) }) } func TestFile_Panorama(t *testing.T) { t.Run("3000", func(t *testing.T) { file := &File{Photo: nil, FileType: "jpg", FileSidecar: false, FileWidth: 3000, FileHeight: 1000} assert.True(t, file.Panorama()) }) t.Run("1999", func(t *testing.T) { file := &File{Photo: nil, FileType: "jpg", FileSidecar: false, FileWidth: 1999, FileHeight: 1000} assert.False(t, file.Panorama()) }) t.Run("2000", func(t *testing.T) { file := &File{Photo: nil, FileType: "jpg", FileSidecar: false, FileWidth: 2000, FileHeight: 1000} assert.True(t, file.Panorama()) }) t.Run("false", func(t *testing.T) { file := &File{Photo: nil, FileType: "jpg", FileSidecar: false, FileWidth: 1500, FileHeight: 1000} assert.False(t, file.Panorama()) }) t.Run("equirectangular", func(t *testing.T) { file := &File{Photo: nil, FileType: "jpg", FileSidecar: false, FileWidth: 1500, FileHeight: 1000, FileProjection: ProjectionEquirectangular} assert.True(t, file.Panorama()) }) t.Run("sidecar", func(t *testing.T) { file := &File{Photo: nil, FileType: "xmp", FileSidecar: true, FileWidth: 3000, FileHeight: 1000} assert.False(t, file.Panorama()) }) } func TestFile_Delete(t *testing.T) { t.Run("permanently", func(t *testing.T) { file := &File{FileType: "jpg", FileSize: 500, FileName: "ToBePermanentlyDeleted", FileRoot: "", PhotoID: 5678} err := file.Save() if err != nil { t.Fatal(err) } assert.Equal(t, "ToBePermanentlyDeleted", file.FileName) err2 := file.Delete(true) assert.Nil(t, err2) }) t.Run("not permanently", func(t *testing.T) { file := &File{FileType: "jpg", FileSize: 500, FileName: "ToBeDeleted", FileRoot: "", PhotoID: 5678} err := file.Save() if err != nil { t.Fatal(err) } assert.Equal(t, "ToBeDeleted", file.FileName) err2 := file.Delete(false) assert.Nil(t, err2) }) }