package entity import ( "testing" "github.com/photoprism/photoprism/internal/classify" "github.com/stretchr/testify/assert" ) func TestNewPhotoLabel(t *testing.T) { t.Run("name Christmas 2018", func(t *testing.T) { photoLabel := NewPhotoLabel(1, 3, 80, "source") assert.Equal(t, uint(0x1), photoLabel.PhotoID) assert.Equal(t, uint(0x3), photoLabel.LabelID) assert.Equal(t, 80, photoLabel.Uncertainty) assert.Equal(t, "source", photoLabel.LabelSrc) }) } func TestPhotoLabel_TableName(t *testing.T) { photoLabel := &PhotoLabel{} tableName := photoLabel.TableName() assert.Equal(t, "photos_labels", tableName) } func TestFirstOrCreatePhotoLabel(t *testing.T) { model := LabelFixtures.PhotoLabel(1000000, "flower", 38, "image") result := FirstOrCreatePhotoLabel(&model) if result == nil { t.Fatal("result should not be nil") } if result.PhotoID != model.PhotoID { t.Errorf("PhotoID should be the same: %d %d", result.PhotoID, model.PhotoID) } if result.LabelID != model.LabelID { t.Errorf("LabelID should be the same: %d %d", result.LabelID, model.LabelID) } } func TestPhotoLabel_ClassifyLabel(t *testing.T) { t.Run("success", func(t *testing.T) { pl := LabelFixtures.PhotoLabel(1000000, "flower", 38, "image") r := pl.ClassifyLabel() assert.Equal(t, "Flower", r.Name) assert.Equal(t, 38, r.Uncertainty) assert.Equal(t, "image", r.Source) }) t.Run("label = nil", func(t *testing.T) { photoLabel := NewPhotoLabel(1, 3, 80, "source") result := photoLabel.ClassifyLabel() assert.Equal(t, classify.Label{}, result) }) } func TestPhotoLabel_Save(t *testing.T) { t.Run("success", func(t *testing.T) { photoLabel := NewPhotoLabel(13, 1000, 99, "image") err := photoLabel.Save() if err != nil { t.Fatal(err) } }) }