photoprism/internal/query/photo_selection_test.go
2021-09-18 15:32:39 +02:00

63 lines
1.2 KiB
Go

package query
import (
"testing"
"github.com/photoprism/photoprism/internal/entity"
"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) {
f := form.Selection{
Photos: []string{},
}
r, err := PhotoSelection(f)
assert.Equal(t, "no items selected", err.Error())
assert.Empty(t, r)
})
t.Run("photos selected", func(t *testing.T) {
f := form.Selection{
Photos: []string{"pt9jtdre2lvl0yh7", "pt9jtdre2lvl0yh8"},
}
r, err := PhotoSelection(f)
if err != nil {
t.Fatal(err)
}
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)
})
}