1
0
mirror of https://github.com/pocketbase/pocketbase.git synced 2024-11-24 17:07:00 +02:00
pocketbase/tools/search/sort_test.go
2023-01-07 22:27:11 +02:00

71 lines
2.1 KiB
Go

package search_test
import (
"encoding/json"
"testing"
"github.com/pocketbase/pocketbase/tools/search"
)
func TestSortFieldBuildExpr(t *testing.T) {
resolver := search.NewSimpleFieldResolver("test1", "test2", "test3", "test4.sub")
scenarios := []struct {
sortField search.SortField
expectError bool
expectExpression string
}{
// empty
{search.SortField{"", search.SortDesc}, true, ""},
// unknown field
{search.SortField{"unknown", search.SortAsc}, true, ""},
// placeholder field
{search.SortField{"'test'", search.SortAsc}, true, ""},
// null field
{search.SortField{"null", search.SortAsc}, true, ""},
// allowed field - asc
{search.SortField{"test1", search.SortAsc}, false, "[[test1]] ASC"},
// allowed field - desc
{search.SortField{"test1", search.SortDesc}, false, "[[test1]] DESC"},
// special @random field (ignore direction)
{search.SortField{"@random", search.SortDesc}, false, "RANDOM()"},
}
for i, s := range scenarios {
result, err := s.sortField.BuildExpr(resolver)
hasErr := err != nil
if hasErr != s.expectError {
t.Errorf("(%d) Expected hasErr %v, got %v (%v)", i, s.expectError, hasErr, err)
continue
}
if result != s.expectExpression {
t.Errorf("(%d) Expected expression %v, got %v", i, s.expectExpression, result)
}
}
}
func TestParseSortFromString(t *testing.T) {
scenarios := []struct {
value string
expectedJson string
}{
{"", `[{"name":"","direction":"ASC"}]`},
{"test", `[{"name":"test","direction":"ASC"}]`},
{"+test", `[{"name":"test","direction":"ASC"}]`},
{"-test", `[{"name":"test","direction":"DESC"}]`},
{"test1,-test2,+test3", `[{"name":"test1","direction":"ASC"},{"name":"test2","direction":"DESC"},{"name":"test3","direction":"ASC"}]`},
{"@random,-test", `[{"name":"@random","direction":"ASC"},{"name":"test","direction":"DESC"}]`},
}
for i, s := range scenarios {
result := search.ParseSortFromString(s.value)
encoded, _ := json.Marshal(result)
if string(encoded) != s.expectedJson {
t.Errorf("(%d) Expected expression %v, got %v", i, s.expectedJson, string(encoded))
}
}
}