photoprism/internal/query/labels_test.go

106 lines
2.3 KiB
Go
Raw Normal View History

2020-01-28 20:59:02 +01:00
package query
import (
"github.com/photoprism/photoprism/internal/entity"
2020-01-28 20:59:02 +01:00
"github.com/photoprism/photoprism/internal/form"
"github.com/stretchr/testify/assert"
2020-05-13 11:57:54 +02:00
"testing"
2020-01-28 20:59:02 +01:00
)
func TestLabels(t *testing.T) {
2020-01-28 20:59:02 +01:00
t.Run("search with query", func(t *testing.T) {
query := form.NewLabelSearch("Query:C Count:1005 Order:slug")
result, err := Labels(query)
2020-01-28 20:59:02 +01:00
if err != nil {
t.Fatal(err)
}
t.Logf("results: %+v", result)
assert.LessOrEqual(t, 2, len(result))
for _, r := range result {
assert.IsType(t, LabelResult{}, r)
assert.NotEmpty(t, r.ID)
assert.NotEmpty(t, r.LabelName)
assert.NotEmpty(t, r.LabelSlug)
assert.NotEmpty(t, r.CustomSlug)
if fix, ok := entity.LabelFixtures[r.LabelSlug]; ok {
assert.Equal(t, fix.LabelName, r.LabelName)
assert.Equal(t, fix.LabelSlug, r.LabelSlug)
assert.Equal(t, fix.CustomSlug, r.CustomSlug)
}
}
2020-01-28 20:59:02 +01:00
})
t.Run("search for favorites", func(t *testing.T) {
2020-05-13 11:57:54 +02:00
query := form.NewLabelSearch("Favorites:true Count:15")
result, err := Labels(query)
2020-01-28 20:59:02 +01:00
if err != nil {
t.Fatal(err)
}
assert.LessOrEqual(t, 2, len(result))
for _, r := range result {
assert.True(t, r.LabelFavorite)
assert.IsType(t, LabelResult{}, r)
assert.NotEmpty(t, r.ID)
assert.NotEmpty(t, r.LabelName)
assert.NotEmpty(t, r.LabelSlug)
assert.NotEmpty(t, r.CustomSlug)
if fix, ok := entity.LabelFixtures[r.LabelSlug]; ok {
assert.Equal(t, fix.LabelName, r.LabelName)
assert.Equal(t, fix.LabelSlug, r.LabelSlug)
assert.Equal(t, fix.CustomSlug, r.CustomSlug)
}
}
2020-01-28 20:59:02 +01:00
})
t.Run("search with empty query", func(t *testing.T) {
query := form.NewLabelSearch("")
result, err := Labels(query)
if err != nil {
t.Fatal(err)
}
2020-05-13 11:57:54 +02:00
t.Log(result)
assert.LessOrEqual(t, 3, len(result))
})
t.Run("search with invalid query string", func(t *testing.T) {
query := form.NewLabelSearch("xxx:bla")
result, err := Labels(query)
assert.Error(t, err, "unknown filter")
assert.Empty(t, result)
})
2020-05-13 11:57:54 +02:00
t.Run("search for ID", func(t *testing.T) {
f := form.LabelSearch{
Query: "",
ID: "lt9k3pw1wowuy3c4",
Slug: "",
Name: "",
All: false,
Favorites: false,
Count: 0,
Offset: 0,
Order: "",
}
result, err := Labels(f)
if err != nil {
t.Fatal(err)
}
assert.Equal(t, "cake", result[0].LabelSlug)
})
2020-01-28 20:59:02 +01:00
}