package query import ( "testing" "github.com/stretchr/testify/assert" ) func TestLabelBySlug(t *testing.T) { t.Run("file found", func(t *testing.T) { label, err := LabelBySlug("flower") if err != nil { t.Fatal(err) } assert.Equal(t, "Flower", label.LabelName) }) t.Run("no file found", func(t *testing.T) { label, err := LabelBySlug("111") assert.Error(t, err, "record not found") assert.Empty(t, label.ID) }) } func TestLabelByUUID(t *testing.T) { t.Run("file found", func(t *testing.T) { label, err := LabelByUUID("lt9k3pw1wowuy3c5") if err != nil { t.Fatal(err) } assert.Equal(t, "COW", label.LabelName) }) t.Run("no file found", func(t *testing.T) { label, err := LabelByUUID("111") assert.Error(t, err, "record not found") assert.Empty(t, label.ID) }) } func TestLabelThumbBySlug(t *testing.T) { t.Run("file found", func(t *testing.T) { file, err := LabelThumbBySlug("flower") if err != nil { t.Fatal(err) } assert.Equal(t, "exampleFileName.jpg", file.FileName) }) t.Run("no file found", func(t *testing.T) { file, err := LabelThumbBySlug("cow") assert.Error(t, err, "record not found") t.Log(file) }) } func TestLabelThumbByUUID(t *testing.T) { t.Run("file found", func(t *testing.T) { file, err := LabelThumbByUUID("lt9k3pw1wowuy3c4") if err != nil { t.Fatal(err) } assert.Equal(t, "exampleFileName.jpg", file.FileName) }) t.Run("no file found", func(t *testing.T) { file, err := LabelThumbByUUID("14") assert.Error(t, err, "record not found") t.Log(file) }) } func TestPhotoLabel(t *testing.T) { t.Run("photo label found", func(t *testing.T) { r, err := PhotoLabel(uint(1000000), uint(1000001)) if err != nil { t.Fatal(err) } assert.Equal(t, 38, r.Uncertainty) }) t.Run("no photo label found", func(t *testing.T) { r, err := PhotoLabel(uint(1000000), uint(1000003)) assert.Equal(t, "record not found", err.Error()) t.Log(r) }) }