noth/surrealpatch
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Update SQL package tests

Browse source
This commit is contained in:
Tobie Morgan Hitchcock 2016-10-29 18:47:09 +01:00
parent d4491d557b
commit 872f4abf43
1 changed files with 253 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

255
sql/sql_test.go
View file

@ -140,11 +140,11 @@ func Test_Parse_Queries_Malformed(t *testing.T) {
},
{
sql: `!`,
err: "Found `!` but expected `USE, INFO, LET, BEGIN, CANCEL, COMMIT, ROLLBACK, SELECT, CREATE, UPDATE, INSERT, UPSERT, MODIFY, DELETE, RELATE, DEFINE, REMOVE`",
err: "Found `!` but expected `USE, INFO, LET, BEGIN, CANCEL, COMMIT, ROLLBACK, RETURN, SELECT, CREATE, UPDATE, INSERT, UPSERT, MODIFY, DELETE, RELATE, DEFINE, REMOVE`",
},
{
sql: `SELECT * FROM person;;;`,
err: "Found `;` but expected `USE, INFO, LET, BEGIN, CANCEL, COMMIT, ROLLBACK, SELECT, CREATE, UPDATE, INSERT, UPSERT, MODIFY, DELETE, RELATE, DEFINE, REMOVE`",
err: "Found `;` but expected `USE, INFO, LET, BEGIN, CANCEL, COMMIT, ROLLBACK, RETURN, SELECT, CREATE, UPDATE, INSERT, UPSERT, MODIFY, DELETE, RELATE, DEFINE, REMOVE`",
},
}
@ -249,6 +249,206 @@ func Test_Parse_Queries_Use(t *testing.T) {
}
func Test_Parse_Queries_Let(t *testing.T) {
var tests = []tester{
{
sql: `LET`,
err: "Found `` but expected `IDENT`",
},
{
sql: `LET name`,
err: "Found `` but expected `=`",
},
{
sql: `LET name =`,
err: "Found `=` but expected `NULL, NOW, DATE, TIME, TRUE, FALSE, STRING, NUMBER, DOUBLE, THING, JSON, ARRAY, PARAM`",
},
{
sql: `LET name = true`,
res: &Query{Statements: []Statement{&LetStatement{
Name: "name",
Expr: true,
}}},
},
{
sql: `LET name = false`,
res: &Query{Statements: []Statement{&LetStatement{
Name: "name",
Expr: false,
}}},
},
{
sql: `LET name = "test"`,
res: &Query{Statements: []Statement{&LetStatement{
Name: "name",
Expr: "test",
}}},
},
{
sql: `LET name = 1`,
res: &Query{Statements: []Statement{&LetStatement{
Name: "name",
Expr: int64(1),
}}},
},
{
sql: `LET name = 1.0`,
res: &Query{Statements: []Statement{&LetStatement{
Name: "name",
Expr: float64(1),
}}},
},
{
sql: `LET name = 1.1`,
res: &Query{Statements: []Statement{&LetStatement{
Name: "name",
Expr: float64(1.1),
}}},
},
{
sql: `LET name = @thing:test`,
res: &Query{Statements: []Statement{&LetStatement{
Name: "name",
Expr: &Thing{TB: "thing", ID: "test"},
}}},
},
{
sql: `LET name = {"key": "val"}`,
res: &Query{Statements: []Statement{&LetStatement{
Name: "name",
Expr: map[string]interface{}{"key": "val"},
}}},
},
{
sql: `LET name = ["key", "val"]`,
res: &Query{Statements: []Statement{&LetStatement{
Name: "name",
Expr: []interface{}{"key", "val"},
}}},
},
{
sql: `LET name = $test`,
res: &Query{Statements: []Statement{&LetStatement{
Name: "name",
Expr: &Param{ID: "test"},
}}},
},
{
sql: `LET name = {"key"::"val"}`,
err: "Invalid JSON: {\"key\"::\"val\"}",
},
{
sql: `LET name = "test" something`,
err: "Found `something` but expected `EOF, ;`",
},
}
for _, test := range tests {
testsql(t, test)
}
}
func Test_Parse_Queries_Return(t *testing.T) {
var tests = []tester{
{
sql: `RETURN`,
err: "Found `` but expected `NULL, NOW, DATE, TIME, TRUE, FALSE, STRING, NUMBER, DOUBLE, THING, JSON, ARRAY, PARAM`",
},
{
sql: `RETURN true`,
res: &Query{Statements: []Statement{&ReturnStatement{
What: []Expr{true},
}}},
},
{
sql: `RETURN true`,
res: &Query{Statements: []Statement{&ReturnStatement{
What: []Expr{true},
}}},
},
{
sql: `RETURN false`,
res: &Query{Statements: []Statement{&ReturnStatement{
What: []Expr{false},
}}},
},
{
sql: `RETURN "test"`,
res: &Query{Statements: []Statement{&ReturnStatement{
What: []Expr{"test"},
}}},
},
{
sql: `RETURN 1`,
res: &Query{Statements: []Statement{&ReturnStatement{
What: []Expr{int64(1)},
}}},
},
{
sql: `RETURN 1.0`,
res: &Query{Statements: []Statement{&ReturnStatement{
What: []Expr{float64(1)},
}}},
},
{
sql: `RETURN 1.1`,
res: &Query{Statements: []Statement{&ReturnStatement{
What: []Expr{float64(1.1)},
}}},
},
{
sql: `RETURN @thing:test`,
res: &Query{Statements: []Statement{&ReturnStatement{
What: []Expr{&Thing{TB: "thing", ID: "test"}},
}}},
},
{
sql: `RETURN {"key": "val"}`,
res: &Query{Statements: []Statement{&ReturnStatement{
What: []Expr{map[string]interface{}{"key": "val"}},
}}},
},
{
sql: `RETURN ["key", "val"]`,
res: &Query{Statements: []Statement{&ReturnStatement{
What: []Expr{[]interface{}{"key", "val"}},
}}},
},
{
sql: `RETURN $test`,
res: &Query{Statements: []Statement{&ReturnStatement{
What: []Expr{&Param{ID: "test"}},
}}},
},
{
sql: `RETURN $test, $test, $test`,
res: &Query{Statements: []Statement{&ReturnStatement{
What: []Expr{
&Param{ID: "test"},
&Param{ID: "test"},
&Param{ID: "test"},
},
}}},
},
{
sql: `RETURN {"key"::"val"}`,
err: "Invalid JSON: {\"key\"::\"val\"}",
},
{
sql: `RETURN $test something`,
err: "Found `something` but expected `EOF, ;`",
},
}
for _, test := range tests {
testsql(t, test)
}
}
func Test_Parse_Queries_Select(t *testing.T) {
date, _ := time.Parse("2006-01-02", "1987-06-22")
@ -757,6 +957,38 @@ func Test_Parse_Queries_Select(t *testing.T) {
Cond: []Expr{&BinaryExpression{LHS: &Ident{"test"}, Op: INS, RHS: []interface{}{"London", "Paris"}}},
}}},
},
{
sql: `SELECT * FROM person WHERE test IS IN ["London","Paris"]`,
res: &Query{Statements: []Statement{&SelectStatement{
Expr: []*Field{{Expr: &All{}, Alias: "*"}},
What: []Expr{&Table{"person"}},
Cond: []Expr{&BinaryExpression{LHS: &Ident{"test"}, Op: INS, RHS: []interface{}{"London", "Paris"}}},
}}},
},
{
sql: `SELECT * FROM person WHERE test IS NOT IN ["London","Paris"]`,
res: &Query{Statements: []Statement{&SelectStatement{
Expr: []*Field{{Expr: &All{}, Alias: "*"}},
What: []Expr{&Table{"person"}},
Cond: []Expr{&BinaryExpression{LHS: &Ident{"test"}, Op: NIS, RHS: []interface{}{"London", "Paris"}}},
}}},
},
{
sql: `SELECT * FROM person WHERE ["London","Paris"] CONTAINS test`,
res: &Query{Statements: []Statement{&SelectStatement{
Expr: []*Field{{Expr: &All{}, Alias: "*"}},
What: []Expr{&Table{"person"}},
Cond: []Expr{&BinaryExpression{LHS: []interface{}{"London", "Paris"}, Op: SIN, RHS: &Ident{"test"}}},
}}},
},
{
sql: `SELECT * FROM person WHERE ["London","Paris"] CONTAINS NOT test`,
res: &Query{Statements: []Statement{&SelectStatement{
Expr: []*Field{{Expr: &All{}, Alias: "*"}},
What: []Expr{&Table{"person"}},
Cond: []Expr{&BinaryExpression{LHS: []interface{}{"London", "Paris"}, Op: SNI, RHS: &Ident{"test"}}},
}}},
},
{
sql: `SELECT * FROM person WHERE test = {`,
err: "Found `{` but expected `field value`",
@ -765,6 +997,10 @@ func Test_Parse_Queries_Select(t *testing.T) {
sql: `SELECT * FROM person WHERE test = {"name","London"}`,
err: `Invalid JSON: {"name","London"}`,
},
{
sql: "SELECT * FROM person WHERE {\"name\":\"\x0A\"} = test",
err: "Invalid JSON: {\"name\":\"\n\"}",
},
{
sql: "SELECT * FROM person WHERE test = {\"name\":\"\x0A\"}",
err: "Invalid JSON: {\"name\":\"\n\"}",
@ -1993,6 +2229,21 @@ func Test_Parse_Queries_Remove(t *testing.T) {
sql: `REMOVE INDEX temp ON person something`,
err: "Found `something` but expected `EOF, ;`",
},
// ----------------------------------------------------------------------
{
sql: `REMOVE VIEW`,
err: "Found `` but expected `name`",
},
{
sql: `REMOVE VIEW temp`,
res: &Query{Statements: []Statement{&RemoveViewStatement{
Name: "temp",
}}},
},
{
sql: `REMOVE VIEW temp something`,
err: "Found `something` but expected `EOF, ;`",
},
}
for _, test := range tests {