mirror of
https://github.com/chaisql/chai.git
synced 2025-10-05 07:36:56 +08:00
9918cd6f55
Previously, expressions and params were evaluated during the planning phase. This change builds the query plan without evaluating params and expressions which are then evaluated only during the execution phase.
84 lines
2.7 KiB
Go
84 lines
2.7 KiB
Go
package parser_test
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/genjidb/genji/document"
|
|
"github.com/genjidb/genji/internal/query/statement"
|
|
"github.com/genjidb/genji/internal/sql/parser"
|
|
"github.com/genjidb/genji/internal/stream"
|
|
"github.com/genjidb/genji/internal/testutil"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestParserUpdate(t *testing.T) {
|
|
tests := []struct {
|
|
name string
|
|
s string
|
|
expected *stream.Stream
|
|
errored bool
|
|
}{
|
|
{"SET/No cond", "UPDATE test SET a = 1",
|
|
stream.New(stream.SeqScan("test")).
|
|
Pipe(stream.Set(document.Path(testutil.ParsePath(t, "a")), testutil.IntegerValue(1))).
|
|
Pipe(stream.TableReplace("test")),
|
|
false,
|
|
},
|
|
{"SET/With cond", "UPDATE test SET a = 1, b = 2 WHERE age = 10",
|
|
stream.New(stream.SeqScan("test")).
|
|
Pipe(stream.Filter(parser.MustParseExpr("age = 10"))).
|
|
Pipe(stream.Set(document.Path(testutil.ParsePath(t, "a")), testutil.IntegerValue(1))).
|
|
Pipe(stream.Set(document.Path(testutil.ParsePath(t, "b")), parser.MustParseExpr("2"))).
|
|
Pipe(stream.TableReplace("test")),
|
|
false,
|
|
},
|
|
{"SET/No cond path with backquotes", "UPDATE test SET ` some \"path\" ` = 1",
|
|
stream.New(stream.SeqScan("test")).
|
|
Pipe(stream.Set(document.Path(testutil.ParsePath(t, "` some \"path\" `")), testutil.IntegerValue(1))).
|
|
Pipe(stream.TableReplace("test")),
|
|
false,
|
|
},
|
|
{"SET/No cond nested path", "UPDATE test SET a.b = 1",
|
|
stream.New(stream.SeqScan("test")).
|
|
Pipe(stream.Set(document.Path(testutil.ParsePath(t, "a.b")), testutil.IntegerValue(1))).
|
|
Pipe(stream.TableReplace("test")),
|
|
false,
|
|
},
|
|
{"UNSET/No cond", "UPDATE test UNSET a",
|
|
stream.New(stream.SeqScan("test")).
|
|
Pipe(stream.Unset("a")).
|
|
Pipe(stream.TableReplace("test")),
|
|
false,
|
|
},
|
|
{"UNSET/With cond", "UPDATE test UNSET a, b WHERE age = 10",
|
|
stream.New(stream.SeqScan("test")).
|
|
Pipe(stream.Filter(parser.MustParseExpr("age = 10"))).
|
|
Pipe(stream.Unset("a")).
|
|
Pipe(stream.Unset("b")).
|
|
Pipe(stream.TableReplace("test")),
|
|
false,
|
|
},
|
|
{"Trailing comma", "UPDATE test SET a = 1, WHERE age = 10", nil, true},
|
|
{"No SET", "UPDATE test WHERE age = 10", nil, true},
|
|
{"No pair", "UPDATE test SET WHERE age = 10", nil, true},
|
|
{"query.Field only", "UPDATE test SET a WHERE age = 10", nil, true},
|
|
{"No value", "UPDATE test SET a = WHERE age = 10", nil, true},
|
|
}
|
|
|
|
for _, test := range tests {
|
|
t.Run(test.name, func(t *testing.T) {
|
|
q, err := parser.ParseQuery(test.s)
|
|
if test.errored {
|
|
require.Error(t, err)
|
|
return
|
|
}
|
|
|
|
require.NoError(t, err)
|
|
require.Len(t, q.Statements, 1)
|
|
stmt := q.Statements[0].(*statement.StreamStmt)
|
|
require.False(t, stmt.IsReadOnly())
|
|
require.EqualValues(t, test.expected, q.Statements[0].(*statement.StreamStmt).Stream)
|
|
})
|
|
}
|
|
}
|