You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
934 lines
30 KiB
Go
934 lines
30 KiB
Go
package db
|
|
|
|
import (
|
|
"database/sql/driver"
|
|
"encoding/json"
|
|
"fmt"
|
|
"io/ioutil"
|
|
"os"
|
|
"path"
|
|
"strings"
|
|
"testing"
|
|
|
|
"github.com/rqlite/rqlite/testdata/chinook"
|
|
)
|
|
|
|
/*
|
|
* Lowest-layer database tests
|
|
*/
|
|
|
|
func Test_DbFileCreation(t *testing.T) {
|
|
dir, err := ioutil.TempDir("", "rqlite-test-")
|
|
defer os.RemoveAll(dir)
|
|
|
|
db, err := Open(path.Join(dir, "test_db"))
|
|
if err != nil {
|
|
t.Fatalf("failed to open new database: %s", err.Error())
|
|
}
|
|
if db == nil {
|
|
t.Fatal("database is nil")
|
|
}
|
|
err = db.Close()
|
|
if err != nil {
|
|
t.Fatalf("failed to close database: %s", err.Error())
|
|
}
|
|
}
|
|
|
|
func Test_TableCreation(t *testing.T) {
|
|
db, path := mustCreateDatabase()
|
|
defer db.Close()
|
|
defer os.Remove(path)
|
|
|
|
_, err := db.ExecuteStringStmt("CREATE TABLE foo (id INTEGER NOT NULL PRIMARY KEY, name TEXT)")
|
|
if err != nil {
|
|
t.Fatalf("failed to create table: %s", err.Error())
|
|
}
|
|
|
|
r, err := db.QueryStringStmt("SELECT * FROM foo")
|
|
if err != nil {
|
|
t.Fatalf("failed to query empty table: %s", err.Error())
|
|
}
|
|
if exp, got := `[{"columns":["id","name"],"types":["integer","text"]}]`, asJSON(r); exp != got {
|
|
t.Fatalf("unexpected results for query, expected %s, got %s", exp, got)
|
|
}
|
|
}
|
|
|
|
func Test_SQLiteMasterTable(t *testing.T) {
|
|
db, path := mustCreateDatabase()
|
|
defer db.Close()
|
|
defer os.Remove(path)
|
|
|
|
_, err := db.ExecuteStringStmt("CREATE TABLE foo (id INTEGER NOT NULL PRIMARY KEY, name TEXT)")
|
|
if err != nil {
|
|
t.Fatalf("failed to create table: %s", err.Error())
|
|
}
|
|
|
|
r, err := db.QueryStringStmt("SELECT * FROM sqlite_master")
|
|
if err != nil {
|
|
t.Fatalf("failed to query master table: %s", err.Error())
|
|
}
|
|
if exp, got := `[{"columns":["type","name","tbl_name","rootpage","sql"],"types":["text","text","text","int","text"],"values":[["table","foo","foo",2,"CREATE TABLE foo (id INTEGER NOT NULL PRIMARY KEY, name TEXT)"]]}]`, asJSON(r); exp != got {
|
|
t.Fatalf("unexpected results for query, expected %s, got %s", exp, got)
|
|
}
|
|
}
|
|
|
|
func Test_LoadInMemory(t *testing.T) {
|
|
db, path := mustCreateDatabase()
|
|
defer db.Close()
|
|
defer os.Remove(path)
|
|
|
|
_, err := db.ExecuteStringStmt("CREATE TABLE foo (id INTEGER NOT NULL PRIMARY KEY, name TEXT)")
|
|
if err != nil {
|
|
t.Fatalf("failed to create table: %s", err.Error())
|
|
}
|
|
|
|
r, err := db.QueryStringStmt("SELECT * FROM foo")
|
|
if err != nil {
|
|
t.Fatalf("failed to query empty table: %s", err.Error())
|
|
}
|
|
if exp, got := `[{"columns":["id","name"],"types":["integer","text"]}]`, asJSON(r); exp != got {
|
|
t.Fatalf("unexpected results for query, expected %s, got %s", exp, got)
|
|
}
|
|
|
|
inmem, err := LoadInMemoryWithDSN(path, "")
|
|
if err != nil {
|
|
t.Fatalf("failed to create loaded in-memory database: %s", err.Error())
|
|
}
|
|
|
|
// Ensure it has been loaded correctly into the database
|
|
r, err = inmem.QueryStringStmt("SELECT * FROM foo")
|
|
if err != nil {
|
|
t.Fatalf("failed to query empty table: %s", err.Error())
|
|
}
|
|
if exp, got := `[{"columns":["id","name"],"types":["integer","text"]}]`, asJSON(r); exp != got {
|
|
t.Fatalf("unexpected results for query, expected %s, got %s", exp, got)
|
|
}
|
|
}
|
|
|
|
func Test_EmptyStatements(t *testing.T) {
|
|
db, path := mustCreateDatabase()
|
|
defer db.Close()
|
|
defer os.Remove(path)
|
|
|
|
_, err := db.Execute([]Statement{{"", nil}}, false, false)
|
|
if err != nil {
|
|
t.Fatalf("failed to execute empty statement: %s", err.Error())
|
|
}
|
|
_, err = db.Execute([]Statement{{";", nil}}, false, false)
|
|
if err != nil {
|
|
t.Fatalf("failed to execute empty statement with semicolon: %s", err.Error())
|
|
}
|
|
}
|
|
|
|
func Test_SimpleSingleStatements(t *testing.T) {
|
|
db, path := mustCreateDatabase()
|
|
defer db.Close()
|
|
defer os.Remove(path)
|
|
|
|
_, err := db.ExecuteStringStmt("CREATE TABLE foo (id INTEGER NOT NULL PRIMARY KEY, name TEXT)")
|
|
if err != nil {
|
|
t.Fatalf("failed to create table: %s", err.Error())
|
|
}
|
|
|
|
_, err = db.ExecuteStringStmt(`INSERT INTO foo(name) VALUES("fiona")`)
|
|
if err != nil {
|
|
t.Fatalf("failed to insert record: %s", err.Error())
|
|
}
|
|
|
|
_, err = db.ExecuteStringStmt(`INSERT INTO foo(name) VALUES("aoife")`)
|
|
if err != nil {
|
|
t.Fatalf("failed to insert record: %s", err.Error())
|
|
}
|
|
|
|
r, err := db.QueryStringStmt(`SELECT * FROM foo`)
|
|
if err != nil {
|
|
t.Fatalf("failed to query table: %s", err.Error())
|
|
}
|
|
if exp, got := `[{"columns":["id","name"],"types":["integer","text"],"values":[[1,"fiona"],[2,"aoife"]]}]`, asJSON(r); exp != got {
|
|
t.Fatalf("unexpected results for query\nexp: %s\ngot: %s", exp, got)
|
|
}
|
|
|
|
r, err = db.QueryStringStmt(`SELECT * FROM foo WHERE name="aoife"`)
|
|
if err != nil {
|
|
t.Fatalf("failed to query table: %s", err.Error())
|
|
}
|
|
if exp, got := `[{"columns":["id","name"],"types":["integer","text"],"values":[[2,"aoife"]]}]`, asJSON(r); exp != got {
|
|
t.Fatalf("unexpected results for query\nexp: %s\ngot: %s", exp, got)
|
|
}
|
|
|
|
r, err = db.QueryStringStmt(`SELECT * FROM foo WHERE name="dana"`)
|
|
if err != nil {
|
|
t.Fatalf("failed to query table: %s", err.Error())
|
|
}
|
|
if exp, got := `[{"columns":["id","name"],"types":["integer","text"]}]`, asJSON(r); exp != got {
|
|
t.Fatalf("unexpected results for query\nexp: %s\ngot: %s", exp, got)
|
|
}
|
|
|
|
r, err = db.QueryStringStmt(`SELECT * FROM foo ORDER BY name`)
|
|
if err != nil {
|
|
t.Fatalf("failed to query table: %s", err.Error())
|
|
}
|
|
if exp, got := `[{"columns":["id","name"],"types":["integer","text"],"values":[[2,"aoife"],[1,"fiona"]]}]`, asJSON(r); exp != got {
|
|
t.Fatalf("unexpected results for query\nexp: %s\ngot: %s", exp, got)
|
|
}
|
|
|
|
r, err = db.QueryStringStmt(`SELECT *,name FROM foo`)
|
|
if err != nil {
|
|
t.Fatalf("failed to query table: %s", err.Error())
|
|
}
|
|
if exp, got := `[{"columns":["id","name","name"],"types":["integer","text","text"],"values":[[1,"fiona","fiona"],[2,"aoife","aoife"]]}]`, asJSON(r); exp != got {
|
|
t.Fatalf("unexpected results for query\nexp: %s\ngot: %s", exp, got)
|
|
}
|
|
}
|
|
|
|
func Test_SimpleSingleJSONStatements(t *testing.T) {
|
|
db, path := mustCreateDatabase()
|
|
defer db.Close()
|
|
defer os.Remove(path)
|
|
|
|
_, err := db.ExecuteStringStmt("CREATE TABLE foo (c0 VARCHAR(36), c1 JSON, c2 NCHAR, c3 NVARCHAR, c4 CLOB)")
|
|
if err != nil {
|
|
t.Fatalf("failed to create table: %s", err.Error())
|
|
}
|
|
|
|
_, err = db.ExecuteStringStmt(`INSERT INTO foo(c0, c1, c2, c3, c4) VALUES("fiona", '{"mittens": "foobar"}', "bob", "dana", "declan")`)
|
|
if err != nil {
|
|
t.Fatalf("failed to insert record: %s", err.Error())
|
|
}
|
|
|
|
r, err := db.QueryStringStmt("SELECT * FROM foo")
|
|
if err != nil {
|
|
t.Fatalf("failed to query: %s", err.Error())
|
|
}
|
|
if exp, got := `[{"columns":["c0","c1","c2","c3","c4"],"types":["varchar(36)","json","nchar","nvarchar","clob"],"values":[["fiona","{\"mittens\": \"foobar\"}","bob","dana","declan"]]}]`, asJSON(r); exp != got {
|
|
t.Fatalf("unexpected results for query, expected %s, got %s", exp, got)
|
|
}
|
|
}
|
|
|
|
func Test_SimpleJoinStatements(t *testing.T) {
|
|
db, path := mustCreateDatabase()
|
|
defer db.Close()
|
|
defer os.Remove(path)
|
|
|
|
_, err := db.ExecuteStringStmt("CREATE TABLE names (id INTEGER NOT NULL PRIMARY KEY, name TEXT, ssn TEXT)")
|
|
if err != nil {
|
|
t.Fatalf("failed to create table: %s", err.Error())
|
|
}
|
|
|
|
_, err = db.Execute([]Statement{
|
|
{`INSERT INTO "names" VALUES(1,'bob','123-45-678')`, nil},
|
|
{`INSERT INTO "names" VALUES(2,'tom','111-22-333')`, nil},
|
|
{`INSERT INTO "names" VALUES(3,'matt','222-22-333')`, nil},
|
|
}, false, false)
|
|
if err != nil {
|
|
t.Fatalf("failed to insert record: %s", err.Error())
|
|
}
|
|
|
|
_, err = db.ExecuteStringStmt("CREATE TABLE staff (id INTEGER NOT NULL PRIMARY KEY, employer TEXT, ssn TEXT)")
|
|
if err != nil {
|
|
t.Fatalf("failed to create table: %s", err.Error())
|
|
}
|
|
|
|
_, err = db.ExecuteStringStmt(`INSERT INTO "staff" VALUES(1,'acme','222-22-333')`)
|
|
if err != nil {
|
|
t.Fatalf("failed to insert record: %s", err.Error())
|
|
}
|
|
|
|
r, err := db.QueryStringStmt(`SELECT names.id,name,names.ssn,employer FROM names INNER JOIN staff ON staff.ssn = names.ssn`)
|
|
if err != nil {
|
|
t.Fatalf("failed to query table using JOIN: %s", err.Error())
|
|
}
|
|
if exp, got := `[{"columns":["id","name","ssn","employer"],"types":["integer","text","text","text"],"values":[[3,"matt","222-22-333","acme"]]}]`, asJSON(r); exp != got {
|
|
t.Fatalf("unexpected results for query\nexp: %s\ngot: %s", exp, got)
|
|
}
|
|
}
|
|
|
|
func Test_SimpleSingleConcatStatements(t *testing.T) {
|
|
db, path := mustCreateDatabase()
|
|
defer db.Close()
|
|
defer os.Remove(path)
|
|
|
|
_, err := db.ExecuteStringStmt("CREATE TABLE foo (id INTEGER NOT NULL PRIMARY KEY, name TEXT)")
|
|
if err != nil {
|
|
t.Fatalf("failed to create table: %s", err.Error())
|
|
}
|
|
|
|
_, err = db.ExecuteStringStmt(`INSERT INTO foo(name) VALUES("fiona")`)
|
|
if err != nil {
|
|
t.Fatalf("failed to insert record: %s", err.Error())
|
|
}
|
|
|
|
r, err := db.QueryStringStmt(`SELECT id || "_bar", name FROM foo`)
|
|
if err != nil {
|
|
t.Fatalf("failed to query table: %s", err.Error())
|
|
}
|
|
if exp, got := `[{"columns":["id || \"_bar\"","name"],"types":["","text"],"values":[["1_bar","fiona"]]}]`, asJSON(r); exp != got {
|
|
t.Fatalf("unexpected results for query\nexp: %s\ngot: %s", exp, got)
|
|
}
|
|
}
|
|
|
|
func Test_SimpleMultiStatements(t *testing.T) {
|
|
db, path := mustCreateDatabase()
|
|
defer db.Close()
|
|
defer os.Remove(path)
|
|
|
|
_, err := db.ExecuteStringStmt("CREATE TABLE foo (id INTEGER NOT NULL PRIMARY KEY, name TEXT)")
|
|
if err != nil {
|
|
t.Fatalf("failed to create table: %s", err.Error())
|
|
}
|
|
|
|
re, err := db.Execute([]Statement{
|
|
{`INSERT INTO foo(name) VALUES("fiona")`, nil},
|
|
{`INSERT INTO foo(name) VALUES("dana")`, nil},
|
|
}, false, false)
|
|
if err != nil {
|
|
t.Fatalf("failed to insert record: %s", err.Error())
|
|
}
|
|
if exp, got := `[{"last_insert_id":1,"rows_affected":1},{"last_insert_id":2,"rows_affected":1}]`, asJSON(re); exp != got {
|
|
t.Fatalf("unexpected results for query\nexp: %s\ngot: %s", exp, got)
|
|
}
|
|
|
|
ro, err := db.Query([]Statement{
|
|
{`SELECT * FROM foo`, nil},
|
|
{`SELECT * FROM foo`, nil},
|
|
}, false, false)
|
|
if err != nil {
|
|
t.Fatalf("failed to query empty table: %s", err.Error())
|
|
}
|
|
if exp, got := `[{"columns":["id","name"],"types":["integer","text"],"values":[[1,"fiona"],[2,"dana"]]},{"columns":["id","name"],"types":["integer","text"],"values":[[1,"fiona"],[2,"dana"]]}]`, asJSON(ro); exp != got {
|
|
t.Fatalf("unexpected results for query\nexp: %s\ngot: %s", exp, got)
|
|
}
|
|
}
|
|
|
|
func Test_SimpleSingleMultiLineStatements(t *testing.T) {
|
|
db, path := mustCreateDatabase()
|
|
defer db.Close()
|
|
defer os.Remove(path)
|
|
|
|
_, err := db.Execute([]Statement{
|
|
{`
|
|
CREATE TABLE foo (
|
|
id INTEGER NOT NULL PRIMARY KEY,
|
|
name TEXT
|
|
)`, nil}}, false, false)
|
|
if err != nil {
|
|
t.Fatalf("failed to create table: %s", err.Error())
|
|
}
|
|
|
|
re, err := db.Execute(
|
|
[]Statement{
|
|
{`INSERT INTO foo(name) VALUES("fiona")`, nil},
|
|
{`INSERT INTO foo(name) VALUES("dana")`, nil},
|
|
}, false, false)
|
|
if err != nil {
|
|
t.Fatalf("failed to insert record: %s", err.Error())
|
|
}
|
|
if exp, got := `[{"last_insert_id":1,"rows_affected":1},{"last_insert_id":2,"rows_affected":1}]`, asJSON(re); exp != got {
|
|
t.Fatalf("unexpected results for query\nexp: %s\ngot: %s", exp, got)
|
|
}
|
|
}
|
|
|
|
func Test_SimpleFailingStatements_Execute(t *testing.T) {
|
|
db, path := mustCreateDatabase()
|
|
defer db.Close()
|
|
defer os.Remove(path)
|
|
|
|
r, err := db.ExecuteStringStmt(`INSERT INTO foo(name) VALUES("fiona")`)
|
|
if err != nil {
|
|
t.Fatalf("error executing insertion into non-existent table: %s", err.Error())
|
|
}
|
|
if exp, got := `[{"error":"no such table: foo"}]`, asJSON(r); exp != got {
|
|
t.Fatalf("unexpected results for query\nexp: %s\ngot: %s", exp, got)
|
|
}
|
|
|
|
r, err = db.ExecuteStringStmt(`CREATE TABLE foo (id INTEGER NOT NULL PRIMARY KEY, name TEXT)`)
|
|
if err != nil {
|
|
t.Fatalf("failed to create table: %s", err.Error())
|
|
}
|
|
if exp, got := `[{}]`, asJSON(r); exp != got {
|
|
t.Fatalf("unexpected results for query\nexp: %s\ngot: %s", exp, got)
|
|
}
|
|
r, err = db.ExecuteStringStmt(`CREATE TABLE foo (id INTEGER NOT NULL PRIMARY KEY, name TEXT)`)
|
|
if err != nil {
|
|
t.Fatalf("failed to attempt creation of duplicate table: %s", err.Error())
|
|
}
|
|
if exp, got := `[{"error":"table foo already exists"}]`, asJSON(r); exp != got {
|
|
t.Fatalf("unexpected results for query\nexp: %s\ngot: %s", exp, got)
|
|
}
|
|
|
|
r, err = db.ExecuteStringStmt(`INSERT INTO foo(id, name) VALUES(11, "fiona")`)
|
|
if err != nil {
|
|
t.Fatalf("failed to insert record: %s", err.Error())
|
|
}
|
|
if exp, got := `[{"last_insert_id":11,"rows_affected":1}]`, asJSON(r); exp != got {
|
|
t.Fatalf("unexpected results for query\nexp: %s\ngot: %s", exp, got)
|
|
}
|
|
r, err = db.ExecuteStringStmt(`INSERT INTO foo(id, name) VALUES(11, "fiona")`)
|
|
if err != nil {
|
|
t.Fatalf("failed to attempt duplicate record insertion: %s", err.Error())
|
|
}
|
|
if exp, got := `[{"error":"UNIQUE constraint failed: foo.id"}]`, asJSON(r); exp != got {
|
|
t.Fatalf("unexpected results for query\nexp: %s\ngot: %s", exp, got)
|
|
}
|
|
|
|
r, err = db.ExecuteStringStmt(`utter nonsense`)
|
|
if err != nil {
|
|
if exp, got := `[{"error":"near \"utter\": syntax error"}]`, asJSON(r); exp != got {
|
|
t.Fatalf("unexpected results for query\nexp: %s\ngot: %s", exp, got)
|
|
}
|
|
}
|
|
}
|
|
|
|
func Test_SimpleFailingStatements_Query(t *testing.T) {
|
|
db, path := mustCreateDatabase()
|
|
defer db.Close()
|
|
defer os.Remove(path)
|
|
|
|
ro, err := db.QueryStringStmt(`SELECT * FROM bar`)
|
|
if err != nil {
|
|
t.Fatalf("failed to attempt query of non-existent table: %s", err.Error())
|
|
}
|
|
if exp, got := `[{"error":"no such table: bar"}]`, asJSON(ro); exp != got {
|
|
t.Fatalf("unexpected results for query\nexp: %s\ngot: %s", exp, got)
|
|
}
|
|
|
|
ro, err = db.QueryStringStmt(`SELECTxx * FROM foo`)
|
|
if err != nil {
|
|
t.Fatalf("failed to attempt nonsense query: %s", err.Error())
|
|
}
|
|
if exp, got := `[{"error":"near \"SELECTxx\": syntax error"}]`, asJSON(ro); exp != got {
|
|
t.Fatalf("unexpected results for query\nexp: %s\ngot: %s", exp, got)
|
|
}
|
|
r, err := db.QueryStringStmt(`utter nonsense`)
|
|
if err != nil {
|
|
if exp, got := `[{"error":"near \"utter\": syntax error"}]`, asJSON(r); exp != got {
|
|
t.Fatalf("unexpected results for query\nexp: %s\ngot: %s", exp, got)
|
|
}
|
|
}
|
|
}
|
|
|
|
func Test_SimplePragmaTableInfo(t *testing.T) {
|
|
db, path := mustCreateDatabase()
|
|
defer db.Close()
|
|
defer os.Remove(path)
|
|
|
|
r, err := db.ExecuteStringStmt(`CREATE TABLE foo (id INTEGER NOT NULL PRIMARY KEY, name TEXT)`)
|
|
if err != nil {
|
|
t.Fatalf("failed to create table: %s", err.Error())
|
|
}
|
|
if exp, got := `[{}]`, asJSON(r); exp != got {
|
|
t.Fatalf("unexpected results for query\nexp: %s\ngot: %s", exp, got)
|
|
}
|
|
|
|
res, err := db.QueryStringStmt(`PRAGMA table_info("foo")`)
|
|
if err != nil {
|
|
t.Fatalf("failed to query a common table expression: %s", err.Error())
|
|
}
|
|
if exp, got := `[{"columns":["cid","name","type","notnull","dflt_value","pk"],"types":["","","","","",""],"values":[[0,"id","INTEGER",1,null,1],[1,"name","TEXT",0,null,0]]}]`, asJSON(res); exp != got {
|
|
t.Fatalf("unexpected results for query\nexp: %s\ngot: %s", exp, got)
|
|
}
|
|
}
|
|
|
|
func Test_SimpleParameterizedStatements(t *testing.T) {
|
|
db, path := mustCreateDatabase()
|
|
defer db.Close()
|
|
defer os.Remove(path)
|
|
|
|
_, err := db.ExecuteStringStmt("CREATE TABLE foo (id INTEGER NOT NULL PRIMARY KEY, name TEXT)")
|
|
if err != nil {
|
|
t.Fatalf("failed to create table: %s", err.Error())
|
|
}
|
|
|
|
s := Statement{
|
|
SQL: "INSERT INTO foo(name) VALUES(?)",
|
|
Parameters: []driver.Value{"fiona"},
|
|
}
|
|
_, err = db.Execute([]Statement{s}, false, false)
|
|
if err != nil {
|
|
t.Fatalf("failed to insert record: %s", err.Error())
|
|
}
|
|
|
|
s.Parameters = []driver.Value{"aoife"}
|
|
_, err = db.Execute([]Statement{s}, false, false)
|
|
if err != nil {
|
|
t.Fatalf("failed to insert record: %s", err.Error())
|
|
}
|
|
|
|
r, err := db.QueryStringStmt(`SELECT * FROM foo`)
|
|
if err != nil {
|
|
t.Fatalf("failed to query table: %s", err.Error())
|
|
}
|
|
if exp, got := `[{"columns":["id","name"],"types":["integer","text"],"values":[[1,"fiona"],[2,"aoife"]]}]`, asJSON(r); exp != got {
|
|
t.Fatalf("unexpected results for query\nexp: %s\ngot: %s", exp, got)
|
|
}
|
|
|
|
s.SQL = "SELECT * FROM foo WHERE name=?"
|
|
s.Parameters = []driver.Value{"aoife"}
|
|
r, err = db.Query([]Statement{s}, false, false)
|
|
if err != nil {
|
|
t.Fatalf("failed to query table: %s", err.Error())
|
|
}
|
|
if exp, got := `[{"columns":["id","name"],"types":["integer","text"],"values":[[2,"aoife"]]}]`, asJSON(r); exp != got {
|
|
t.Fatalf("unexpected results for query\nexp: %s\ngot: %s", exp, got)
|
|
}
|
|
|
|
s.Parameters = []driver.Value{"fiona"}
|
|
r, err = db.Query([]Statement{s}, false, false)
|
|
if err != nil {
|
|
t.Fatalf("failed to query table: %s", err.Error())
|
|
}
|
|
if exp, got := `[{"columns":["id","name"],"types":["integer","text"],"values":[[1,"fiona"]]}]`, asJSON(r); exp != got {
|
|
t.Fatalf("unexpected results for query\nexp: %s\ngot: %s", exp, got)
|
|
}
|
|
|
|
stmts := []Statement{
|
|
{"SELECT * FROM foo WHERE NAME=?", []driver.Value{"fiona"}},
|
|
{"SELECT * FROM foo WHERE NAME=?", []driver.Value{"aoife"}},
|
|
}
|
|
r, err = db.Query(stmts, false, false)
|
|
if err != nil {
|
|
t.Fatalf("failed to query table: %s", err.Error())
|
|
}
|
|
if exp, got := `[{"columns":["id","name"],"types":["integer","text"],"values":[[1,"fiona"]]},{"columns":["id","name"],"types":["integer","text"],"values":[[2,"aoife"]]}]`, asJSON(r); exp != got {
|
|
t.Fatalf("unexpected results for query\nexp: %s\ngot: %s", exp, got)
|
|
}
|
|
}
|
|
|
|
func Test_CommonTableExpressions(t *testing.T) {
|
|
db, path := mustCreateDatabase()
|
|
defer db.Close()
|
|
defer os.Remove(path)
|
|
|
|
_, err := db.ExecuteStringStmt("CREATE TABLE test(x foo)")
|
|
if err != nil {
|
|
t.Fatalf("failed to create table: %s", err.Error())
|
|
}
|
|
|
|
_, err = db.ExecuteStringStmt(`INSERT INTO test VALUES(1)`)
|
|
if err != nil {
|
|
t.Fatalf("failed to insert record: %s", err.Error())
|
|
}
|
|
|
|
r, err := db.QueryStringStmt(`WITH bar AS (SELECT * FROM test) SELECT * FROM test WHERE x = 1`)
|
|
if err != nil {
|
|
t.Fatalf("failed to query a common table expression: %s", err.Error())
|
|
}
|
|
if exp, got := `[{"columns":["x"],"types":["foo"],"values":[[1]]}]`, asJSON(r); exp != got {
|
|
t.Fatalf("unexpected results for query\nexp: %s\ngot: %s", exp, got)
|
|
}
|
|
|
|
r, err = db.QueryStringStmt(`WITH bar AS (SELECT * FROM test) SELECT * FROM test WHERE x = 2`)
|
|
if err != nil {
|
|
t.Fatalf("failed to query a common table expression: %s", err.Error())
|
|
}
|
|
if exp, got := `[{"columns":["x"],"types":["foo"]}]`, asJSON(r); exp != got {
|
|
t.Fatalf("unexpected results for query\nexp: %s\ngot: %s", exp, got)
|
|
}
|
|
}
|
|
|
|
func Test_ForeignKeyConstraints(t *testing.T) {
|
|
db, path := mustCreateDatabase()
|
|
defer db.Close()
|
|
defer os.Remove(path)
|
|
|
|
_, err := db.ExecuteStringStmt("CREATE TABLE foo (id INTEGER NOT NULL PRIMARY KEY, ref INTEGER REFERENCES foo(id))")
|
|
if err != nil {
|
|
t.Fatalf("failed to create table: %s", err.Error())
|
|
}
|
|
|
|
// Explicitly disable constraints.
|
|
if err := db.EnableFKConstraints(false); err != nil {
|
|
t.Fatalf("failed to enable foreign key constraints: %s", err.Error())
|
|
}
|
|
|
|
// Check constraints
|
|
fk, err := db.FKConstraints()
|
|
if err != nil {
|
|
t.Fatalf("failed to check FK constraints: %s", err.Error())
|
|
}
|
|
if fk != false {
|
|
t.Fatal("FK constraints are not disabled")
|
|
}
|
|
|
|
r, err := db.ExecuteStringStmt(`INSERT INTO foo(id, ref) VALUES(1, 2)`)
|
|
if err != nil {
|
|
t.Fatalf("failed to execute FK test statement: %s", err.Error())
|
|
}
|
|
if exp, got := `[{"last_insert_id":1,"rows_affected":1}]`, asJSON(r); exp != got {
|
|
t.Fatalf("unexpected results for query\nexp: %s\ngot: %s", exp, got)
|
|
}
|
|
|
|
// Explicitly enable constraints.
|
|
if err := db.EnableFKConstraints(true); err != nil {
|
|
t.Fatalf("failed to enable foreign key constraints: %s", err.Error())
|
|
}
|
|
|
|
// Check constraints
|
|
fk, err = db.FKConstraints()
|
|
if err != nil {
|
|
t.Fatalf("failed to check FK constraints: %s", err.Error())
|
|
}
|
|
if fk != true {
|
|
t.Fatal("FK constraints are not enabled")
|
|
}
|
|
|
|
r, err = db.ExecuteStringStmt(`INSERT INTO foo(id, ref) VALUES(1, 3)`)
|
|
if err != nil {
|
|
t.Fatalf("failed to execute FK test statement: %s", err.Error())
|
|
}
|
|
if exp, got := `[{"error":"UNIQUE constraint failed: foo.id"}]`, asJSON(r); exp != got {
|
|
t.Fatalf("unexpected results for query\nexp: %s\ngot: %s", exp, got)
|
|
}
|
|
}
|
|
|
|
func Test_UniqueConstraints(t *testing.T) {
|
|
db, path := mustCreateDatabase()
|
|
defer db.Close()
|
|
defer os.Remove(path)
|
|
|
|
_, err := db.ExecuteStringStmt("CREATE TABLE foo (id INTEGER NOT NULL PRIMARY KEY, name TEXT, CONSTRAINT name_unique UNIQUE (name))")
|
|
if err != nil {
|
|
t.Fatalf("failed to create table: %s", err.Error())
|
|
}
|
|
|
|
r, err := db.ExecuteStringStmt(`INSERT INTO foo(name) VALUES("fiona")`)
|
|
if err != nil {
|
|
t.Fatalf("error executing insertion into table: %s", err.Error())
|
|
}
|
|
if exp, got := `[{"last_insert_id":1,"rows_affected":1}]`, asJSON(r); exp != got {
|
|
t.Fatalf("unexpected results for INSERT\nexp: %s\ngot: %s", exp, got)
|
|
}
|
|
|
|
// UNIQUE constraint should fire.
|
|
r, err = db.ExecuteStringStmt(`INSERT INTO foo(name) VALUES("fiona")`)
|
|
if err != nil {
|
|
t.Fatalf("error executing insertion into table: %s", err.Error())
|
|
}
|
|
if exp, got := `[{"error":"UNIQUE constraint failed: foo.name"}]`, asJSON(r); exp != got {
|
|
t.Fatalf("unexpected results for INSERT\nexp: %s\ngot: %s", exp, got)
|
|
}
|
|
}
|
|
|
|
func Test_ActiveTransaction(t *testing.T) {
|
|
db, path := mustCreateDatabase()
|
|
defer db.Close()
|
|
defer os.Remove(path)
|
|
|
|
if db.TransactionActive() {
|
|
t.Fatal("transaction incorrectly marked as active")
|
|
}
|
|
|
|
if _, err := db.ExecuteStringStmt(`BEGIN`); err != nil {
|
|
t.Fatalf("error starting transaction: %s", err.Error())
|
|
}
|
|
|
|
if !db.TransactionActive() {
|
|
t.Fatal("transaction incorrectly marked as inactive")
|
|
}
|
|
|
|
if _, err := db.ExecuteStringStmt(`COMMIT`); err != nil {
|
|
t.Fatalf("error starting transaction: %s", err.Error())
|
|
}
|
|
|
|
if db.TransactionActive() {
|
|
t.Fatal("transaction incorrectly marked as active")
|
|
}
|
|
|
|
if _, err := db.ExecuteStringStmt(`BEGIN`); err != nil {
|
|
t.Fatalf("error starting transaction: %s", err.Error())
|
|
}
|
|
|
|
if !db.TransactionActive() {
|
|
t.Fatal("transaction incorrectly marked as inactive")
|
|
}
|
|
|
|
if _, err := db.ExecuteStringStmt(`ROLLBACK`); err != nil {
|
|
t.Fatalf("error starting transaction: %s", err.Error())
|
|
}
|
|
|
|
if db.TransactionActive() {
|
|
t.Fatal("transaction incorrectly marked as active")
|
|
}
|
|
}
|
|
|
|
func Test_AbortTransaction(t *testing.T) {
|
|
db, path := mustCreateDatabase()
|
|
defer db.Close()
|
|
defer os.Remove(path)
|
|
|
|
if err := db.AbortTransaction(); err != nil {
|
|
t.Fatalf("error abrorting non-active transaction: %s", err.Error())
|
|
}
|
|
|
|
if _, err := db.ExecuteStringStmt(`BEGIN`); err != nil {
|
|
t.Fatalf("error starting transaction: %s", err.Error())
|
|
}
|
|
|
|
if !db.TransactionActive() {
|
|
t.Fatal("transaction incorrectly marked as inactive")
|
|
}
|
|
|
|
if err := db.AbortTransaction(); err != nil {
|
|
t.Fatalf("error abrorting non-active transaction: %s", err.Error())
|
|
}
|
|
|
|
if db.TransactionActive() {
|
|
t.Fatal("transaction incorrectly marked as active")
|
|
}
|
|
}
|
|
|
|
func Test_PartialFail(t *testing.T) {
|
|
db, path := mustCreateDatabase()
|
|
defer db.Close()
|
|
defer os.Remove(path)
|
|
|
|
_, err := db.ExecuteStringStmt("CREATE TABLE foo (id INTEGER NOT NULL PRIMARY KEY, name TEXT)")
|
|
if err != nil {
|
|
t.Fatalf("failed to create table: %s", err.Error())
|
|
}
|
|
|
|
stmts := []Statement{
|
|
{`INSERT INTO foo(id, name) VALUES(1, "fiona")`, nil},
|
|
{`INSERT INTO foo(id, name) VALUES(2, "fiona")`, nil},
|
|
{`INSERT INTO foo(id, name) VALUES(1, "fiona")`, nil},
|
|
{`INSERT INTO foo(id, name) VALUES(4, "fiona")`, nil},
|
|
}
|
|
r, err := db.Execute(stmts, false, false)
|
|
if err != nil {
|
|
t.Fatalf("failed to insert records: %s", err.Error())
|
|
}
|
|
if exp, got := `[{"last_insert_id":1,"rows_affected":1},{"last_insert_id":2,"rows_affected":1},{"error":"UNIQUE constraint failed: foo.id"},{"last_insert_id":4,"rows_affected":1}]`, asJSON(r); exp != got {
|
|
t.Fatalf("unexpected results for query\nexp: %s\ngot: %s", exp, got)
|
|
}
|
|
ro, err := db.QueryStringStmt(`SELECT * FROM foo`)
|
|
if err != nil {
|
|
t.Fatalf("failed to query table: %s", err.Error())
|
|
}
|
|
if exp, got := `[{"columns":["id","name"],"types":["integer","text"],"values":[[1,"fiona"],[2,"fiona"],[4,"fiona"]]}]`, asJSON(ro); exp != got {
|
|
t.Fatalf("unexpected results for query\nexp: %s\ngot: %s", exp, got)
|
|
}
|
|
}
|
|
|
|
func Test_SimpleTransaction(t *testing.T) {
|
|
db, path := mustCreateDatabase()
|
|
defer db.Close()
|
|
defer os.Remove(path)
|
|
|
|
_, err := db.ExecuteStringStmt("CREATE TABLE foo (id INTEGER NOT NULL PRIMARY KEY, name TEXT)")
|
|
if err != nil {
|
|
t.Fatalf("failed to create table: %s", err.Error())
|
|
}
|
|
|
|
stmts := []Statement{
|
|
{`INSERT INTO foo(id, name) VALUES(1, "fiona")`, nil},
|
|
{`INSERT INTO foo(id, name) VALUES(2, "fiona")`, nil},
|
|
{`INSERT INTO foo(id, name) VALUES(3, "fiona")`, nil},
|
|
{`INSERT INTO foo(id, name) VALUES(4, "fiona")`, nil},
|
|
}
|
|
r, err := db.Execute(stmts, true, false)
|
|
if err != nil {
|
|
t.Fatalf("failed to insert records: %s", err.Error())
|
|
}
|
|
if exp, got := `[{"last_insert_id":1,"rows_affected":1},{"last_insert_id":2,"rows_affected":1},{"last_insert_id":3,"rows_affected":1},{"last_insert_id":4,"rows_affected":1}]`, asJSON(r); exp != got {
|
|
t.Fatalf("unexpected results for query\nexp: %s\ngot: %s", exp, got)
|
|
}
|
|
ro, err := db.QueryStringStmt(`SELECT * FROM foo`)
|
|
if err != nil {
|
|
t.Fatalf("failed to query table: %s", err.Error())
|
|
}
|
|
if exp, got := `[{"columns":["id","name"],"types":["integer","text"],"values":[[1,"fiona"],[2,"fiona"],[3,"fiona"],[4,"fiona"]]}]`, asJSON(ro); exp != got {
|
|
t.Fatalf("unexpected results for query\nexp: %s\ngot: %s", exp, got)
|
|
}
|
|
}
|
|
|
|
func Test_PartialFailTransaction(t *testing.T) {
|
|
db, path := mustCreateDatabase()
|
|
defer db.Close()
|
|
defer os.Remove(path)
|
|
|
|
_, err := db.ExecuteStringStmt("CREATE TABLE foo (id INTEGER NOT NULL PRIMARY KEY, name TEXT)")
|
|
if err != nil {
|
|
t.Fatalf("failed to create table: %s", err.Error())
|
|
}
|
|
|
|
stmts := []Statement{
|
|
{`INSERT INTO foo(id, name) VALUES(1, "fiona")`, nil},
|
|
{`INSERT INTO foo(id, name) VALUES(2, "fiona")`, nil},
|
|
{`INSERT INTO foo(id, name) VALUES(1, "fiona")`, nil},
|
|
{`INSERT INTO foo(id, name) VALUES(4, "fiona")`, nil},
|
|
}
|
|
r, err := db.Execute(stmts, true, false)
|
|
if err != nil {
|
|
t.Fatalf("failed to insert records: %s", err.Error())
|
|
}
|
|
if exp, got := `[{"last_insert_id":1,"rows_affected":1},{"last_insert_id":2,"rows_affected":1},{"error":"UNIQUE constraint failed: foo.id"}]`, asJSON(r); exp != got {
|
|
t.Fatalf("unexpected results for query\nexp: %s\ngot: %s", exp, got)
|
|
}
|
|
ro, err := db.QueryStringStmt(`SELECT * FROM foo`)
|
|
if err != nil {
|
|
t.Fatalf("failed to query table: %s", err.Error())
|
|
}
|
|
if exp, got := `[{"columns":["id","name"],"types":["integer","text"]}]`, asJSON(ro); exp != got {
|
|
t.Fatalf("unexpected results for query\nexp: %s\ngot: %s", exp, got)
|
|
}
|
|
}
|
|
|
|
func Test_Backup(t *testing.T) {
|
|
db, path := mustCreateDatabase()
|
|
defer db.Close()
|
|
defer os.Remove(path)
|
|
|
|
_, err := db.ExecuteStringStmt("CREATE TABLE foo (id INTEGER NOT NULL PRIMARY KEY, name TEXT)")
|
|
if err != nil {
|
|
t.Fatalf("failed to create table: %s", err.Error())
|
|
}
|
|
|
|
stmts := []Statement{
|
|
{`INSERT INTO foo(id, name) VALUES(1, "fiona")`, nil},
|
|
{`INSERT INTO foo(id, name) VALUES(2, "fiona")`, nil},
|
|
{`INSERT INTO foo(id, name) VALUES(3, "fiona")`, nil},
|
|
{`INSERT INTO foo(id, name) VALUES(4, "fiona")`, nil},
|
|
}
|
|
_, err = db.Execute(stmts, true, false)
|
|
if err != nil {
|
|
t.Fatalf("failed to insert records: %s", err.Error())
|
|
}
|
|
|
|
dstDB, err := ioutil.TempFile("", "rqlilte-bak-")
|
|
if err != nil {
|
|
t.Fatalf("failed to create temp file: %s", err.Error())
|
|
}
|
|
dstDB.Close()
|
|
defer os.Remove(dstDB.Name())
|
|
|
|
err = db.Backup(dstDB.Name())
|
|
if err != nil {
|
|
t.Fatalf("failed to backup database: %s", err.Error())
|
|
}
|
|
|
|
newDB, err := Open(dstDB.Name())
|
|
if err != nil {
|
|
t.Fatalf("failed to open backup database: %s", err.Error())
|
|
}
|
|
defer newDB.Close()
|
|
defer os.Remove(dstDB.Name())
|
|
ro, err := newDB.QueryStringStmt(`SELECT * FROM foo`)
|
|
if err != nil {
|
|
t.Fatalf("failed to query table: %s", err.Error())
|
|
}
|
|
if exp, got := `[{"columns":["id","name"],"types":["integer","text"],"values":[[1,"fiona"],[2,"fiona"],[3,"fiona"],[4,"fiona"]]}]`, asJSON(ro); exp != got {
|
|
t.Fatalf("unexpected results for query\nexp: %s\ngot: %s", exp, got)
|
|
}
|
|
}
|
|
|
|
func Test_Dump(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
db, path := mustCreateDatabase()
|
|
defer db.Close()
|
|
defer os.Remove(path)
|
|
|
|
_, err := db.ExecuteStringStmt(chinook.DB)
|
|
if err != nil {
|
|
t.Fatalf("failed to load chinook dump: %s", err.Error())
|
|
}
|
|
|
|
var b strings.Builder
|
|
if err := db.Dump(&b); err != nil {
|
|
t.Fatalf("failed to dump database: %s", err.Error())
|
|
}
|
|
|
|
if b.String() != chinook.DB {
|
|
t.Fatal("dumped database does not equal entered database")
|
|
}
|
|
}
|
|
|
|
func Test_DumpMemory(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
db, path := mustCreateDatabase()
|
|
defer db.Close()
|
|
defer os.Remove(path)
|
|
|
|
inmem, err := LoadInMemoryWithDSN(path, "")
|
|
if err != nil {
|
|
t.Fatalf("failed to create loaded in-memory database: %s", err.Error())
|
|
}
|
|
|
|
_, err = inmem.ExecuteStringStmt(chinook.DB)
|
|
if err != nil {
|
|
t.Fatalf("failed to load chinook dump: %s", err.Error())
|
|
}
|
|
|
|
var b strings.Builder
|
|
if err := inmem.Dump(&b); err != nil {
|
|
t.Fatalf("failed to dump database: %s", err.Error())
|
|
}
|
|
|
|
if b.String() != chinook.DB {
|
|
t.Fatal("dumped database does not equal entered database")
|
|
}
|
|
}
|
|
|
|
func mustCreateDatabase() (*DB, string) {
|
|
var err error
|
|
f, err := ioutil.TempFile("", "rqlilte-test-")
|
|
if err != nil {
|
|
panic("failed to create temp file")
|
|
}
|
|
f.Close()
|
|
|
|
db, err := Open(f.Name())
|
|
if err != nil {
|
|
panic("failed to open database")
|
|
}
|
|
|
|
return db, f.Name()
|
|
}
|
|
|
|
func mustWriteAndOpenDatabase(b []byte) (*DB, string) {
|
|
var err error
|
|
f, err := ioutil.TempFile("", "rqlilte-test-write-")
|
|
if err != nil {
|
|
panic("failed to create temp file")
|
|
}
|
|
f.Close()
|
|
|
|
err = ioutil.WriteFile(f.Name(), b, 0660)
|
|
if err != nil {
|
|
panic("failed to write file")
|
|
}
|
|
|
|
db, err := Open(f.Name())
|
|
if err != nil {
|
|
panic("failed to open database")
|
|
}
|
|
return db, f.Name()
|
|
}
|
|
|
|
// mustExecute executes a statement, and panics on failure. Used for statements
|
|
// that should never fail, even taking into account test setup.
|
|
func mustExecute(db *DB, stmt string) {
|
|
_, err := db.ExecuteStringStmt(stmt)
|
|
if err != nil {
|
|
panic(fmt.Sprintf("failed to execute statement: %s", err.Error()))
|
|
}
|
|
}
|
|
|
|
// mustQuery executes a statement, and panics on failure. Used for statements
|
|
// that should never fail, even taking into account test setup.
|
|
func mustQuery(db *DB, stmt string) {
|
|
_, err := db.QueryStringStmt(stmt)
|
|
if err != nil {
|
|
panic(fmt.Sprintf("failed to query: %s", err.Error()))
|
|
}
|
|
}
|
|
|
|
func asJSON(v interface{}) string {
|
|
b, err := json.Marshal(v)
|
|
if err != nil {
|
|
panic("failed to JSON marshal value")
|
|
}
|
|
return string(b)
|
|
}
|