306 lines
9.3 KiB
Go
306 lines
9.3 KiB
Go
package entity
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/photoprism/photoprism/internal/classify"
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestPhoto_HasTitle(t *testing.T) {
|
|
t.Run("false", func(t *testing.T) {
|
|
m := PhotoFixtures.Get("Photo03")
|
|
assert.False(t, m.HasTitle())
|
|
})
|
|
t.Run("true", func(t *testing.T) {
|
|
m := PhotoFixtures.Get("Photo04")
|
|
assert.True(t, m.HasTitle())
|
|
})
|
|
}
|
|
|
|
func TestPhoto_NoTitle(t *testing.T) {
|
|
t.Run("true", func(t *testing.T) {
|
|
m := PhotoFixtures.Get("Photo03")
|
|
assert.True(t, m.NoTitle())
|
|
})
|
|
t.Run("false", func(t *testing.T) {
|
|
m := PhotoFixtures.Get("Photo04")
|
|
assert.False(t, m.NoTitle())
|
|
})
|
|
}
|
|
|
|
func TestPhoto_SetTitle(t *testing.T) {
|
|
t.Run("empty title", func(t *testing.T) {
|
|
m := PhotoFixtures.Get("Photo15")
|
|
assert.Equal(t, "TitleToBeSet", m.PhotoTitle)
|
|
m.SetTitle("", SrcManual)
|
|
assert.Equal(t, "TitleToBeSet", m.PhotoTitle)
|
|
})
|
|
t.Run("title not from the same source", func(t *testing.T) {
|
|
m := PhotoFixtures.Get("Photo15")
|
|
assert.Equal(t, "TitleToBeSet", m.PhotoTitle)
|
|
m.SetTitle("NewTitleSet", SrcAuto)
|
|
assert.Equal(t, "TitleToBeSet", m.PhotoTitle)
|
|
})
|
|
t.Run("success", func(t *testing.T) {
|
|
m := PhotoFixtures.Get("Photo15")
|
|
assert.Equal(t, "TitleToBeSet", m.PhotoTitle)
|
|
m.SetTitle("NewTitleSet", SrcName)
|
|
assert.Equal(t, "NewTitleSet", m.PhotoTitle)
|
|
})
|
|
}
|
|
|
|
func TestPhoto_UpdateTitle(t *testing.T) {
|
|
t.Run("wont update title was modified", func(t *testing.T) {
|
|
m := PhotoFixtures.Get("Photo08")
|
|
classifyLabels := &classify.Labels{}
|
|
assert.Equal(t, "Black beach", m.PhotoTitle)
|
|
err := m.UpdateTitle(*classifyLabels)
|
|
if err == nil {
|
|
t.Fatal()
|
|
}
|
|
assert.Equal(t, "Black beach", m.PhotoTitle)
|
|
})
|
|
t.Run("photo with location without city and label", func(t *testing.T) {
|
|
m := PhotoFixtures.Get("Photo10")
|
|
classifyLabels := &classify.Labels{{Name: "tree", Uncertainty: 30, Source: "manual", Priority: 5, Categories: []string{"plant"}}}
|
|
assert.Equal(t, "Title", m.PhotoTitle)
|
|
err := m.UpdateTitle(*classifyLabels)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
// TODO: Unstable
|
|
if len(m.SubjectNames()) > 0 {
|
|
assert.Equal(t, "Actor A / Germany / 2016", m.PhotoTitle)
|
|
} else {
|
|
assert.Equal(t, "Tree / Germany / 2016", m.PhotoTitle)
|
|
}
|
|
})
|
|
t.Run("photo with location and short city and label", func(t *testing.T) {
|
|
m := PhotoFixtures.Get("Photo09")
|
|
classifyLabels := &classify.Labels{{Name: "tree", Uncertainty: 30, Source: "manual", Priority: 5, Categories: []string{"plant"}}}
|
|
assert.Equal(t, "Title", m.PhotoTitle)
|
|
err := m.UpdateTitle(*classifyLabels)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
assert.Equal(t, "Tree / Teotihuacán / 2016", m.PhotoTitle)
|
|
})
|
|
t.Run("photo with location and locname >45", func(t *testing.T) {
|
|
m := PhotoFixtures.Get("Photo13")
|
|
classifyLabels := &classify.Labels{}
|
|
assert.Equal(t, "Title", m.PhotoTitle)
|
|
err := m.UpdateTitle(*classifyLabels)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
assert.Equal(t, "LonglonglonglonglonglonglonglonglonglonglonglonglongName", m.PhotoTitle)
|
|
})
|
|
t.Run("photo with location and locname >20", func(t *testing.T) {
|
|
m := PhotoFixtures.Get("Photo14")
|
|
classifyLabels := &classify.Labels{}
|
|
assert.Equal(t, "Title", m.PhotoTitle)
|
|
err := m.UpdateTitle(*classifyLabels)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
assert.Equal(t, "longlonglonglonglonglongName / 2018", m.PhotoTitle)
|
|
})
|
|
|
|
t.Run("photo with location and short city", func(t *testing.T) {
|
|
m := PhotoFixtures.Get("Photo09")
|
|
classifyLabels := &classify.Labels{}
|
|
assert.Equal(t, "Title", m.PhotoTitle)
|
|
err := m.UpdateTitle(*classifyLabels)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
assert.Equal(t, "Adosada Platform / Teotihuacán / 2016", m.PhotoTitle)
|
|
})
|
|
t.Run("photo with location without city", func(t *testing.T) {
|
|
m := PhotoFixtures.Get("Photo10")
|
|
classifyLabels := &classify.Labels{}
|
|
assert.Equal(t, "Title", m.PhotoTitle)
|
|
err := m.UpdateTitle(*classifyLabels)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
// TODO: Unstable
|
|
if len(m.SubjectNames()) > 0 {
|
|
assert.Equal(t, "Actor A / Germany / 2016", m.PhotoTitle)
|
|
} else {
|
|
assert.Equal(t, "Holiday Park / Germany / 2016", m.PhotoTitle)
|
|
}
|
|
})
|
|
|
|
t.Run("photo with location without loc name and long city", func(t *testing.T) {
|
|
m := PhotoFixtures.Get("Photo11")
|
|
classifyLabels := &classify.Labels{}
|
|
assert.Equal(t, "Title", m.PhotoTitle)
|
|
err := m.UpdateTitle(*classifyLabels)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
assert.Equal(t, "longlonglonglonglongcity / 2016", m.PhotoTitle)
|
|
})
|
|
t.Run("photo with location without loc name and short city", func(t *testing.T) {
|
|
m := PhotoFixtures.Get("Photo12")
|
|
classifyLabels := &classify.Labels{}
|
|
assert.Equal(t, "Title", m.PhotoTitle)
|
|
err := m.UpdateTitle(*classifyLabels)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
assert.Equal(t, "shortcity / Germany / 2016", m.PhotoTitle)
|
|
})
|
|
t.Run("no location original name", func(t *testing.T) {
|
|
m := PhotoFixtures.Get("19800101_000002_D640C559")
|
|
classifyLabels := &classify.Labels{{Name: "classify", Uncertainty: 30, Source: SrcManual, Priority: 5, Categories: []string{"flower", "plant"}}}
|
|
assert.Equal(t, "Lake / 2790", m.PhotoTitle)
|
|
err := m.UpdateTitle(*classifyLabels)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
assert.Equal(t, "Franzilein & Actress / 2008", m.PhotoTitle)
|
|
})
|
|
t.Run("no location", func(t *testing.T) {
|
|
m := PhotoFixtures.Get("Photo01")
|
|
classifyLabels := &classify.Labels{{Name: "classify", Uncertainty: 30, Source: SrcManual, Priority: 5, Categories: []string{"flower", "plant"}}}
|
|
assert.Equal(t, "", m.PhotoTitle)
|
|
err := m.UpdateTitle(*classifyLabels)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
assert.Equal(t, "Classify / Germany / 2006", m.PhotoTitle)
|
|
})
|
|
|
|
t.Run("no location no labels", func(t *testing.T) {
|
|
m := PhotoFixtures.Get("Photo02")
|
|
classifyLabels := &classify.Labels{}
|
|
assert.Equal(t, "", m.PhotoTitle)
|
|
err := m.UpdateTitle(*classifyLabels)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
// TODO: Unstable
|
|
if len(m.SubjectNames()) > 0 {
|
|
assert.Equal(t, "Actress A / 1990", m.PhotoTitle)
|
|
} else {
|
|
assert.Equal(t, "Bridge / 1990", m.PhotoTitle)
|
|
}
|
|
})
|
|
t.Run("no location no labels no takenAt", func(t *testing.T) {
|
|
m := PhotoFixtures.Get("Photo20")
|
|
classifyLabels := &classify.Labels{}
|
|
assert.Equal(t, "", m.PhotoTitle)
|
|
err := m.UpdateTitle(*classifyLabels)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
assert.Equal(t, "Unknown", m.PhotoTitle)
|
|
})
|
|
t.Run("OnePerson", func(t *testing.T) {
|
|
m := PhotoFixtures.Get("Photo10")
|
|
|
|
assert.Equal(t, SrcAuto, m.TitleSrc)
|
|
assert.Equal(t, SrcAuto, m.DescriptionSrc)
|
|
assert.Equal(t, "Title", m.PhotoTitle)
|
|
assert.Equal(t, "", m.PhotoDescription)
|
|
|
|
err := m.UpdateTitle(classify.Labels{})
|
|
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
assert.Equal(t, SrcAuto, m.TitleSrc)
|
|
assert.Equal(t, SrcAuto, m.DescriptionSrc)
|
|
|
|
// TODO: Unstable
|
|
if len(m.SubjectNames()) > 0 {
|
|
assert.Equal(t, "Actor A / Germany / 2016", m.PhotoTitle)
|
|
} else {
|
|
assert.Equal(t, "Holiday Park / Germany / 2016", m.PhotoTitle)
|
|
}
|
|
|
|
assert.Equal(t, "", m.PhotoDescription)
|
|
})
|
|
t.Run("People", func(t *testing.T) {
|
|
m := PhotoFixtures.Get("Photo04")
|
|
|
|
assert.Equal(t, SrcAuto, m.TitleSrc)
|
|
assert.Equal(t, SrcAuto, m.DescriptionSrc)
|
|
assert.Equal(t, "Neckarbrücke", m.PhotoTitle)
|
|
assert.Equal(t, "", m.PhotoDescription)
|
|
|
|
err := m.UpdateTitle(classify.Labels{})
|
|
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
assert.Equal(t, SrcAuto, m.TitleSrc)
|
|
assert.Equal(t, SrcAuto, m.DescriptionSrc)
|
|
assert.Equal(t, "Corn & Jens / Germany / 2014", m.PhotoTitle)
|
|
assert.Equal(t, "", m.PhotoDescription)
|
|
})
|
|
}
|
|
|
|
func TestPhoto_FileTitle(t *testing.T) {
|
|
t.Run("non-latin", func(t *testing.T) {
|
|
photo := Photo{PhotoName: "桥", PhotoPath: "", OriginalName: ""}
|
|
result := photo.FileTitle()
|
|
assert.Equal(t, "桥", result)
|
|
})
|
|
t.Run("changing-of-the-guard--buckingham-palace_7925318070_o.jpg", func(t *testing.T) {
|
|
photo := Photo{PhotoName: "20200102_194030_9EFA9E5E", PhotoPath: "2000/05", OriginalName: "flickr import/changing-of-the-guard--buckingham-palace_7925318070_o.jpg"}
|
|
result := photo.FileTitle()
|
|
assert.Equal(t, "Changing of the Guard / Buckingham Palace", result)
|
|
})
|
|
t.Run("empty title", func(t *testing.T) {
|
|
photo := Photo{PhotoName: "", PhotoPath: "", OriginalName: ""}
|
|
result := photo.FileTitle()
|
|
assert.Equal(t, "", result)
|
|
})
|
|
t.Run("return title", func(t *testing.T) {
|
|
photo := Photo{PhotoName: "sun, beach, fun", PhotoPath: "", OriginalName: "", PhotoTitle: ""}
|
|
result := photo.FileTitle()
|
|
assert.Equal(t, "Sun, Beach, Fun", result)
|
|
})
|
|
t.Run("return title", func(t *testing.T) {
|
|
photo := Photo{PhotoName: "", PhotoPath: "vacation", OriginalName: "20200102_194030_9EFA9E5E", PhotoTitle: ""}
|
|
result := photo.FileTitle()
|
|
assert.Equal(t, "Vacation", result)
|
|
})
|
|
}
|
|
|
|
func TestPhoto_SubjectNames(t *testing.T) {
|
|
t.Run("Photo09", func(t *testing.T) {
|
|
m := PhotoFixtures.Get("Photo09")
|
|
|
|
if names := m.SubjectNames(); len(names) > 0 {
|
|
t.Errorf("no name expected: %#v", names)
|
|
}
|
|
})
|
|
t.Run("Photo10", func(t *testing.T) {
|
|
m := PhotoFixtures.Get("Photo10")
|
|
|
|
if names := m.SubjectNames(); len(names) == 1 {
|
|
assert.Equal(t, "Actor A", names[0])
|
|
} else {
|
|
t.Logf("unstable subject list: %#v", names)
|
|
}
|
|
})
|
|
t.Run("Photo04", func(t *testing.T) {
|
|
m := PhotoFixtures.Get("Photo04")
|
|
|
|
if names := m.SubjectNames(); len(names) != 2 {
|
|
t.Errorf("two names expected: %#v", names)
|
|
} else {
|
|
assert.Equal(t, []string{"Corn McCornface", "Jens Mander"}, names)
|
|
}
|
|
})
|
|
}
|