mirror of https://github.com/milvus-io/milvus.git
542 lines
23 KiB
Go
542 lines
23 KiB
Go
|
package planparserv2
|
||
|
|
|
||
|
|
import (
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"github.com/stretchr/testify/suite"
|
||
|
|
|
||
|
|
"github.com/milvus-io/milvus-proto/go-api/v2/schemapb"
|
||
|
|
"github.com/milvus-io/milvus/pkg/util/typeutil"
|
||
|
|
)
|
||
|
|
|
||
|
|
type FillExpressionValueSuite struct {
|
||
|
|
suite.Suite
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestFillExpressionValue(t *testing.T) {
|
||
|
|
suite.Run(t, new(FillExpressionValueSuite))
|
||
|
|
}
|
||
|
|
|
||
|
|
type testcase struct {
|
||
|
|
expr string
|
||
|
|
values map[string]*schemapb.TemplateValue
|
||
|
|
}
|
||
|
|
|
||
|
|
func (s *FillExpressionValueSuite) assertValidExpr(helper *typeutil.SchemaHelper, exprStr string, templateValues map[string]*schemapb.TemplateValue) {
|
||
|
|
expr, err := ParseExpr(helper, exprStr, templateValues)
|
||
|
|
s.NoError(err, exprStr)
|
||
|
|
s.NotNil(expr, exprStr)
|
||
|
|
ShowExpr(expr)
|
||
|
|
}
|
||
|
|
|
||
|
|
func (s *FillExpressionValueSuite) assertInvalidExpr(helper *typeutil.SchemaHelper, exprStr string, templateValues map[string]*schemapb.TemplateValue) {
|
||
|
|
expr, err := ParseExpr(helper, exprStr, templateValues)
|
||
|
|
s.Error(err, exprStr)
|
||
|
|
s.Nil(expr, exprStr)
|
||
|
|
}
|
||
|
|
|
||
|
|
func (s *FillExpressionValueSuite) TestTermExpr() {
|
||
|
|
s.Run("normal case", func() {
|
||
|
|
testcases := []testcase{
|
||
|
|
{`Int64Field in {age}`, map[string]*schemapb.TemplateValue{
|
||
|
|
"age": generateExpressionFieldData(schemapb.DataType_Array, []interface{}{
|
||
|
|
generateExpressionFieldData(schemapb.DataType_Int64, int64(1)),
|
||
|
|
generateExpressionFieldData(schemapb.DataType_Int64, int64(2)),
|
||
|
|
generateExpressionFieldData(schemapb.DataType_Int64, int64(3)),
|
||
|
|
generateExpressionFieldData(schemapb.DataType_Int64, int64(4)),
|
||
|
|
}),
|
||
|
|
}},
|
||
|
|
{`FloatField in {age}`, map[string]*schemapb.TemplateValue{
|
||
|
|
"age": generateExpressionFieldData(schemapb.DataType_Array, []interface{}{
|
||
|
|
generateExpressionFieldData(schemapb.DataType_Float, 1.1),
|
||
|
|
generateExpressionFieldData(schemapb.DataType_Float, 2.2),
|
||
|
|
generateExpressionFieldData(schemapb.DataType_Float, 3.3),
|
||
|
|
generateExpressionFieldData(schemapb.DataType_Float, 4.4),
|
||
|
|
}),
|
||
|
|
}},
|
||
|
|
{`A in {list}`, map[string]*schemapb.TemplateValue{
|
||
|
|
"list": generateExpressionFieldData(schemapb.DataType_Array, []interface{}{
|
||
|
|
generateExpressionFieldData(schemapb.DataType_Int64, int64(1)),
|
||
|
|
generateExpressionFieldData(schemapb.DataType_Float, 2.2),
|
||
|
|
generateExpressionFieldData(schemapb.DataType_String, "abc"),
|
||
|
|
generateExpressionFieldData(schemapb.DataType_Bool, false),
|
||
|
|
}),
|
||
|
|
}},
|
||
|
|
{`ArrayField in {list}`, map[string]*schemapb.TemplateValue{
|
||
|
|
"list": generateExpressionFieldData(schemapb.DataType_Array, []interface{}{
|
||
|
|
generateExpressionFieldData(schemapb.DataType_Array, []interface{}{
|
||
|
|
generateExpressionFieldData(schemapb.DataType_Int64, int64(1)),
|
||
|
|
generateExpressionFieldData(schemapb.DataType_Int64, int64(2)),
|
||
|
|
generateExpressionFieldData(schemapb.DataType_Int64, int64(3)),
|
||
|
|
}),
|
||
|
|
generateExpressionFieldData(schemapb.DataType_Array, []interface{}{
|
||
|
|
generateExpressionFieldData(schemapb.DataType_Int64, int64(4)),
|
||
|
|
generateExpressionFieldData(schemapb.DataType_Int64, int64(5)),
|
||
|
|
generateExpressionFieldData(schemapb.DataType_Int64, int64(6)),
|
||
|
|
}),
|
||
|
|
generateExpressionFieldData(schemapb.DataType_Array, []interface{}{
|
||
|
|
generateExpressionFieldData(schemapb.DataType_Int64, int64(7)),
|
||
|
|
generateExpressionFieldData(schemapb.DataType_Int64, int64(8)),
|
||
|
|
generateExpressionFieldData(schemapb.DataType_Int64, int64(9)),
|
||
|
|
}),
|
||
|
|
}),
|
||
|
|
}},
|
||
|
|
{`ArrayField[0] in {list}`, map[string]*schemapb.TemplateValue{
|
||
|
|
"list": generateExpressionFieldData(schemapb.DataType_Array, []interface{}{
|
||
|
|
generateExpressionFieldData(schemapb.DataType_Int64, int64(1)),
|
||
|
|
generateExpressionFieldData(schemapb.DataType_Int64, int64(2)),
|
||
|
|
generateExpressionFieldData(schemapb.DataType_Int64, int64(3)),
|
||
|
|
}),
|
||
|
|
}},
|
||
|
|
{`Int64Field in {empty_list}`, map[string]*schemapb.TemplateValue{
|
||
|
|
"empty_list": generateExpressionFieldData(schemapb.DataType_Array, []interface{}{}),
|
||
|
|
}},
|
||
|
|
}
|
||
|
|
schemaH := newTestSchemaHelper(s.T())
|
||
|
|
for _, c := range testcases {
|
||
|
|
s.assertValidExpr(schemaH, c.expr, c.values)
|
||
|
|
}
|
||
|
|
})
|
||
|
|
|
||
|
|
s.Run("failed case", func() {
|
||
|
|
testcases := []testcase{
|
||
|
|
{`Int64Field in {age}`, map[string]*schemapb.TemplateValue{
|
||
|
|
"age": generateExpressionFieldData(schemapb.DataType_Array, []interface{}{
|
||
|
|
generateExpressionFieldData(schemapb.DataType_String, "abc"),
|
||
|
|
generateExpressionFieldData(schemapb.DataType_String, "def"),
|
||
|
|
}),
|
||
|
|
}},
|
||
|
|
{`StringField in {list}`, map[string]*schemapb.TemplateValue{
|
||
|
|
"list": generateExpressionFieldData(schemapb.DataType_Array, []interface{}{
|
||
|
|
generateExpressionFieldData(schemapb.DataType_Int64, int64(1)),
|
||
|
|
generateExpressionFieldData(schemapb.DataType_String, "abc"),
|
||
|
|
generateExpressionFieldData(schemapb.DataType_Float, 2.2),
|
||
|
|
generateExpressionFieldData(schemapb.DataType_Bool, false),
|
||
|
|
}),
|
||
|
|
}},
|
||
|
|
{"ArrayField[0] in {list}", map[string]*schemapb.TemplateValue{
|
||
|
|
"list": generateExpressionFieldData(schemapb.DataType_Array, []interface{}{
|
||
|
|
generateExpressionFieldData(schemapb.DataType_Int64, int64(1)),
|
||
|
|
generateExpressionFieldData(schemapb.DataType_String, "abc"),
|
||
|
|
generateExpressionFieldData(schemapb.DataType_Float, 3.2),
|
||
|
|
}),
|
||
|
|
}},
|
||
|
|
{"Int64Field not in {not_list}", map[string]*schemapb.TemplateValue{
|
||
|
|
"not_list": generateExpressionFieldData(schemapb.DataType_Int64, int64(33)),
|
||
|
|
}},
|
||
|
|
{"Int64Field not in {not_list}", map[string]*schemapb.TemplateValue{
|
||
|
|
"age": generateExpressionFieldData(schemapb.DataType_Int64, int64(33)),
|
||
|
|
}},
|
||
|
|
}
|
||
|
|
schemaH := newTestSchemaHelper(s.T())
|
||
|
|
for _, c := range testcases {
|
||
|
|
s.assertInvalidExpr(schemaH, c.expr, c.values)
|
||
|
|
}
|
||
|
|
})
|
||
|
|
}
|
||
|
|
|
||
|
|
func (s *FillExpressionValueSuite) TestUnaryRange() {
|
||
|
|
s.Run("normal case", func() {
|
||
|
|
testcases := []testcase{
|
||
|
|
{`Int64Field == 10`, nil},
|
||
|
|
{`Int64Field > {target}`, map[string]*schemapb.TemplateValue{
|
||
|
|
"target": generateExpressionFieldData(schemapb.DataType_Int64, int64(11)),
|
||
|
|
}},
|
||
|
|
{`FloatField < {target}`, map[string]*schemapb.TemplateValue{
|
||
|
|
"target": generateExpressionFieldData(schemapb.DataType_Float, float64(12.3)),
|
||
|
|
}},
|
||
|
|
{`DoubleField != {target}`, map[string]*schemapb.TemplateValue{
|
||
|
|
"target": generateExpressionFieldData(schemapb.DataType_Double, 3.5),
|
||
|
|
}},
|
||
|
|
{`ArrayField[0] >= {target}`, map[string]*schemapb.TemplateValue{
|
||
|
|
"target": generateExpressionFieldData(schemapb.DataType_Int64, int64(3)),
|
||
|
|
}},
|
||
|
|
{`BoolField == {bool}`, map[string]*schemapb.TemplateValue{
|
||
|
|
"bool": generateExpressionFieldData(schemapb.DataType_Bool, false),
|
||
|
|
}},
|
||
|
|
{`{str} != StringField`, map[string]*schemapb.TemplateValue{
|
||
|
|
"str": generateExpressionFieldData(schemapb.DataType_String, "abc"),
|
||
|
|
}},
|
||
|
|
{`{target} > Int64Field`, map[string]*schemapb.TemplateValue{
|
||
|
|
"target": generateExpressionFieldData(schemapb.DataType_Int64, int64(11)),
|
||
|
|
}},
|
||
|
|
}
|
||
|
|
schemaH := newTestSchemaHelper(s.T())
|
||
|
|
for _, c := range testcases {
|
||
|
|
s.assertValidExpr(schemaH, c.expr, c.values)
|
||
|
|
}
|
||
|
|
})
|
||
|
|
|
||
|
|
s.Run("failed case", func() {
|
||
|
|
testcases := []testcase{
|
||
|
|
{`Int64Field == 10.5`, nil},
|
||
|
|
{`Int64Field > {target}`, map[string]*schemapb.TemplateValue{
|
||
|
|
"target": generateExpressionFieldData(schemapb.DataType_Double, 11.2),
|
||
|
|
}},
|
||
|
|
{`FloatField < {target}`, map[string]*schemapb.TemplateValue{
|
||
|
|
"target": generateExpressionFieldData(schemapb.DataType_String, "abc"),
|
||
|
|
}},
|
||
|
|
{`DoubleField != {target}`, map[string]*schemapb.TemplateValue{
|
||
|
|
"target": generateExpressionFieldData(schemapb.DataType_Bool, false),
|
||
|
|
}},
|
||
|
|
{`ArrayField[0] >= {target}`, map[string]*schemapb.TemplateValue{
|
||
|
|
"target": generateExpressionFieldData(schemapb.DataType_Double, 3.5),
|
||
|
|
}},
|
||
|
|
{`BoolField == {bool}`, map[string]*schemapb.TemplateValue{
|
||
|
|
"bool": generateExpressionFieldData(schemapb.DataType_String, "abc"),
|
||
|
|
}},
|
||
|
|
{`{str} != StringField`, map[string]*schemapb.TemplateValue{
|
||
|
|
"str": generateExpressionFieldData(schemapb.DataType_Int64, int64(5)),
|
||
|
|
}},
|
||
|
|
{`{int} != StringField`, map[string]*schemapb.TemplateValue{
|
||
|
|
"int": generateExpressionFieldData(schemapb.DataType_Int64, int64(5)),
|
||
|
|
}},
|
||
|
|
{`{str} != StringField`, map[string]*schemapb.TemplateValue{
|
||
|
|
"int": generateExpressionFieldData(schemapb.DataType_Int64, int64(5)),
|
||
|
|
}},
|
||
|
|
}
|
||
|
|
schemaH := newTestSchemaHelper(s.T())
|
||
|
|
for _, c := range testcases {
|
||
|
|
s.assertInvalidExpr(schemaH, c.expr, c.values)
|
||
|
|
}
|
||
|
|
})
|
||
|
|
}
|
||
|
|
|
||
|
|
func (s *FillExpressionValueSuite) TestBinaryRange() {
|
||
|
|
s.Run("normal case", func() {
|
||
|
|
testcases := []testcase{
|
||
|
|
{`10 < Int64Field < 20`, nil},
|
||
|
|
{`{max} > Int64Field > {min}`, map[string]*schemapb.TemplateValue{
|
||
|
|
"min": generateExpressionFieldData(schemapb.DataType_Int64, int64(11)),
|
||
|
|
"max": generateExpressionFieldData(schemapb.DataType_Int64, int64(22)),
|
||
|
|
}},
|
||
|
|
{`{min} <= FloatField <= {max}`, map[string]*schemapb.TemplateValue{
|
||
|
|
"min": generateExpressionFieldData(schemapb.DataType_Float, float64(11)),
|
||
|
|
"max": generateExpressionFieldData(schemapb.DataType_Float, float64(22)),
|
||
|
|
}},
|
||
|
|
{`{min} < DoubleField < {max}`, map[string]*schemapb.TemplateValue{
|
||
|
|
"min": generateExpressionFieldData(schemapb.DataType_Double, float64(11)),
|
||
|
|
"max": generateExpressionFieldData(schemapb.DataType_Double, float64(22)),
|
||
|
|
}},
|
||
|
|
{`{max} >= ArrayField[0] >= {min}`, map[string]*schemapb.TemplateValue{
|
||
|
|
"min": generateExpressionFieldData(schemapb.DataType_Int64, int64(11)),
|
||
|
|
"max": generateExpressionFieldData(schemapb.DataType_Int64, int64(22)),
|
||
|
|
}},
|
||
|
|
{`{max} > Int64Field >= 10`, map[string]*schemapb.TemplateValue{
|
||
|
|
"max": generateExpressionFieldData(schemapb.DataType_Int64, int64(22)),
|
||
|
|
}},
|
||
|
|
{`30 >= Int64Field > {min}`, map[string]*schemapb.TemplateValue{
|
||
|
|
"min": generateExpressionFieldData(schemapb.DataType_Int64, int64(11)),
|
||
|
|
}},
|
||
|
|
{`10 < Int64Field <= {max}`, map[string]*schemapb.TemplateValue{
|
||
|
|
"max": generateExpressionFieldData(schemapb.DataType_Int64, int64(22)),
|
||
|
|
}},
|
||
|
|
{`{min} <= Int64Field < 20`, map[string]*schemapb.TemplateValue{
|
||
|
|
"min": generateExpressionFieldData(schemapb.DataType_Int64, int64(11)),
|
||
|
|
}},
|
||
|
|
}
|
||
|
|
|
||
|
|
schemaH := newTestSchemaHelper(s.T())
|
||
|
|
for _, c := range testcases {
|
||
|
|
s.assertValidExpr(schemaH, c.expr, c.values)
|
||
|
|
}
|
||
|
|
})
|
||
|
|
|
||
|
|
s.Run("failed case", func() {
|
||
|
|
testcases := []testcase{
|
||
|
|
{`10 < Int64Field < 20.5`, nil},
|
||
|
|
{`{max} > Int64Field > {min}`, map[string]*schemapb.TemplateValue{
|
||
|
|
"min": generateExpressionFieldData(schemapb.DataType_Int64, int64(11)),
|
||
|
|
"max": generateExpressionFieldData(schemapb.DataType_Double, 22.5),
|
||
|
|
}},
|
||
|
|
{`{min} <= FloatField <= {max}`, map[string]*schemapb.TemplateValue{
|
||
|
|
"min": generateExpressionFieldData(schemapb.DataType_String, "abc"),
|
||
|
|
"max": generateExpressionFieldData(schemapb.DataType_Int64, int64(11)),
|
||
|
|
}},
|
||
|
|
{`{min} < DoubleField < {max}`, map[string]*schemapb.TemplateValue{
|
||
|
|
"min": generateExpressionFieldData(schemapb.DataType_Int64, int64(33)),
|
||
|
|
"max": generateExpressionFieldData(schemapb.DataType_Int64, int64(22)),
|
||
|
|
}},
|
||
|
|
{`{max} >= ArrayField[0] >= {min}`, map[string]*schemapb.TemplateValue{
|
||
|
|
"min": generateExpressionFieldData(schemapb.DataType_Double, 11.5),
|
||
|
|
"max": generateExpressionFieldData(schemapb.DataType_Int64, int64(22)),
|
||
|
|
}},
|
||
|
|
{`{max} >= Int64Field >= {min}`, map[string]*schemapb.TemplateValue{
|
||
|
|
"max": generateExpressionFieldData(schemapb.DataType_Int64, int64(22)),
|
||
|
|
}},
|
||
|
|
{`{max} > Int64Field`, map[string]*schemapb.TemplateValue{
|
||
|
|
"max": generateExpressionFieldData(schemapb.DataType_Bool, false),
|
||
|
|
}},
|
||
|
|
{`{$meta} > Int64Field`, map[string]*schemapb.TemplateValue{
|
||
|
|
"$meta": generateExpressionFieldData(schemapb.DataType_Int64, int64(22)),
|
||
|
|
}},
|
||
|
|
}
|
||
|
|
|
||
|
|
schemaH := newTestSchemaHelper(s.T())
|
||
|
|
for _, c := range testcases {
|
||
|
|
s.assertInvalidExpr(schemaH, c.expr, c.values)
|
||
|
|
}
|
||
|
|
})
|
||
|
|
}
|
||
|
|
|
||
|
|
func (s *FillExpressionValueSuite) TestBinaryArithOpEvalRange() {
|
||
|
|
s.Run("normal case", func() {
|
||
|
|
testcases := []testcase{
|
||
|
|
{`Int64Field + 5.5 == 10.5`, nil},
|
||
|
|
{`Int64Field - {offset} >= {target}`, map[string]*schemapb.TemplateValue{
|
||
|
|
"offset": generateExpressionFieldData(schemapb.DataType_Double, 3.5),
|
||
|
|
"target": generateExpressionFieldData(schemapb.DataType_Double, 11.5),
|
||
|
|
}},
|
||
|
|
{`Int64Field * 3.5 <= {target}`, map[string]*schemapb.TemplateValue{
|
||
|
|
"target": generateExpressionFieldData(schemapb.DataType_Double, 11.5),
|
||
|
|
}},
|
||
|
|
{`Int64Field / {offset} > 11.5`, map[string]*schemapb.TemplateValue{
|
||
|
|
"offset": generateExpressionFieldData(schemapb.DataType_Double, 3.5),
|
||
|
|
}},
|
||
|
|
{`ArrayField[0] % {offset} < 11`, map[string]*schemapb.TemplateValue{
|
||
|
|
"offset": generateExpressionFieldData(schemapb.DataType_Int64, int64(3)),
|
||
|
|
}},
|
||
|
|
{`array_length(ArrayField) == {length}`, map[string]*schemapb.TemplateValue{
|
||
|
|
"length": generateExpressionFieldData(schemapb.DataType_Int64, int64(3)),
|
||
|
|
}},
|
||
|
|
{`array_length(ArrayField) > {length}`, map[string]*schemapb.TemplateValue{
|
||
|
|
"length": generateExpressionFieldData(schemapb.DataType_Int64, int64(3)),
|
||
|
|
}},
|
||
|
|
}
|
||
|
|
|
||
|
|
schemaH := newTestSchemaHelper(s.T())
|
||
|
|
for _, c := range testcases {
|
||
|
|
s.assertValidExpr(schemaH, c.expr, c.values)
|
||
|
|
}
|
||
|
|
})
|
||
|
|
|
||
|
|
s.Run("failed case", func() {
|
||
|
|
testcases := []testcase{
|
||
|
|
{`Int64Field + 6 == 12.5`, nil},
|
||
|
|
{`Int64Field - {offset} == {target}`, map[string]*schemapb.TemplateValue{
|
||
|
|
"offset": generateExpressionFieldData(schemapb.DataType_Int64, int64(4)),
|
||
|
|
"target": generateExpressionFieldData(schemapb.DataType_Double, 13.5),
|
||
|
|
}},
|
||
|
|
{`Int64Field * 6 == {target}`, map[string]*schemapb.TemplateValue{
|
||
|
|
"target": generateExpressionFieldData(schemapb.DataType_Double, 13.5),
|
||
|
|
}},
|
||
|
|
{`Int64Field / {offset} == 11.5`, map[string]*schemapb.TemplateValue{
|
||
|
|
"offset": generateExpressionFieldData(schemapb.DataType_Int64, int64(6)),
|
||
|
|
}},
|
||
|
|
{`Int64Field % {offset} < 11`, map[string]*schemapb.TemplateValue{
|
||
|
|
"offset": generateExpressionFieldData(schemapb.DataType_Double, 3.5),
|
||
|
|
}},
|
||
|
|
{`Int64Field + {offset} < {target}`, map[string]*schemapb.TemplateValue{
|
||
|
|
"target": generateExpressionFieldData(schemapb.DataType_Double, 3.5),
|
||
|
|
}},
|
||
|
|
{`Int64Field + {offset} < {target}`, map[string]*schemapb.TemplateValue{
|
||
|
|
"offset": generateExpressionFieldData(schemapb.DataType_String, "abc"),
|
||
|
|
"target": generateExpressionFieldData(schemapb.DataType_Int64, int64(15)),
|
||
|
|
}},
|
||
|
|
{`Int64Field + {offset} < {target}`, map[string]*schemapb.TemplateValue{
|
||
|
|
"offset": generateExpressionFieldData(schemapb.DataType_Int64, int64(15)),
|
||
|
|
"target": generateExpressionFieldData(schemapb.DataType_String, "def"),
|
||
|
|
}},
|
||
|
|
{`ArrayField + {offset} < {target}`, map[string]*schemapb.TemplateValue{
|
||
|
|
"offset": generateExpressionFieldData(schemapb.DataType_Double, 3.5),
|
||
|
|
"target": generateExpressionFieldData(schemapb.DataType_Int64, int64(5)),
|
||
|
|
}},
|
||
|
|
{`ArrayField[0] + {offset} < {target}`, map[string]*schemapb.TemplateValue{
|
||
|
|
"offset": generateExpressionFieldData(schemapb.DataType_Array, []interface{}{
|
||
|
|
generateExpressionFieldData(schemapb.DataType_Int64, int64(1)),
|
||
|
|
generateExpressionFieldData(schemapb.DataType_Int64, int64(2)),
|
||
|
|
generateExpressionFieldData(schemapb.DataType_Int64, int64(3)),
|
||
|
|
}),
|
||
|
|
"target": generateExpressionFieldData(schemapb.DataType_Int64, int64(5)),
|
||
|
|
}},
|
||
|
|
{`array_length(ArrayField) == {length}`, map[string]*schemapb.TemplateValue{
|
||
|
|
"length": generateExpressionFieldData(schemapb.DataType_String, "abc"),
|
||
|
|
}},
|
||
|
|
}
|
||
|
|
|
||
|
|
schemaH := newTestSchemaHelper(s.T())
|
||
|
|
for _, c := range testcases {
|
||
|
|
s.assertInvalidExpr(schemaH, c.expr, c.values)
|
||
|
|
}
|
||
|
|
})
|
||
|
|
}
|
||
|
|
|
||
|
|
func (s *FillExpressionValueSuite) TestJSONContainsExpression() {
|
||
|
|
s.Run("normal case", func() {
|
||
|
|
testcases := []testcase{
|
||
|
|
{`json_contains(A, 5)`, nil},
|
||
|
|
{`json_contains(A, {age})`, map[string]*schemapb.TemplateValue{
|
||
|
|
"age": generateExpressionFieldData(schemapb.DataType_Int64, int64(18)),
|
||
|
|
}},
|
||
|
|
{`json_contains(A, {str})`, map[string]*schemapb.TemplateValue{
|
||
|
|
"str": generateExpressionFieldData(schemapb.DataType_String, "abc"),
|
||
|
|
}},
|
||
|
|
{`json_contains(A, {bool})`, map[string]*schemapb.TemplateValue{
|
||
|
|
"bool": generateExpressionFieldData(schemapb.DataType_Bool, false),
|
||
|
|
}},
|
||
|
|
{`json_contains_any(JSONField, {array})`, map[string]*schemapb.TemplateValue{
|
||
|
|
"array": generateExpressionFieldData(schemapb.DataType_Array, []interface{}{
|
||
|
|
generateExpressionFieldData(schemapb.DataType_Int64, int64(1)),
|
||
|
|
generateExpressionFieldData(schemapb.DataType_String, "abc"),
|
||
|
|
generateExpressionFieldData(schemapb.DataType_Double, 2.2),
|
||
|
|
generateExpressionFieldData(schemapb.DataType_Bool, false),
|
||
|
|
}),
|
||
|
|
}},
|
||
|
|
{`json_contains_any(JSONField, {array})`, map[string]*schemapb.TemplateValue{
|
||
|
|
"array": generateExpressionFieldData(schemapb.DataType_Array, []interface{}{
|
||
|
|
generateExpressionFieldData(schemapb.DataType_Int64, int64(1)),
|
||
|
|
generateExpressionFieldData(schemapb.DataType_Int64, int64(2)),
|
||
|
|
generateExpressionFieldData(schemapb.DataType_Int64, int64(3)),
|
||
|
|
generateExpressionFieldData(schemapb.DataType_Int64, int64(4)),
|
||
|
|
}),
|
||
|
|
}},
|
||
|
|
{`json_contains_any(JSONField["A"], {array})`, map[string]*schemapb.TemplateValue{
|
||
|
|
"array": generateExpressionFieldData(schemapb.DataType_Array, []interface{}{
|
||
|
|
generateExpressionFieldData(schemapb.DataType_Int64, int64(1)),
|
||
|
|
generateExpressionFieldData(schemapb.DataType_Int64, int64(2)),
|
||
|
|
generateExpressionFieldData(schemapb.DataType_Int64, int64(3)),
|
||
|
|
generateExpressionFieldData(schemapb.DataType_Int64, int64(4)),
|
||
|
|
}),
|
||
|
|
}},
|
||
|
|
{`json_contains_all(JSONField["A"], {array})`, map[string]*schemapb.TemplateValue{
|
||
|
|
"array": generateExpressionFieldData(schemapb.DataType_Array, []interface{}{
|
||
|
|
generateExpressionFieldData(schemapb.DataType_Int64, int64(1)),
|
||
|
|
generateExpressionFieldData(schemapb.DataType_String, "abc"),
|
||
|
|
generateExpressionFieldData(schemapb.DataType_Double, 2.2),
|
||
|
|
generateExpressionFieldData(schemapb.DataType_Bool, false),
|
||
|
|
}),
|
||
|
|
}},
|
||
|
|
{`json_contains_all(JSONField["A"], {array})`, map[string]*schemapb.TemplateValue{
|
||
|
|
"array": generateExpressionFieldData(schemapb.DataType_Array, []interface{}{
|
||
|
|
generateExpressionFieldData(schemapb.DataType_Array, []interface{}{
|
||
|
|
generateExpressionFieldData(schemapb.DataType_Int64, int64(1)),
|
||
|
|
generateExpressionFieldData(schemapb.DataType_Int64, int64(2)),
|
||
|
|
generateExpressionFieldData(schemapb.DataType_Int64, int64(3)),
|
||
|
|
}),
|
||
|
|
generateExpressionFieldData(schemapb.DataType_Array, []interface{}{
|
||
|
|
generateExpressionFieldData(schemapb.DataType_Int64, int64(4)),
|
||
|
|
generateExpressionFieldData(schemapb.DataType_Int64, int64(5)),
|
||
|
|
generateExpressionFieldData(schemapb.DataType_Int64, int64(6)),
|
||
|
|
}),
|
||
|
|
}),
|
||
|
|
}},
|
||
|
|
{`json_contains(ArrayField, {int})`, map[string]*schemapb.TemplateValue{
|
||
|
|
"int": generateExpressionFieldData(schemapb.DataType_Int64, int64(1)),
|
||
|
|
}},
|
||
|
|
{`json_contains_any(ArrayField, {list})`, map[string]*schemapb.TemplateValue{
|
||
|
|
"list": generateExpressionFieldData(schemapb.DataType_Array, []interface{}{
|
||
|
|
generateExpressionFieldData(schemapb.DataType_Int64, int64(1)),
|
||
|
|
generateExpressionFieldData(schemapb.DataType_Int64, int64(2)),
|
||
|
|
generateExpressionFieldData(schemapb.DataType_Int64, int64(3)),
|
||
|
|
generateExpressionFieldData(schemapb.DataType_Int64, int64(4)),
|
||
|
|
}),
|
||
|
|
}},
|
||
|
|
}
|
||
|
|
|
||
|
|
schemaH := newTestSchemaHelper(s.T())
|
||
|
|
|
||
|
|
for _, c := range testcases {
|
||
|
|
s.assertValidExpr(schemaH, c.expr, c.values)
|
||
|
|
}
|
||
|
|
})
|
||
|
|
|
||
|
|
s.Run("failed case", func() {
|
||
|
|
testcases := []testcase{
|
||
|
|
{`json_contains(ArrayField[0], {str})`, map[string]*schemapb.TemplateValue{
|
||
|
|
"str": generateExpressionFieldData(schemapb.DataType_String, "abc"),
|
||
|
|
}},
|
||
|
|
{`json_contains_any(JSONField, {not_array})`, map[string]*schemapb.TemplateValue{
|
||
|
|
"not_array": generateExpressionFieldData(schemapb.DataType_Int64, int64(1)),
|
||
|
|
}},
|
||
|
|
{`json_contains_all(JSONField, {not_array})`, map[string]*schemapb.TemplateValue{
|
||
|
|
"not_array": generateExpressionFieldData(schemapb.DataType_Int64, int64(1)),
|
||
|
|
}},
|
||
|
|
{`json_contains_all(JSONField, {not_array})`, map[string]*schemapb.TemplateValue{
|
||
|
|
"array": generateExpressionFieldData(schemapb.DataType_Array, []interface{}{
|
||
|
|
generateExpressionFieldData(schemapb.DataType_Int64, int64(1)),
|
||
|
|
generateExpressionFieldData(schemapb.DataType_Int64, int64(2)),
|
||
|
|
generateExpressionFieldData(schemapb.DataType_Int64, int64(3)),
|
||
|
|
}),
|
||
|
|
}},
|
||
|
|
{`json_contains_all(ArrayField, {array})`, map[string]*schemapb.TemplateValue{
|
||
|
|
"array": generateExpressionFieldData(schemapb.DataType_Array, []interface{}{
|
||
|
|
generateExpressionFieldData(schemapb.DataType_Int64, int64(1)),
|
||
|
|
generateExpressionFieldData(schemapb.DataType_String, "abc"),
|
||
|
|
generateExpressionFieldData(schemapb.DataType_Double, 2.2),
|
||
|
|
generateExpressionFieldData(schemapb.DataType_Bool, false),
|
||
|
|
}),
|
||
|
|
}},
|
||
|
|
{`json_contains(ArrayField, {array})`, map[string]*schemapb.TemplateValue{
|
||
|
|
"array": generateExpressionFieldData(schemapb.DataType_Array, []interface{}{
|
||
|
|
generateExpressionFieldData(schemapb.DataType_Int64, int64(1)),
|
||
|
|
generateExpressionFieldData(schemapb.DataType_Int64, int64(2)),
|
||
|
|
generateExpressionFieldData(schemapb.DataType_Int64, int64(3)),
|
||
|
|
}),
|
||
|
|
}},
|
||
|
|
{`json_contains_any(ArrayField, {array})`, map[string]*schemapb.TemplateValue{
|
||
|
|
"array": generateExpressionFieldData(schemapb.DataType_Array, []interface{}{
|
||
|
|
generateExpressionFieldData(schemapb.DataType_Int64, int64(1)),
|
||
|
|
generateExpressionFieldData(schemapb.DataType_String, "abc"),
|
||
|
|
generateExpressionFieldData(schemapb.DataType_Double, 2.2),
|
||
|
|
generateExpressionFieldData(schemapb.DataType_Bool, false),
|
||
|
|
}),
|
||
|
|
}},
|
||
|
|
}
|
||
|
|
|
||
|
|
schemaH := newTestSchemaHelper(s.T())
|
||
|
|
|
||
|
|
for _, c := range testcases {
|
||
|
|
s.assertInvalidExpr(schemaH, c.expr, c.values)
|
||
|
|
}
|
||
|
|
})
|
||
|
|
}
|
||
|
|
|
||
|
|
func (s *FillExpressionValueSuite) TestBinaryExpression() {
|
||
|
|
s.Run("normal case", func() {
|
||
|
|
testcases := []testcase{
|
||
|
|
{`Int64Field > {int} && StringField in {list}`, map[string]*schemapb.TemplateValue{
|
||
|
|
"int": generateExpressionFieldData(schemapb.DataType_Int64, int64(10)),
|
||
|
|
"list": generateExpressionFieldData(schemapb.DataType_Array, []interface{}{
|
||
|
|
generateExpressionFieldData(schemapb.DataType_VarChar, "abc"),
|
||
|
|
generateExpressionFieldData(schemapb.DataType_VarChar, "def"),
|
||
|
|
generateExpressionFieldData(schemapb.DataType_VarChar, "ghi"),
|
||
|
|
}),
|
||
|
|
}},
|
||
|
|
{`{max} > FloatField >= {min} or BoolField == {bool}`, map[string]*schemapb.TemplateValue{
|
||
|
|
"min": generateExpressionFieldData(schemapb.DataType_Int64, int64(10)),
|
||
|
|
"max": generateExpressionFieldData(schemapb.DataType_Float, 22.22),
|
||
|
|
"bool": generateExpressionFieldData(schemapb.DataType_Bool, true),
|
||
|
|
}},
|
||
|
|
}
|
||
|
|
|
||
|
|
schemaH := newTestSchemaHelper(s.T())
|
||
|
|
|
||
|
|
for _, c := range testcases {
|
||
|
|
s.assertValidExpr(schemaH, c.expr, c.values)
|
||
|
|
}
|
||
|
|
})
|
||
|
|
|
||
|
|
s.Run("failed case", func() {
|
||
|
|
testcases := []testcase{
|
||
|
|
{`Int64Field > {int} && StringField in {list}`, map[string]*schemapb.TemplateValue{
|
||
|
|
"int": generateExpressionFieldData(schemapb.DataType_String, "abc"),
|
||
|
|
"list": generateExpressionFieldData(schemapb.DataType_Array, []interface{}{
|
||
|
|
generateExpressionFieldData(schemapb.DataType_VarChar, "abc"),
|
||
|
|
generateExpressionFieldData(schemapb.DataType_Int64, int64(10)),
|
||
|
|
generateExpressionFieldData(schemapb.DataType_VarChar, "ghi"),
|
||
|
|
}),
|
||
|
|
}},
|
||
|
|
{`{max} > FloatField >= {min} or BoolField == {bool}`, map[string]*schemapb.TemplateValue{
|
||
|
|
"min": generateExpressionFieldData(schemapb.DataType_Int64, int64(10)),
|
||
|
|
"bool": generateExpressionFieldData(schemapb.DataType_Bool, true),
|
||
|
|
}},
|
||
|
|
}
|
||
|
|
|
||
|
|
schemaH := newTestSchemaHelper(s.T())
|
||
|
|
|
||
|
|
for _, c := range testcases {
|
||
|
|
s.assertInvalidExpr(schemaH, c.expr, c.values)
|
||
|
|
}
|
||
|
|
})
|
||
|
|
}
|