Add support for CockorachDB.
This commit is contained in:
go-jet
parent
3ff9241eea
commit
bc776f947b
33 changed files with 1040 additions and 1037 deletions
|
|
@ -2,6 +2,7 @@ package sqlite
|
|||
|
||||
import (
|
||||
"context"
|
||||
"database/sql"
|
||||
"math/rand"
|
||||
|
||||
"testing"
|
||||
|
|
@ -15,9 +16,6 @@ import (
|
|||
)
|
||||
|
||||
func TestInsertValues(t *testing.T) {
|
||||
tx := beginSampleDBTx(t)
|
||||
defer tx.Rollback()
|
||||
|
||||
insertQuery := Link.INSERT(Link.ID, Link.URL, Link.Name, Link.Description).
|
||||
VALUES(100, "http://www.postgresqltutorial.com", "PostgreSQL Tutorial", nil).
|
||||
VALUES(101, "http://www.google.com", "Google", "Search engine").
|
||||
|
|
@ -32,31 +30,32 @@ VALUES (?, ?, ?, ?),
|
|||
101, "http://www.google.com", "Google", "Search engine",
|
||||
102, "http://www.yahoo.com", "Yahoo", nil)
|
||||
|
||||
_, err := insertQuery.Exec(tx)
|
||||
require.NoError(t, err)
|
||||
requireLogged(t, insertQuery)
|
||||
testutils.ExecuteInTxAndRollback(t, sampleDB, func(tx *sql.Tx) {
|
||||
testutils.AssertExec(t, insertQuery, tx)
|
||||
requireLogged(t, insertQuery)
|
||||
|
||||
insertedLinks := []model.Link{}
|
||||
var insertedLinks []model.Link
|
||||
|
||||
err = SELECT(Link.AllColumns).
|
||||
FROM(Link).
|
||||
WHERE(Link.ID.GT_EQ(Int(100))).
|
||||
ORDER_BY(Link.ID).
|
||||
Query(tx, &insertedLinks)
|
||||
err := SELECT(Link.AllColumns).
|
||||
FROM(Link).
|
||||
WHERE(Link.ID.GT_EQ(Int(100))).
|
||||
ORDER_BY(Link.ID).
|
||||
Query(tx, &insertedLinks)
|
||||
|
||||
require.NoError(t, err)
|
||||
require.Equal(t, len(insertedLinks), 3)
|
||||
testutils.AssertDeepEqual(t, insertedLinks[0], postgreTutorial)
|
||||
testutils.AssertDeepEqual(t, insertedLinks[1], model.Link{
|
||||
ID: 101,
|
||||
URL: "http://www.google.com",
|
||||
Name: "Google",
|
||||
Description: testutils.StringPtr("Search engine"),
|
||||
})
|
||||
testutils.AssertDeepEqual(t, insertedLinks[2], model.Link{
|
||||
ID: 102,
|
||||
URL: "http://www.yahoo.com",
|
||||
Name: "Yahoo",
|
||||
require.NoError(t, err)
|
||||
require.Equal(t, len(insertedLinks), 3)
|
||||
testutils.AssertDeepEqual(t, insertedLinks[0], postgreTutorial)
|
||||
testutils.AssertDeepEqual(t, insertedLinks[1], model.Link{
|
||||
ID: 101,
|
||||
URL: "http://www.google.com",
|
||||
Name: "Google",
|
||||
Description: testutils.StringPtr("Search engine"),
|
||||
})
|
||||
testutils.AssertDeepEqual(t, insertedLinks[2], model.Link{
|
||||
ID: 102,
|
||||
URL: "http://www.yahoo.com",
|
||||
Name: "Yahoo",
|
||||
})
|
||||
})
|
||||
}
|
||||
|
||||
|
|
@ -67,41 +66,35 @@ var postgreTutorial = model.Link{
|
|||
}
|
||||
|
||||
func TestInsertEmptyColumnList(t *testing.T) {
|
||||
tx := beginSampleDBTx(t)
|
||||
defer tx.Rollback()
|
||||
|
||||
expectedSQL := `
|
||||
INSERT INTO link
|
||||
VALUES (100, 'http://www.postgresqltutorial.com', 'PostgreSQL Tutorial', NULL);
|
||||
`
|
||||
|
||||
stmt := Link.INSERT().
|
||||
VALUES(100, "http://www.postgresqltutorial.com", "PostgreSQL Tutorial", nil)
|
||||
|
||||
testutils.AssertDebugStatementSql(t, stmt, expectedSQL,
|
||||
testutils.AssertDebugStatementSql(t, stmt, `
|
||||
INSERT INTO link
|
||||
VALUES (100, 'http://www.postgresqltutorial.com', 'PostgreSQL Tutorial', NULL);
|
||||
`,
|
||||
100, "http://www.postgresqltutorial.com", "PostgreSQL Tutorial", nil)
|
||||
|
||||
_, err := stmt.Exec(tx)
|
||||
require.NoError(t, err)
|
||||
requireLogged(t, stmt)
|
||||
testutils.ExecuteInTxAndRollback(t, sampleDB, func(tx *sql.Tx) {
|
||||
_, err := stmt.Exec(tx)
|
||||
require.NoError(t, err)
|
||||
requireLogged(t, stmt)
|
||||
|
||||
insertedLinks := []model.Link{}
|
||||
var insertedLinks []model.Link
|
||||
|
||||
err = SELECT(Link.AllColumns).
|
||||
FROM(Link).
|
||||
WHERE(Link.ID.GT_EQ(Int(100))).
|
||||
ORDER_BY(Link.ID).
|
||||
Query(tx, &insertedLinks)
|
||||
err = SELECT(Link.AllColumns).
|
||||
FROM(Link).
|
||||
WHERE(Link.ID.GT_EQ(Int(100))).
|
||||
ORDER_BY(Link.ID).
|
||||
Query(tx, &insertedLinks)
|
||||
|
||||
require.NoError(t, err)
|
||||
require.Equal(t, len(insertedLinks), 1)
|
||||
testutils.AssertDeepEqual(t, insertedLinks[0], postgreTutorial)
|
||||
require.NoError(t, err)
|
||||
require.Equal(t, len(insertedLinks), 1)
|
||||
testutils.AssertDeepEqual(t, insertedLinks[0], postgreTutorial)
|
||||
})
|
||||
}
|
||||
|
||||
func TestInsertModelObject(t *testing.T) {
|
||||
tx := beginSampleDBTx(t)
|
||||
defer tx.Rollback()
|
||||
|
||||
linkData := model.Link{
|
||||
URL: "http://www.duckduckgo.com",
|
||||
Name: "Duck Duck go",
|
||||
|
|
@ -115,19 +108,13 @@ INSERT INTO link (url, name)
|
|||
VALUES ('http://www.duckduckgo.com', 'Duck Duck go');
|
||||
`, "http://www.duckduckgo.com", "Duck Duck go")
|
||||
|
||||
_, err := query.Exec(tx)
|
||||
require.NoError(t, err)
|
||||
testutils.ExecuteInTxAndRollback(t, sampleDB, func(tx *sql.Tx) {
|
||||
_, err := query.Exec(tx)
|
||||
require.NoError(t, err)
|
||||
})
|
||||
}
|
||||
|
||||
func TestInsertModelObjectEmptyColumnList(t *testing.T) {
|
||||
tx := beginSampleDBTx(t)
|
||||
defer tx.Rollback()
|
||||
|
||||
var expectedSQL = `
|
||||
INSERT INTO link
|
||||
VALUES (1000, 'http://www.duckduckgo.com', 'Duck Duck go', NULL);
|
||||
`
|
||||
|
||||
linkData := model.Link{
|
||||
ID: 1000,
|
||||
URL: "http://www.duckduckgo.com",
|
||||
|
|
@ -138,23 +125,18 @@ VALUES (1000, 'http://www.duckduckgo.com', 'Duck Duck go', NULL);
|
|||
INSERT().
|
||||
MODEL(linkData)
|
||||
|
||||
testutils.AssertDebugStatementSql(t, query, expectedSQL, int32(1000), "http://www.duckduckgo.com", "Duck Duck go", nil)
|
||||
testutils.AssertDebugStatementSql(t, query, `
|
||||
INSERT INTO link
|
||||
VALUES (1000, 'http://www.duckduckgo.com', 'Duck Duck go', NULL);
|
||||
`, int32(1000), "http://www.duckduckgo.com", "Duck Duck go", nil)
|
||||
|
||||
_, err := query.Exec(tx)
|
||||
require.NoError(t, err)
|
||||
testutils.ExecuteInTxAndRollback(t, sampleDB, func(tx *sql.Tx) {
|
||||
_, err := query.Exec(tx)
|
||||
require.NoError(t, err)
|
||||
})
|
||||
}
|
||||
|
||||
func TestInsertModelsObject(t *testing.T) {
|
||||
tx := beginSampleDBTx(t)
|
||||
defer tx.Rollback()
|
||||
|
||||
expectedSQL := `
|
||||
INSERT INTO link (url, name)
|
||||
VALUES ('http://www.postgresqltutorial.com', 'PostgreSQL Tutorial'),
|
||||
('http://www.google.com', 'Google'),
|
||||
('http://www.yahoo.com', 'Yahoo');
|
||||
`
|
||||
|
||||
tutorial := model.Link{
|
||||
URL: "http://www.postgresqltutorial.com",
|
||||
Name: "PostgreSQL Tutorial",
|
||||
|
|
@ -176,27 +158,20 @@ VALUES ('http://www.postgresqltutorial.com', 'PostgreSQL Tutorial'),
|
|||
yahoo,
|
||||
})
|
||||
|
||||
testutils.AssertDebugStatementSql(t, query, expectedSQL,
|
||||
testutils.AssertDebugStatementSql(t, query, `
|
||||
INSERT INTO link (url, name)
|
||||
VALUES ('http://www.postgresqltutorial.com', 'PostgreSQL Tutorial'),
|
||||
('http://www.google.com', 'Google'),
|
||||
('http://www.yahoo.com', 'Yahoo');
|
||||
`,
|
||||
"http://www.postgresqltutorial.com", "PostgreSQL Tutorial",
|
||||
"http://www.google.com", "Google",
|
||||
"http://www.yahoo.com", "Yahoo")
|
||||
|
||||
_, err := query.Exec(tx)
|
||||
require.NoError(t, err)
|
||||
testutils.AssertExecAndRollback(t, query, sampleDB)
|
||||
}
|
||||
|
||||
func TestInsertUsingMutableColumns(t *testing.T) {
|
||||
tx := beginSampleDBTx(t)
|
||||
defer tx.Rollback()
|
||||
|
||||
var expectedSQL = `
|
||||
INSERT INTO link (url, name, description)
|
||||
VALUES ('http://www.postgresqltutorial.com', 'PostgreSQL Tutorial', NULL),
|
||||
('http://www.google.com', 'Google', NULL),
|
||||
('http://www.google.com', 'Google', NULL),
|
||||
('http://www.yahoo.com', 'Yahoo', NULL);
|
||||
`
|
||||
|
||||
google := model.Link{
|
||||
URL: "http://www.google.com",
|
||||
Name: "Google",
|
||||
|
|
@ -213,20 +188,22 @@ VALUES ('http://www.postgresqltutorial.com', 'PostgreSQL Tutorial', NULL),
|
|||
MODEL(google).
|
||||
MODELS([]model.Link{google, yahoo})
|
||||
|
||||
testutils.AssertDebugStatementSql(t, stmt, expectedSQL,
|
||||
testutils.AssertDebugStatementSql(t, stmt, `
|
||||
INSERT INTO link (url, name, description)
|
||||
VALUES ('http://www.postgresqltutorial.com', 'PostgreSQL Tutorial', NULL),
|
||||
('http://www.google.com', 'Google', NULL),
|
||||
('http://www.google.com', 'Google', NULL),
|
||||
('http://www.yahoo.com', 'Yahoo', NULL);
|
||||
`,
|
||||
"http://www.postgresqltutorial.com", "PostgreSQL Tutorial", nil,
|
||||
"http://www.google.com", "Google", nil,
|
||||
"http://www.google.com", "Google", nil,
|
||||
"http://www.yahoo.com", "Yahoo", nil)
|
||||
|
||||
_, err := stmt.Exec(tx)
|
||||
require.NoError(t, err)
|
||||
testutils.AssertExecAndRollback(t, stmt, sampleDB)
|
||||
}
|
||||
|
||||
func TestInsertQuery(t *testing.T) {
|
||||
tx := beginSampleDBTx(t)
|
||||
defer tx.Rollback()
|
||||
|
||||
var expectedSQL = `
|
||||
INSERT INTO link (url, name)
|
||||
SELECT link.url AS "link.url",
|
||||
|
|
@ -242,24 +219,22 @@ WHERE link.id = 24;
|
|||
)
|
||||
|
||||
testutils.AssertDebugStatementSql(t, query, expectedSQL, int64(24))
|
||||
testutils.ExecuteInTxAndRollback(t, sampleDB, func(tx *sql.Tx) {
|
||||
_, err := query.Exec(tx)
|
||||
require.NoError(t, err)
|
||||
|
||||
_, err := query.Exec(tx)
|
||||
require.NoError(t, err)
|
||||
var youtubeLinks []model.Link
|
||||
err = Link.
|
||||
SELECT(Link.AllColumns).
|
||||
WHERE(Link.Name.EQ(String("Bing"))).
|
||||
Query(tx, &youtubeLinks)
|
||||
|
||||
youtubeLinks := []model.Link{}
|
||||
err = Link.
|
||||
SELECT(Link.AllColumns).
|
||||
WHERE(Link.Name.EQ(String("Bing"))).
|
||||
Query(tx, &youtubeLinks)
|
||||
|
||||
require.NoError(t, err)
|
||||
require.Equal(t, len(youtubeLinks), 2)
|
||||
require.NoError(t, err)
|
||||
require.Equal(t, len(youtubeLinks), 2)
|
||||
})
|
||||
}
|
||||
|
||||
func TestInsert_DEFAULT_VALUES_RETURNING(t *testing.T) {
|
||||
tx := beginSampleDBTx(t)
|
||||
defer tx.Rollback()
|
||||
|
||||
stmt := Link.INSERT().
|
||||
DEFAULT_VALUES().
|
||||
RETURNING(Link.AllColumns)
|
||||
|
|
@ -273,24 +248,23 @@ RETURNING link.id AS "link.id",
|
|||
link.description AS "link.description";
|
||||
`)
|
||||
|
||||
var link model.Link
|
||||
err := stmt.Query(tx, &link)
|
||||
require.NoError(t, err)
|
||||
testutils.ExecuteInTxAndRollback(t, sampleDB, func(tx *sql.Tx) {
|
||||
var link model.Link
|
||||
err := stmt.Query(tx, &link)
|
||||
require.NoError(t, err)
|
||||
|
||||
require.EqualValues(t, link, model.Link{
|
||||
ID: 25,
|
||||
URL: "www.",
|
||||
Name: "_",
|
||||
Description: nil,
|
||||
require.EqualValues(t, link, model.Link{
|
||||
ID: 25,
|
||||
URL: "www.",
|
||||
Name: "_",
|
||||
Description: nil,
|
||||
})
|
||||
})
|
||||
}
|
||||
|
||||
func TestInsertOnConflict(t *testing.T) {
|
||||
|
||||
t.Run("do nothing", func(t *testing.T) {
|
||||
tx := beginSampleDBTx(t)
|
||||
defer tx.Rollback()
|
||||
|
||||
link := model.Link{ID: rand.Int31()}
|
||||
|
||||
stmt := Link.INSERT(Link.AllColumns).
|
||||
|
|
@ -304,14 +278,11 @@ VALUES (?, ?, ?, ?),
|
|||
(?, ?, ?, ?)
|
||||
ON CONFLICT (id) DO NOTHING;
|
||||
`)
|
||||
testutils.AssertExec(t, stmt, tx, 1)
|
||||
testutils.AssertExecAndRollback(t, stmt, sampleDB, 1)
|
||||
requireLogged(t, stmt)
|
||||
})
|
||||
|
||||
t.Run("do update", func(t *testing.T) {
|
||||
tx := beginSampleDBTx(t)
|
||||
defer tx.Rollback()
|
||||
|
||||
stmt := Link.INSERT(Link.ID, Link.URL, Link.Name, Link.Description).
|
||||
VALUES(21, "http://www.postgresqltutorial.com", "PostgreSQL Tutorial", nil).
|
||||
VALUES(22, "http://www.postgresqltutorial.com", "PostgreSQL Tutorial", nil).
|
||||
|
|
@ -336,14 +307,11 @@ RETURNING link.id AS "link.id",
|
|||
link.description AS "link.description";
|
||||
`)
|
||||
|
||||
testutils.AssertExec(t, stmt, tx)
|
||||
testutils.AssertExecAndRollback(t, stmt, sampleDB)
|
||||
requireLogged(t, stmt)
|
||||
})
|
||||
|
||||
t.Run("do update complex", func(t *testing.T) {
|
||||
tx := beginSampleDBTx(t)
|
||||
defer tx.Rollback()
|
||||
|
||||
stmt := Link.INSERT(Link.ID, Link.URL, Link.Name, Link.Description).
|
||||
VALUES(21, "http://www.postgresqltutorial.com", "PostgreSQL Tutorial", nil).
|
||||
ON_CONFLICT(Link.ID).
|
||||
|
|
@ -370,7 +338,7 @@ ON CONFLICT (id) WHERE (id * 2) > 10 DO UPDATE
|
|||
WHERE link.description IS NOT NULL;
|
||||
`)
|
||||
|
||||
testutils.AssertExec(t, stmt, tx)
|
||||
testutils.AssertExecAndRollback(t, stmt, sampleDB)
|
||||
requireLogged(t, stmt)
|
||||
})
|
||||
}
|
||||
|
|
@ -384,7 +352,7 @@ func TestInsertContextDeadlineExceeded(t *testing.T) {
|
|||
|
||||
time.Sleep(10 * time.Millisecond)
|
||||
|
||||
dest := []model.Link{}
|
||||
var dest []model.Link
|
||||
err := stmt.QueryContext(ctx, sampleDB, &dest)
|
||||
require.Error(t, err, "context deadline exceeded")
|
||||
|
||||
|
|
|
|||
|
|
@ -35,6 +35,7 @@ func TestMain(m *testing.M) {
|
|||
var err error
|
||||
db, err = sql.Open("sqlite3", "file:"+dbconfig.SakilaDBPath)
|
||||
throw.OnError(err)
|
||||
defer db.Close()
|
||||
|
||||
_, err = db.Exec(fmt.Sprintf("ATTACH DATABASE '%s' as 'chinook';", dbconfig.ChinookDBPath))
|
||||
throw.OnError(err)
|
||||
|
|
@ -42,8 +43,6 @@ func TestMain(m *testing.M) {
|
|||
sampleDB, err = sql.Open("sqlite3", dbconfig.TestSampleDBPath)
|
||||
throw.OnError(err)
|
||||
|
||||
defer db.Close()
|
||||
|
||||
ret := m.Run()
|
||||
|
||||
if ret != 0 {
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue