mirror of
https://github.com/chaisql/chai.git
synced 2025-10-28 17:51:48 +08:00
f966172cee
This replaces the Value struct by an interface to allow us to override some values behavior in the future. It also introduces a new package types, which contains type definitions, comparison, and arithmetics. Concerning encoding, Genji now only uses on type of encoding for values. This simplifies indexing logic as well as table access in general.
125 lines
3.1 KiB
Go
125 lines
3.1 KiB
Go
package expr_test
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
"testing"
|
|
|
|
"github.com/genjidb/genji/document"
|
|
"github.com/genjidb/genji/internal/environment"
|
|
"github.com/genjidb/genji/internal/expr"
|
|
"github.com/genjidb/genji/internal/sql/parser"
|
|
"github.com/genjidb/genji/internal/testutil"
|
|
"github.com/genjidb/genji/types"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestPathExpr(t *testing.T) {
|
|
tests := []struct {
|
|
expr string
|
|
res types.Value
|
|
fails bool
|
|
}{
|
|
{"a", types.NewIntegerValue(1), false},
|
|
{"b", func() types.Value {
|
|
fb := document.NewFieldBuffer()
|
|
err := json.Unmarshal([]byte(`{"foo bar": [1, 2]}`), fb)
|
|
require.NoError(t, err)
|
|
return types.NewDocumentValue(fb)
|
|
}(),
|
|
false},
|
|
{"b.`foo bar`[0]", types.NewIntegerValue(1), false},
|
|
{"_v.b.`foo bar`[0]", types.NewNullValue(), false},
|
|
{"b.`foo bar`[1]", types.NewIntegerValue(2), false},
|
|
{"b.`foo bar`[2]", nullLiteral, false},
|
|
{"b[0]", nullLiteral, false},
|
|
{"c[0]", types.NewIntegerValue(1), false},
|
|
{"c[1].foo", types.NewTextValue("bar"), false},
|
|
{"c.foo", nullLiteral, false},
|
|
{"d", nullLiteral, false},
|
|
}
|
|
|
|
d := document.NewFromJSON([]byte(`{
|
|
"a": 1,
|
|
"b": {"foo bar": [1, 2]},
|
|
"c": [1, {"foo": "bar"}, [1, 2]]
|
|
}`))
|
|
|
|
for _, test := range tests {
|
|
t.Run(test.expr, func(t *testing.T) {
|
|
testutil.TestExpr(t, test.expr, environment.New(d), test.res, test.fails)
|
|
})
|
|
}
|
|
|
|
t.Run("empty env", func(t *testing.T) {
|
|
testutil.TestExpr(t, "a", &environment.Environment{}, nullLiteral, true)
|
|
})
|
|
}
|
|
|
|
func TestPathIsEqual(t *testing.T) {
|
|
tests := []struct {
|
|
a, b string
|
|
isEqual bool
|
|
}{
|
|
{`a`, `a`, true},
|
|
{`a[0].b`, `a[0].b`, true},
|
|
{`a[0].b`, `a[1].b`, false},
|
|
}
|
|
|
|
for _, test := range tests {
|
|
t.Run(fmt.Sprintf("%s = %s", test.a, test.b), func(t *testing.T) {
|
|
pa, err := parser.ParsePath(test.a)
|
|
require.NoError(t, err)
|
|
ea := expr.Path(pa)
|
|
|
|
pb, err := parser.ParsePath(test.b)
|
|
require.NoError(t, err)
|
|
eb := expr.Path(pb)
|
|
|
|
require.Equal(t, test.isEqual, ea.IsEqual(eb))
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestEnvPathExpr(t *testing.T) {
|
|
tests := []struct {
|
|
expr string
|
|
res types.Value
|
|
fails bool
|
|
}{
|
|
{"a", types.NewIntegerValue(1), false},
|
|
{"b", func() types.Value {
|
|
fb := document.NewFieldBuffer()
|
|
err := json.Unmarshal([]byte(`{"foo bar": [1, 2]}`), fb)
|
|
require.NoError(t, err)
|
|
return types.NewDocumentValue(fb)
|
|
}(),
|
|
false},
|
|
{"b.`foo bar`[0]", types.NewIntegerValue(1), false},
|
|
{"_v.b.`foo bar`[0]", types.NewNullValue(), false},
|
|
{"b.`foo bar`[1]", types.NewIntegerValue(2), false},
|
|
{"b.`foo bar`[2]", nullLiteral, false},
|
|
{"b[0]", nullLiteral, false},
|
|
{"c[0]", types.NewIntegerValue(1), false},
|
|
{"c[1].foo", types.NewTextValue("bar"), false},
|
|
{"c.foo", nullLiteral, false},
|
|
{"d", nullLiteral, false},
|
|
}
|
|
|
|
d := document.NewFromJSON([]byte(`{
|
|
"a": 1,
|
|
"b": {"foo bar": [1, 2]},
|
|
"c": [1, {"foo": "bar"}, [1, 2]]
|
|
}`))
|
|
|
|
for _, test := range tests {
|
|
t.Run(test.expr, func(t *testing.T) {
|
|
testutil.TestExpr(t, test.expr, environment.New(d), test.res, test.fails)
|
|
})
|
|
}
|
|
|
|
t.Run("empty env", func(t *testing.T) {
|
|
testutil.TestExpr(t, "a", &environment.Environment{}, nullLiteral, true)
|
|
})
|
|
}
|