photoprism/internal/entity/subject_test.go

281 lines
6.3 KiB
Go

package entity
import (
"testing"
"time"
"github.com/stretchr/testify/assert"
)
func TestSubject_TableName(t *testing.T) {
m := &Subject{}
assert.Contains(t, m.TableName(), "subjects")
}
func TestNewSubject(t *testing.T) {
t.Run("Jens_Mander", func(t *testing.T) {
m := NewSubject("Jens Mander", SubjPerson, SrcAuto)
assert.Equal(t, "Jens Mander", m.SubjName)
assert.Equal(t, "jens-mander", m.SubjSlug)
assert.Equal(t, "person", m.SubjType)
})
t.Run("subject Type empty", func(t *testing.T) {
m := NewSubject("Anna Mander", "", SrcAuto)
assert.Equal(t, "Anna Mander", m.SubjName)
assert.Equal(t, "anna-mander", m.SubjSlug)
assert.Equal(t, "person", m.SubjType)
})
t.Run("subject name empty", func(t *testing.T) {
m := NewSubject("", "", SrcAuto)
assert.Nil(t, m)
})
}
func TestSubject_SetName(t *testing.T) {
t.Run("success", func(t *testing.T) {
m := NewSubject("Jens Mander", SubjPerson, SrcAuto)
assert.Equal(t, "Jens Mander", m.SubjName)
assert.Equal(t, "jens-mander", m.SubjSlug)
if err := m.SetName("Foo McBar"); err != nil {
t.Fatal(err)
}
assert.Equal(t, "Foo McBar", m.SubjName)
assert.Equal(t, "foo-mcbar", m.SubjSlug)
})
t.Run("new name empty", func(t *testing.T) {
m := NewSubject("Jens Mander", SubjPerson, SrcAuto)
assert.Equal(t, "Jens Mander", m.SubjName)
assert.Equal(t, "jens-mander", m.SubjSlug)
err := m.SetName("")
if err == nil {
t.Fatal(err)
}
assert.Equal(t, "subject: name must not be empty", err.Error())
assert.Equal(t, "Jens Mander", m.SubjName)
})
}
func TestFirstOrCreatePerson(t *testing.T) {
t.Run("not yet existing person", func(t *testing.T) {
m := NewSubject("Create Me", SubjPerson, SrcAuto)
result := FirstOrCreateSubject(m)
if result == nil {
t.Fatal("result should not be nil")
}
assert.Equal(t, "Create Me", m.SubjName)
assert.Equal(t, "create-me", m.SubjSlug)
})
t.Run("existing person", func(t *testing.T) {
m := SubjectFixtures.Pointer("john-doe")
result := FirstOrCreateSubject(m)
if result == nil {
t.Fatal("result should not be nil")
}
assert.Equal(t, "John Doe", m.SubjName)
assert.Equal(t, "john-doe", m.SubjSlug)
assert.Equal(t, "Short Note", m.SubjNotes)
})
}
func TestSubject_Save(t *testing.T) {
t.Run("success", func(t *testing.T) {
m := NewSubject("Save Me", SubjPerson, SrcAuto)
initialDate := m.UpdatedAt
err := m.Save()
if err != nil {
t.Fatal(err)
}
afterDate := m.UpdatedAt
assert.True(t, afterDate.After(initialDate))
})
}
func TestSubject_Delete(t *testing.T) {
t.Run("success", func(t *testing.T) {
m := NewSubject("Jens Mander", SubjPerson, SrcAuto)
err := m.Save()
assert.False(t, m.Deleted())
var subj Subjects
if err := Db().Where("subj_name = ?", m.SubjName).Find(&subj).Error; err != nil {
t.Fatal(err)
}
assert.Len(t, subj, 1)
err = m.Delete()
if err != nil {
t.Fatal(err)
}
if err := Db().Where("subj_name = ?", m.SubjName).Find(&subj).Error; err != nil {
t.Fatal(err)
}
assert.Len(t, subj, 0)
})
}
func TestSubject_Restore(t *testing.T) {
t.Run("success", func(t *testing.T) {
var deleteTime = time.Date(2020, 1, 1, 0, 0, 0, 0, time.UTC)
m := &Subject{DeletedAt: &deleteTime, SubjName: "ToBeRestored"}
err := m.Save()
if err != nil {
t.Fatal(err)
}
assert.True(t, m.Deleted())
err = m.Restore()
if err != nil {
t.Fatal(err)
}
assert.False(t, m.Deleted())
})
t.Run("subject not deleted", func(t *testing.T) {
m := &Subject{DeletedAt: nil, SubjName: "NotDeleted1234"}
err := m.Restore()
if err != nil {
t.Fatal(err)
}
assert.False(t, m.Deleted())
})
}
func TestFindSubject(t *testing.T) {
t.Run("success", func(t *testing.T) {
m := NewSubject("Find Me", SubjPerson, SrcAuto)
if err := m.Save(); err != nil {
t.Fatal(err)
}
if s := FindSubject(m.SubjName); s != nil {
t.Fatal("result must be nil")
}
if s := FindSubject(m.SubjUID); s != nil {
assert.Equal(t, "Find Me", s.SubjName)
} else {
t.Fatal("result must not be nil")
}
})
t.Run("nil", func(t *testing.T) {
r := FindSubject("XXX")
assert.Nil(t, r)
})
t.Run("empty uid", func(t *testing.T) {
r := FindSubject("")
assert.Nil(t, r)
})
}
func TestSubject_Links(t *testing.T) {
t.Run("no-result", func(t *testing.T) {
m := SubjectFixtures.Pointer("john-doe")
links := m.Links()
assert.Empty(t, links)
})
}
func TestSubject_Update(t *testing.T) {
t.Run("success", func(t *testing.T) {
m := NewSubject("Update Me", SubjPerson, SrcAuto)
if err := m.Save(); err != nil {
t.Fatal(err)
}
if err := m.Update("SubjName", "Updated Name"); err != nil {
t.Fatal(err)
} else {
assert.Equal(t, "Updated Name", m.SubjName)
}
})
}
func TestSubject_Updates(t *testing.T) {
t.Run("success", func(t *testing.T) {
m := NewSubject("Update Me", SubjPerson, SrcAuto)
if err := m.Save(); err != nil {
t.Fatal(err)
}
if err := m.Updates(Subject{SubjName: "UpdatedName", SubjType: "UpdatedType"}); err != nil {
t.Fatal(err)
} else {
assert.Equal(t, "UpdatedName", m.SubjName)
assert.Equal(t, "UpdatedType", m.SubjType)
}
})
}
func TestSubject_UpdateName(t *testing.T) {
t.Run("success", func(t *testing.T) {
m := NewSubject("Test Person", SubjPerson, SrcAuto)
if err := m.Save(); err != nil {
t.Fatal(err)
}
assert.Equal(t, "Test Person", m.SubjName)
assert.Equal(t, "test-person", m.SubjSlug)
if s, err := m.UpdateName("New New"); err != nil {
t.Fatal(err)
} else if s == nil {
t.Fatal("subject is nil")
} else {
assert.Equal(t, "New New", m.SubjName)
assert.Equal(t, "new-new", m.SubjSlug)
assert.Equal(t, "New New", s.SubjName)
assert.Equal(t, "new-new", s.SubjSlug)
}
})
t.Run("empty name", func(t *testing.T) {
m := NewSubject("Test Person2", SubjPerson, SrcAuto)
if err := m.Save(); err != nil {
t.Fatal(err)
}
assert.Equal(t, "Test Person2", m.SubjName)
assert.Equal(t, "test-person2", m.SubjSlug)
if s, err := m.UpdateName(""); err == nil {
t.Error("error expected")
} else if s == nil {
t.Fatal("subject is nil")
} else {
assert.Equal(t, "Test Person2", m.SubjName)
assert.Equal(t, "test-person2", m.SubjSlug)
assert.Equal(t, "Test Person2", s.SubjName)
assert.Equal(t, "test-person2", s.SubjSlug)
}
})
}
func TestSubject_RefreshPhotos(t *testing.T) {
subj := SubjectFixtures.Get("john-doe")
if err := subj.RefreshPhotos(); err != nil {
t.Fatal(err)
}
}