64 lines
1.2 KiB
Go
64 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)
|
|
})
|
|
}
|