photoprism/internal/query/selection_test.go

64 lines
1.2 KiB
Go
Raw Normal View History

2020-05-13 17:23:16 +02:00
package query
import (
"testing"
"github.com/photoprism/photoprism/internal/entity"
2020-05-13 17:23:16 +02:00
"github.com/photoprism/photoprism/internal/form"
"github.com/stretchr/testify/assert"
)
func TestPhotoSelection(t *testing.T) {
t.Run("no items selected", func(t *testing.T) {
2020-05-13 17:23:16 +02:00
f := form.Selection{
Photos: []string{},
}
2020-05-13 17:23:16 +02:00
r, err := PhotoSelection(f)
assert.Equal(t, "no items selected", err.Error())
2020-05-13 17:23:16 +02:00
assert.Empty(t, r)
})
t.Run("photos selected", func(t *testing.T) {
f := form.Selection{
Photos: []string{"pt9jtdre2lvl0yh7", "pt9jtdre2lvl0yh8"},
}
2020-05-13 17:23:16 +02:00
r, err := PhotoSelection(f)
2020-05-13 17:23:16 +02:00
if err != nil {
t.Fatal(err)
}
2020-05-13 17:23:16 +02:00
assert.Equal(t, 2, len(r))
assert.IsType(t, entity.Photos{}, r)
})
}
func TestFileSelection(t *testing.T) {
t.Run("no items selected", func(t *testing.T) {
f := form.Selection{
Photos: []string{},
}
r, err := FileSelection(f)
assert.Equal(t, "no items selected", err.Error())
assert.Empty(t, r)
})
t.Run("files selected", func(t *testing.T) {
f := form.Selection{
Photos: []string{"pt9jtdre2lvl0yh7", "pt9jtdre2lvl0yh8"},
}
r, err := FileSelection(f)
if err != nil {
t.Fatal(err)
}
assert.Equal(t, 3, len(r))
assert.IsType(t, entity.Files{}, r)
2020-05-13 17:23:16 +02:00
})
}