package entity import ( "github.com/photoprism/photoprism/internal/classify" "testing" "time" "github.com/stretchr/testify/assert" ) func TestNewLabel(t *testing.T) { t.Run("name Unicorn2000 priority 5", func(t *testing.T) { label := NewLabel("Unicorn2000", 5) assert.Equal(t, "Unicorn2000", label.LabelName) assert.Equal(t, "unicorn2000", label.LabelSlug) assert.Equal(t, 5, label.LabelPriority) }) t.Run("name Unknown", func(t *testing.T) { label := NewLabel("", -6) assert.Equal(t, "Unknown", label.LabelName) assert.Equal(t, "unknown", label.LabelSlug) assert.Equal(t, -6, label.LabelPriority) }) } func TestLabel_SetName(t *testing.T) { t.Run("set name", func(t *testing.T) { entity := LabelFixtures["landscape"] assert.Equal(t, "Landscape", entity.LabelName) assert.Equal(t, "landscape", entity.LabelSlug) assert.Equal(t, "landscape", entity.CustomSlug) entity.SetName("Landschaft") assert.Equal(t, "Landschaft", entity.LabelName) assert.Equal(t, "landscape", entity.LabelSlug) assert.Equal(t, "landschaft", entity.CustomSlug) }) t.Run("new name empty", func(t *testing.T) { entity := LabelFixtures["flower"] assert.Equal(t, "Flower", entity.LabelName) assert.Equal(t, "flower", entity.LabelSlug) assert.Equal(t, "flower", entity.CustomSlug) entity.SetName("") assert.Equal(t, "Flower", entity.LabelName) assert.Equal(t, "flower", entity.LabelSlug) assert.Equal(t, "flower", entity.CustomSlug) }) } func TestFirstOrCreateLabel(t *testing.T) { label := LabelFixtures.Get("flower") result := FirstOrCreateLabel(&label) if result == nil { t.Fatal("result should not be nil") } if result.LabelName != label.LabelName { t.Errorf("LabelName should be the same: %s %s", result.LabelName, label.LabelName) } if result.LabelSlug != label.LabelSlug { t.Errorf("LabelName should be the same: %s %s", result.LabelSlug, label.LabelSlug) } } func TestLabel_Update(t *testing.T) { t.Run("update priority and label slug", func(t *testing.T) { classifyLabel := &classify.Label{Name: "classify", Uncertainty: 30, Source: "manual", Priority: 5} Label := &Label{LabelName: "label", LabelSlug: "", CustomSlug: "customslug", LabelPriority: 4} assert.Equal(t, 4, Label.LabelPriority) assert.Equal(t, "", Label.LabelSlug) assert.Equal(t, "customslug", Label.CustomSlug) assert.Equal(t, "label", Label.LabelName) err := Label.UpdateClassify(*classifyLabel) if err != nil { t.Fatal(err) } assert.Equal(t, 5, Label.LabelPriority) assert.Equal(t, "customslug", Label.LabelSlug) assert.Equal(t, "classify", Label.CustomSlug) assert.Equal(t, "Classify", Label.LabelName) }) t.Run("update custom slug", func(t *testing.T) { classifyLabel := &classify.Label{Name: "classify", Uncertainty: 30, Source: "manual", Priority: 5} Label := &Label{LabelName: "label12", LabelSlug: "labelslug", CustomSlug: "", LabelPriority: 5} assert.Equal(t, 5, Label.LabelPriority) assert.Equal(t, "labelslug", Label.LabelSlug) assert.Equal(t, "", Label.CustomSlug) assert.Equal(t, "label12", Label.LabelName) err := Label.UpdateClassify(*classifyLabel) if err != nil { t.Fatal(err) } assert.Equal(t, 5, Label.LabelPriority) assert.Equal(t, "labelslug", Label.LabelSlug) assert.Equal(t, "classify", Label.CustomSlug) assert.Equal(t, "Classify", Label.LabelName) }) t.Run("update name and Categories", func(t *testing.T) { classifyLabel := &classify.Label{Name: "classify", Uncertainty: 30, Source: "manual", Priority: 5, Categories: []string{"flower", "plant"}} Label := &Label{LabelName: "label34", LabelSlug: "labelslug2", CustomSlug: "labelslug2", LabelPriority: 5, LabelCategories: []*Label{LabelFixtures.Pointer("flower")}} assert.Equal(t, 5, Label.LabelPriority) assert.Equal(t, "labelslug2", Label.LabelSlug) assert.Equal(t, "labelslug2", Label.CustomSlug) assert.Equal(t, "label34", Label.LabelName) err := Label.UpdateClassify(*classifyLabel) if err != nil { t.Fatal(err) } assert.Equal(t, 5, Label.LabelPriority) assert.Equal(t, "labelslug2", Label.LabelSlug) assert.Equal(t, "classify", Label.CustomSlug) assert.Equal(t, "Classify", Label.LabelName) }) } func TestLabel_Save(t *testing.T) { t.Run("success", func(t *testing.T) { label := NewLabel("Unicorn2000", 5) initialDate := label.UpdatedAt err := label.Save() if err != nil { t.Fatal(err) } afterDate := label.UpdatedAt assert.True(t, afterDate.After(initialDate)) }) } func TestLabel_Delete(t *testing.T) { t.Run("success", func(t *testing.T) { label := NewLabel("LabelToBeDeleted", 5) err := label.Save() assert.False(t, label.Deleted()) var labels Labels if err := Db().Where("label_name = ?", label.LabelName).Find(&labels).Error; err != nil { t.Fatal(err) } assert.Len(t, labels, 1) err = label.Delete() if err != nil { t.Fatal(err) } if err := Db().Where("label_name = ?", label.LabelName).Find(&labels).Error; err != nil { t.Fatal(err) } assert.Len(t, labels, 0) }) } func TestLabel_Restore(t *testing.T) { t.Run("success", func(t *testing.T) { var deleteTime = time.Date(2020, 1, 1, 0, 0, 0, 0, time.UTC) label := &Label{DeletedAt: &deleteTime, LabelName: "ToBeRestored"} err := label.Save() if err != nil { t.Fatal(err) } assert.True(t, label.Deleted()) err = label.Restore() if err != nil { t.Fatal(err) } assert.False(t, label.Deleted()) }) } func TestFindLabel(t *testing.T) { t.Run("success", func(t *testing.T) { label := &Label{LabelSlug: "find-me-label", LabelName: "Find Me"} err := label.Save() if err != nil { t.Fatal(err) } r := FindLabel("find-me-label") assert.Equal(t, "Find Me", r.LabelName) }) t.Run("nil", func(t *testing.T) { r := FindLabel("XXX") assert.Nil(t, r) }) } func TestLabel_Links(t *testing.T) { t.Run("1 result", func(t *testing.T) { label := LabelFixtures.Get("flower") links := label.Links() assert.Equal(t, "6jxf3jfn2k", links[0].LinkToken) }) }