zishuo/chaisql
1
0
Fork 0
mirror of https://github.com/chaisql/chai.git synced 2025-10-05 23:57:01 +08:00
Code Issues Packages Projects Releases Wiki Activity

Move tests to query package

Browse Source
This commit is contained in:
Asdine El Hrychy
2019-12-04 20:47:19 +01:00
parent 56abf30461
commit 5b51b1ac0c
14 changed files with 618 additions and 550 deletions
Download Patch File Download Diff File Expand all files Collapse all files

78
parser/parser_test.go
View File

@@ -1,90 +1,12 @@
package parser
import (
"strings"
"testing"
"github.com/asdine/genji/query"
"github.com/stretchr/testify/require"
)
func TestParserExpr(t *testing.T) {
tests := []struct {
name string
s string
expected query.Expr
}{
{"=", "age = 10", query.Eq(query.FieldSelector("age"), query.Int8Value(10))},
{"!=", "age != 10", query.Neq(query.FieldSelector("age"), query.Int8Value(10))},
{">", "age > 10", query.Gt(query.FieldSelector("age"), query.Int8Value(10))},
{">=", "age >= 10", query.Gte(query.FieldSelector("age"), query.Int8Value(10))},
{"<", "age < 10", query.Lt(query.FieldSelector("age"), query.Int8Value(10))},
{"<=", "age <= 10", query.Lte(query.FieldSelector("age"), query.Int8Value(10))},
{"AND", "age = 10 AND age <= 11",
query.And(
query.Eq(query.FieldSelector("age"), query.Int8Value(10)),
query.Lte(query.FieldSelector("age"), query.Int8Value(11)),
)},
{"OR", "age = 10 OR age = 11",
query.Or(
query.Eq(query.FieldSelector("age"), query.Int8Value(10)),
query.Eq(query.FieldSelector("age"), query.Int8Value(11)),
)},
{"AND then OR", "age >= 10 AND age > $age OR age < 10.4",
query.Or(
query.And(
query.Gte(query.FieldSelector("age"), query.Int8Value(10)),
query.Gt(query.FieldSelector("age"), query.NamedParam("age")),
),
query.Lt(query.FieldSelector("age"), query.Float64Value(10.4)),
)},
{"with NULL", "age > NULL", query.Gt(query.FieldSelector("age"), query.NullValue())},
}
for _, test := range tests {
t.Run(test.name, func(t *testing.T) {
ex, err := NewParser(strings.NewReader(test.s)).ParseExpr()
require.NoError(t, err)
require.EqualValues(t, test.expected, ex)
})
}
}
func TestParserParams(t *testing.T) {
tests := []struct {
name string
s string
expected query.Expr
errored bool
}{
{"one positional", "age = ?", query.Eq(query.FieldSelector("age"), query.PositionalParam(1)), false},
{"multiple positional", "age = ? AND age <= ?",
query.And(
query.Eq(query.FieldSelector("age"), query.PositionalParam(1)),
query.Lte(query.FieldSelector("age"), query.PositionalParam(2)),
), false},
{"one named", "age = $age", query.Eq(query.FieldSelector("age"), query.NamedParam("age")), false},
{"multiple named", "age = $foo OR age = $bar",
query.Or(
query.Eq(query.FieldSelector("age"), query.NamedParam("foo")),
query.Eq(query.FieldSelector("age"), query.NamedParam("bar")),
), false},
{"mixed", "age >= ? AND age > $foo OR age < ?", nil, true},
}
for _, test := range tests {
t.Run(test.name, func(t *testing.T) {
ex, err := NewParser(strings.NewReader(test.s)).ParseExpr()
if test.errored {
require.Error(t, err)
} else {
require.NoError(t, err)
require.EqualValues(t, test.expected, ex)
}
})
}
}
func TestParserMultiStatement(t *testing.T) {
tests := []struct {
name string