GOMODULES
/
ksql
mirror of https://github.com/VinGarcia/ksql.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Add TDD tests for checking if context was cancelled

pull/32/head v1.5.1
Vinícius Garcia 2022-11-29 09:03:02 -03:00
parent f4cac02602
commit 83cf354f35
3 changed files with 138 additions and 92 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
docker-compose.yml
View File

@ -18,13 +18,13 @@ services:
- POSTGRES_DB=${DB_NAME:-ksql}
mysql:
image: mysql
image: mariadb:10.8
restart: always
ports:
- "127.0.0.1:3306:3306"
environment:
MYSQL_ROOT_PASSWORD: mysql
MYSQL_DATABASE: ksql
MARIADB_ROOT_PASSWORD: mysql
MARIADB_DATABASE: ksql
sqlserver:
image: microsoft/mssql-server-linux:2017-latest

22
ksql.go
View File

@ -84,14 +84,14 @@ type ScanArgError struct {
func (s ScanArgError) Error() string {
return fmt.Sprintf(
"error scanning input attribute with index %d: %s",
s.ColumnIndex, s.Err.Error(),
s.ColumnIndex, s.Err,
)
}
func (s ScanArgError) ErrorWithStructNames(structName string, colName string) error {
return fmt.Errorf(
"error scanning %s.%s: %s",
structName, colName, s.Err.Error(),
"error scanning %s.%s: %w",
structName, colName, s.Err,
)
}
@ -192,7 +192,7 @@ func (c DB) Query(
rows, err := c.db.QueryContext(ctx, query, params...)
if err != nil {
return fmt.Errorf("error running query: %s", err)
return fmt.Errorf("error running query: %w", err)
}
defer rows.Close()
@ -282,7 +282,7 @@ func (c DB) QueryOne(
rows, err := c.db.QueryContext(ctx, query, params...)
if err != nil {
return fmt.Errorf("error running query: %s", err)
return fmt.Errorf("error running query: %w", err)
}
defer rows.Close()
@ -441,7 +441,7 @@ func (c DB) Insert(
}
if err := table.validate(); err != nil {
return fmt.Errorf("can't insert in ksql.Table: %s", err)
return fmt.Errorf("can't insert in ksql.Table: %w", err)
}
info, err := structs.GetTagInfo(t.Elem())
@ -586,7 +586,7 @@ func (c DB) Delete(
idOrRecord interface{},
) error {
if err := table.validate(); err != nil {
return fmt.Errorf("can't delete from ksql.Table: %s", err)
return fmt.Errorf("can't delete from ksql.Table: %w", err)
}
idMap, err := normalizeIDsAsMap(table.idColumns, idOrRecord)
@ -605,7 +605,7 @@ func (c DB) Delete(
n, err := result.RowsAffected()
if err != nil {
return fmt.Errorf("unable to check if the record was succesfully deleted: %s", err)
return fmt.Errorf("unable to check if the record was succesfully deleted: %w", err)
}
if n == 0 {
@ -699,7 +699,7 @@ func (c DB) Patch(
n, err := result.RowsAffected()
if err != nil {
return fmt.Errorf(
"unexpected error: unable to fetch how many rows were affected by the update: %s",
"unexpected error: unable to fetch how many rows were affected by the update: %w",
err,
)
}
@ -946,7 +946,7 @@ func (c DB) Transaction(ctx context.Context, fn func(Provider) error) error {
case TxBeginner:
tx, err := txBeginner.BeginTx(ctx)
if err != nil {
return fmt.Errorf("KSQL: error starting transaction: %s", err)
return fmt.Errorf("KSQL: error starting transaction: %w", err)
}
defer func() {
if r := recover(); r != nil {
@ -970,7 +970,7 @@ func (c DB) Transaction(ctx context.Context, fn func(Provider) error) error {
if rollbackErr != nil {
err = fmt.Errorf(
"KSQL: unable to rollback after error: %s, rollback error: %w",
err.Error(), rollbackErr,
err, rollbackErr,
)
}
return err

202
test_adapters.go
View File

@ -4,6 +4,7 @@ import (
"context"
"database/sql"
"encoding/json"
"errors"
"fmt"
"io"
"reflect"
@ -93,7 +94,7 @@ func QueryTest(
) {
ctx := context.Background()
t.Run("QueryTest", func(t *testing.T) {
t.Run("Query", func(t *testing.T) {
variations := []struct {
desc string
queryPrefix string
@ -554,6 +555,21 @@ func QueryTest(
err := c.Query(ctx, &users, `SELECT * FROM users`)
tt.AssertErrContains(t, err, "KSQL", "fakeCloseErr")
})
t.Run("should report error context.Canceled if the context is canceled", func(t *testing.T) {
ctx, cancel := context.WithCancel(ctx)
cancel()
db, closer := newDBAdapter(t)
defer closer.Close()
c := newTestDB(db, driver)
var users []user
err := c.Query(ctx, &users, `SELECT * FROM users`)
tt.AssertErrContains(t, err, "context", "canceled")
tt.AssertEqual(t, errors.Is(err, context.Canceled), true)
})
})
})
}
@ -566,6 +582,8 @@ func QueryOneTest(
connStr string,
newDBAdapter func(t *testing.T) (DBAdapter, io.Closer),
) {
ctx := context.Background()
t.Run("QueryOne", func(t *testing.T) {
variations := []struct {
desc string
@ -591,7 +609,6 @@ func QueryOneTest(
db, closer := newDBAdapter(t)
defer closer.Close()
ctx := context.Background()
c := newTestDB(db, driver)
u := user{}
err := c.QueryOne(ctx, &u, variation.queryPrefix+`FROM users WHERE id=1;`)
@ -602,8 +619,6 @@ func QueryOneTest(
db, closer := newDBAdapter(t)
defer closer.Close()
ctx := context.Background()
_, err := db.ExecContext(ctx, `INSERT INTO users (name, age, address) VALUES ('Bia', 0, '{"country":"BR"}')`)
tt.AssertNoErr(t, err)
@ -623,8 +638,6 @@ func QueryOneTest(
db, closer := newDBAdapter(t)
defer closer.Close()
ctx := context.Background()
_, err := db.ExecContext(ctx, `INSERT INTO users (name, age, address) VALUES ('Andréa Sá', 0, '{"country":"US"}')`)
tt.AssertNoErr(t, err)
@ -652,8 +665,6 @@ func QueryOneTest(
db, closer := newDBAdapter(t)
defer closer.Close()
ctx := context.Background()
_, err := db.ExecContext(ctx, `INSERT INTO users (name, age, address) VALUES ('João Ribeiro', 0, '{"country":"US"}')`)
tt.AssertNoErr(t, err)
var joao user
@ -683,7 +694,6 @@ func QueryOneTest(
db, closer := newDBAdapter(t)
defer closer.Close()
ctx := context.Background()
_, err := db.ExecContext(ctx, `INSERT INTO users (name, age, address) VALUES ('Count Olivia', 0, '{"country":"US"}')`)
tt.AssertNoErr(t, err)
@ -703,8 +713,6 @@ func QueryOneTest(
db, closer := newDBAdapter(t)
defer closer.Close()
ctx := context.Background()
_, err := db.ExecContext(ctx, `INSERT INTO users (name, age, address) VALUES ('Andréa Sá', 0, '{"country":"US"}')`)
tt.AssertNoErr(t, err)
@ -724,7 +732,6 @@ func QueryOneTest(
db, closer := newDBAdapter(t)
defer closer.Close()
ctx := context.Background()
c := newTestDB(db, driver)
var u *user
err := c.QueryOne(ctx, u, `SELECT * FROM users`)
@ -735,7 +742,6 @@ func QueryOneTest(
db, closer := newDBAdapter(t)
defer closer.Close()
ctx := context.Background()
c := newTestDB(db, driver)
var u user
err := c.QueryOne(ctx, &u, `SELECT * FROM not a valid query`)
@ -746,7 +752,6 @@ func QueryOneTest(
db, closer := newDBAdapter(t)
defer closer.Close()
ctx := context.Background()
c := newTestDB(db, driver)
var row struct {
User user `tablename:"users"`
@ -760,7 +765,6 @@ func QueryOneTest(
db, closer := newDBAdapter(t)
defer closer.Close()
ctx := context.Background()
_, err := db.ExecContext(ctx, `INSERT INTO users (name, age, address) VALUES ('Olivia', 0, '{"country":"US"}')`)
tt.AssertNoErr(t, err)
@ -772,6 +776,20 @@ func QueryOneTest(
err = c.QueryOne(ctx, &row, `SELECT count(*) as my_count FROM users`)
tt.AssertErrContains(t, err, "unexported", "my_count")
})
t.Run("should report error context.Canceled the context has been canceled", func(t *testing.T) {
db, closer := newDBAdapter(t)
defer closer.Close()
ctx, cancel := context.WithCancel(ctx)
cancel()
c := newTestDB(db, driver)
var u user
err := c.QueryOne(ctx, &u, `SELECT * FROM users LIMIT 1`)
tt.AssertEqual(t, errors.Is(err, context.Canceled), true)
})
})
}
@ -783,6 +801,8 @@ func InsertTest(
connStr string,
newDBAdapter func(t *testing.T) (DBAdapter, io.Closer),
) {
ctx := context.Background()
t.Run("Insert", func(t *testing.T) {
t.Run("success cases", func(t *testing.T) {
t.Run("single primary key tables", func(t *testing.T) {
@ -795,7 +815,6 @@ func InsertTest(
db, closer := newDBAdapter(t)
defer closer.Close()
ctx := context.Background()
c := newTestDB(db, driver)
u := user{
@ -828,7 +847,6 @@ func InsertTest(
db, closer := newDBAdapter(t)
defer closer.Close()
ctx := context.Background()
c := newTestDB(db, driver)
u := user{
@ -856,7 +874,6 @@ func InsertTest(
db, closer := newDBAdapter(t)
defer closer.Close()
ctx := context.Background()
c := newTestDB(db, driver)
err = c.Insert(ctx, usersTable, &struct {
ID int `ksql:"id"`
@ -870,7 +887,6 @@ func InsertTest(
db, closer := newDBAdapter(t)
defer closer.Close()
ctx := context.Background()
c := newTestDB(db, driver)
usersByName := NewTable("users", "name")
@ -891,7 +907,6 @@ func InsertTest(
db, closer := newDBAdapter(t)
defer closer.Close()
ctx := context.Background()
c := newTestDB(db, driver)
type tsUser struct {
@ -925,7 +940,6 @@ func InsertTest(
db, closer := newDBAdapter(t)
defer closer.Close()
ctx := context.Background()
c := newTestDB(db, driver)
table := NewTable("user_permissions", "id", "user_id", "perm_id")
@ -946,7 +960,6 @@ func InsertTest(
db, closer := newDBAdapter(t)
defer closer.Close()
ctx := context.Background()
c := newTestDB(db, driver)
// Table defined with 3 values, but we'll provide only 2,
@ -983,7 +996,6 @@ func InsertTest(
db, closer := newDBAdapter(t)
defer closer.Close()
ctx := context.Background()
c := newTestDB(db, driver)
// Table defined with 3 values, but we'll provide only 2,
@ -1039,7 +1051,6 @@ func InsertTest(
db, closer := newDBAdapter(t)
defer closer.Close()
ctx := context.Background()
c := newTestDB(db, driver)
err = c.Insert(ctx, usersTable, "foo")
@ -1070,7 +1081,6 @@ func InsertTest(
db, closer := newDBAdapter(t)
defer closer.Close()
ctx := context.Background()
c := newTestDB(db, driver)
// This is an invalid value:
@ -1084,7 +1094,6 @@ func InsertTest(
db, closer := newDBAdapter(t)
defer closer.Close()
ctx := context.Background()
c := newTestDB(db, driver)
var u *user
@ -1096,7 +1105,6 @@ func InsertTest(
db, closer := newDBAdapter(t)
defer closer.Close()
ctx := context.Background()
c := newTestDB(db, driver)
err := c.Insert(ctx, NewTable("users", ""), &user{Name: "fake-name"})
@ -1107,7 +1115,6 @@ func InsertTest(
db, closer := newDBAdapter(t)
defer closer.Close()
ctx := context.Background()
c := newTestDB(db, driver)
err := c.Insert(ctx, NewTable("", "id"), &user{Name: "fake-name"})
@ -1118,7 +1125,6 @@ func InsertTest(
db, closer := newDBAdapter(t)
defer closer.Close()
ctx := context.Background()
c := newTestDB(db, driver)
err = c.Insert(ctx, usersTable, &struct {
@ -1134,7 +1140,6 @@ func InsertTest(
db, closer := newDBAdapter(t)
defer closer.Close()
ctx := context.Background()
c := newTestDB(db, driver)
brokenTable := NewTable("users", "non_existing_id")
@ -1150,7 +1155,6 @@ func InsertTest(
db, closer := newDBAdapter(t)
defer closer.Close()
ctx := context.Background()
c := newTestDB(db, driver)
err = c.Insert(ctx, usersTable, &struct {
@ -1165,6 +1169,20 @@ func InsertTest(
tt.AssertNotEqual(t, u.ID, uint(0))
tt.AssertEqual(t, u.Age, 42)
})
t.Run("should report error context.Canceled the context has been canceled", func(t *testing.T) {
db, closer := newDBAdapter(t)
defer closer.Close()
ctx, cancel := context.WithCancel(ctx)
cancel()
c := newTestDB(db, driver)
err = c.Insert(ctx, usersTable, &user{Age: 42, Name: "FakeUserName"})
tt.AssertEqual(t, errors.Is(err, context.Canceled), true)
})
})
})
}
@ -1195,6 +1213,8 @@ func DeleteTest(
connStr string,
newDBAdapter func(t *testing.T) (DBAdapter, io.Closer),
) {
ctx := context.Background()
t.Run("Delete", func(t *testing.T) {
err := createTables(driver, connStr)
if err != nil {
@ -1231,7 +1251,6 @@ func DeleteTest(
db, closer := newDBAdapter(t)
defer closer.Close()
ctx := context.Background()
c := newTestDB(db, driver)
u1 := user{
@ -1282,7 +1301,6 @@ func DeleteTest(
db, closer := newDBAdapter(t)
defer closer.Close()
ctx := context.Background()
c := newTestDB(db, driver)
// This permission should not be deleted, we'll use the id to check it:
@ -1318,7 +1336,6 @@ func DeleteTest(
db, closer := newDBAdapter(t)
defer closer.Close()
ctx := context.Background()
c := newTestDB(db, driver)
// This permission should not be deleted, we'll use the id to check it:
@ -1358,7 +1375,6 @@ func DeleteTest(
db, closer := newDBAdapter(t)
defer closer.Close()
ctx := context.Background()
c := newTestDB(db, driver)
err = c.Delete(ctx, usersTable, 4200)
@ -1369,7 +1385,6 @@ func DeleteTest(
db, closer := newDBAdapter(t)
defer closer.Close()
ctx := context.Background()
c := newTestDB(db, driver)
var u *user
@ -1383,7 +1398,6 @@ func DeleteTest(
db, closer := newDBAdapter(t)
defer closer.Close()
ctx := context.Background()
c := newTestDB(db, driver)
err := c.Delete(ctx, NewTable("users", "id"), &struct {
@ -1397,7 +1411,6 @@ func DeleteTest(
db, closer := newDBAdapter(t)
defer closer.Close()
ctx := context.Background()
c := newTestDB(db, driver)
err := c.Delete(ctx, NewTable("users", "id"), &struct {
@ -1412,7 +1425,6 @@ func DeleteTest(
db, closer := newDBAdapter(t)
defer closer.Close()
ctx := context.Background()
c := newTestDB(db, driver)
err := c.Delete(ctx, NewTable("users", "id"), &struct {
@ -1429,7 +1441,6 @@ func DeleteTest(
db, closer := newDBAdapter(t)
defer closer.Close()
ctx := context.Background()
c := newTestDB(db, driver)
err := c.Delete(ctx, userPermissionsTable, &struct {
@ -1447,7 +1458,6 @@ func DeleteTest(
db, closer := newDBAdapter(t)
defer closer.Close()
ctx := context.Background()
c := newTestDB(db, driver)
err := c.Delete(ctx, userPermissionsTable, map[string]interface{}{
@ -1462,7 +1472,6 @@ func DeleteTest(
db, closer := newDBAdapter(t)
defer closer.Close()
ctx := context.Background()
c := newTestDB(db, driver)
err := c.Delete(ctx, userPermissionsTable, &struct {
@ -1482,7 +1491,6 @@ func DeleteTest(
db, closer := newDBAdapter(t)
defer closer.Close()
ctx := context.Background()
c := newTestDB(db, driver)
err := c.Delete(ctx, userPermissionsTable, map[string]interface{}{
@ -1498,7 +1506,6 @@ func DeleteTest(
db, closer := newDBAdapter(t)
defer closer.Close()
ctx := context.Background()
c := newTestDB(db, driver)
err := c.Delete(ctx, userPermissionsTable, &struct {
@ -1518,7 +1525,6 @@ func DeleteTest(
db, closer := newDBAdapter(t)
defer closer.Close()
ctx := context.Background()
c := newTestDB(db, driver)
err := c.Delete(ctx, userPermissionsTable, map[string]interface{}{
@ -1536,7 +1542,6 @@ func DeleteTest(
db, closer := newDBAdapter(t)
defer closer.Close()
ctx := context.Background()
c := newTestDB(db, driver)
err := c.Delete(ctx, NewTable("users", ""), &user{ID: 42, Name: "fake-name"})
@ -1547,12 +1552,24 @@ func DeleteTest(
db, closer := newDBAdapter(t)
defer closer.Close()
ctx := context.Background()
c := newTestDB(db, driver)
err := c.Delete(ctx, NewTable("", "id"), &user{Name: "fake-name"})
tt.AssertErrContains(t, err, "ksql.Table", "table name", "empty string")
})
t.Run("should report error context.Canceled the context has been canceled", func(t *testing.T) {
db, closer := newDBAdapter(t)
defer closer.Close()
ctx, cancel := context.WithCancel(ctx)
cancel()
c := newTestDB(db, driver)
err := c.Delete(ctx, usersTable, &user{ID: 42, Name: "fake-name"})
tt.AssertEqual(t, errors.Is(err, context.Canceled), true)
})
})
}
@ -1564,6 +1581,8 @@ func PatchTest(
connStr string,
newDBAdapter func(t *testing.T) (DBAdapter, io.Closer),
) {
ctx := context.Background()
t.Run("Patch", func(t *testing.T) {
err := createTables(driver, connStr)
if err != nil {
@ -1574,7 +1593,6 @@ func PatchTest(
db, closer := newDBAdapter(t)
defer closer.Close()
ctx := context.Background()
c := newTestDB(db, driver)
u := user{
@ -1603,7 +1621,6 @@ func PatchTest(
db, closer := newDBAdapter(t)
defer closer.Close()
ctx := context.Background()
c := newTestDB(db, driver)
u := user{
@ -1632,7 +1649,6 @@ func PatchTest(
db, closer := newDBAdapter(t)
defer closer.Close()
ctx := context.Background()
c := newTestDB(db, driver)
err = createUserPermission(db, c.dialect, userPermission{
@ -1669,7 +1685,6 @@ func PatchTest(
db, closer := newDBAdapter(t)
defer closer.Close()
ctx := context.Background()
c := newTestDB(db, driver)
type partialUser struct {
@ -1706,7 +1721,6 @@ func PatchTest(
db, closer := newDBAdapter(t)
defer closer.Close()
ctx := context.Background()
c := newTestDB(db, driver)
type partialUser struct {
@ -1743,7 +1757,6 @@ func PatchTest(
db, closer := newDBAdapter(t)
defer closer.Close()
ctx := context.Background()
c := newTestDB(db, driver)
err = c.Update(ctx, usersTable, user{
@ -1757,7 +1770,6 @@ func PatchTest(
db, closer := newDBAdapter(t)
defer closer.Close()
ctx := context.Background()
c := newTestDB(db, driver)
err = c.Update(ctx, NewTable("non_existing_table"), user{
@ -1771,7 +1783,6 @@ func PatchTest(
db, closer := newDBAdapter(t)
defer closer.Close()
ctx := context.Background()
c := newTestDB(db, driver)
var u *user
@ -1783,7 +1794,6 @@ func PatchTest(
db, closer := newDBAdapter(t)
defer closer.Close()
ctx := context.Background()
c := newTestDB(db, driver)
err = c.Update(ctx, usersTable, struct {
@ -1802,7 +1812,6 @@ func PatchTest(
db, closer := newDBAdapter(t)
defer closer.Close()
ctx := context.Background()
c := newTestDB(db, driver)
err := c.Update(ctx, usersTable, &user{
@ -1816,7 +1825,6 @@ func PatchTest(
db, closer := newDBAdapter(t)
defer closer.Close()
ctx := context.Background()
c := newTestDB(db, driver)
err := c.Update(ctx, NewTable("user_permissions", "id", "user_id", "perm_id"), &userPermission{
@ -1827,6 +1835,22 @@ func PatchTest(
tt.AssertErrContains(t, err, "invalid value", "0", "'user_id'")
})
})
t.Run("should report error context.Canceled the context has been canceled", func(t *testing.T) {
db, closer := newDBAdapter(t)
defer closer.Close()
ctx, cancel := context.WithCancel(ctx)
cancel()
c := newTestDB(db, driver)
err = c.Update(ctx, usersTable, user{
ID: 1,
Name: "Thayane",
})
tt.AssertEqual(t, errors.Is(err, context.Canceled), true)
})
})
}
@ -1838,6 +1862,8 @@ func QueryChunksTest(
connStr string,
newDBAdapter func(t *testing.T) (DBAdapter, io.Closer),
) {
ctx := context.Background()
t.Run("QueryChunks", func(t *testing.T) {
variations := []struct {
desc string
@ -1863,7 +1889,6 @@ func QueryChunksTest(
db, closer := newDBAdapter(t)
defer closer.Close()
ctx := context.Background()
c := newTestDB(db, driver)
_ = c.Insert(ctx, usersTable, &user{
@ -1903,7 +1928,6 @@ func QueryChunksTest(
db, closer := newDBAdapter(t)
defer closer.Close()
ctx := context.Background()
c := newTestDB(db, driver)
_ = c.Insert(ctx, usersTable, &user{Name: "User1", Address: address{Country: "US"}})
@ -1945,7 +1969,6 @@ func QueryChunksTest(
db, closer := newDBAdapter(t)
defer closer.Close()
ctx := context.Background()
c := newTestDB(db, driver)
_ = c.Insert(ctx, usersTable, &user{Name: "User1", Address: address{Country: "US"}})
@ -1987,7 +2010,6 @@ func QueryChunksTest(
db, closer := newDBAdapter(t)
defer closer.Close()
ctx := context.Background()
c := newTestDB(db, driver)
_ = c.Insert(ctx, usersTable, &user{Name: "User1"})
@ -2038,7 +2060,6 @@ func QueryChunksTest(
Age: 20,
}
ctx := context.Background()
c := newTestDB(db, driver)
_ = c.Insert(ctx, usersTable, &joao)
_ = c.Insert(ctx, usersTable, &thatiana)
@ -2100,7 +2121,6 @@ func QueryChunksTest(
db, closer := newDBAdapter(t)
defer closer.Close()
ctx := context.Background()
c := newTestDB(db, driver)
_ = c.Insert(ctx, usersTable, &user{Name: "User1"})
@ -2139,7 +2159,6 @@ func QueryChunksTest(
db, closer := newDBAdapter(t)
defer closer.Close()
ctx := context.Background()
c := newTestDB(db, driver)
_ = c.Insert(ctx, usersTable, &user{Name: "User1"})
@ -2182,7 +2201,6 @@ func QueryChunksTest(
db, closer := newDBAdapter(t)
defer closer.Close()
ctx := context.Background()
c := newTestDB(db, driver)
_ = c.Insert(ctx, usersTable, &user{Name: "User1"})
@ -2221,7 +2239,6 @@ func QueryChunksTest(
db, closer := newDBAdapter(t)
defer closer.Close()
ctx := context.Background()
c := newTestDB(db, driver)
_ = c.Insert(ctx, usersTable, &user{Name: "User1"})
@ -2259,7 +2276,6 @@ func QueryChunksTest(
db, closer := newDBAdapter(t)
defer closer.Close()
ctx := context.Background()
c := newTestDB(db, driver)
funcs := []interface{}{
@ -2303,7 +2319,6 @@ func QueryChunksTest(
db, closer := newDBAdapter(t)
defer closer.Close()
ctx := context.Background()
c := newTestDB(db, driver)
err := c.QueryChunks(ctx, ChunkParser{
Query: `SELECT * FROM not a valid query`,
@ -2321,7 +2336,6 @@ func QueryChunksTest(
db, closer := newDBAdapter(t)
defer closer.Close()
ctx := context.Background()
c := newTestDB(db, driver)
err := c.QueryChunks(ctx, ChunkParser{
@ -2341,6 +2355,29 @@ func QueryChunksTest(
})
})
}
t.Run("error cases", func(t *testing.T) {
t.Run("should report error context.Canceled the context has been canceled", func(t *testing.T) {
db, closer := newDBAdapter(t)
defer closer.Close()
ctx, cancel := context.WithCancel(ctx)
cancel()
c := newTestDB(db, driver)
err := c.QueryChunks(ctx, ChunkParser{
Query: `SELECT * FROM users u JOIN posts p ON u.id = p.user_id`,
Params: []interface{}{},
ChunkSize: 2,
ForEachChunk: func(rows []user) error {
return nil
},
})
tt.AssertEqual(t, errors.Is(err, context.Canceled), true)
})
})
})
}
@ -2352,6 +2389,8 @@ func TransactionTest(
connStr string,
newDBAdapter func(t *testing.T) (DBAdapter, io.Closer),
) {
ctx := context.Background()
t.Run("Transaction", func(t *testing.T) {
t.Run("should query a single row correctly", func(t *testing.T) {
err := createTables(driver, connStr)
@ -2362,7 +2401,6 @@ func TransactionTest(
db, closer := newDBAdapter(t)
defer closer.Close()
ctx := context.Background()
c := newTestDB(db, driver)
_ = c.Insert(ctx, usersTable, &user{Name: "User1"})
@ -2389,7 +2427,6 @@ func TransactionTest(
db, closer := newDBAdapter(t)
defer closer.Close()
ctx := context.Background()
c := newTestDB(db, driver)
u := user{
@ -2426,7 +2463,6 @@ func TransactionTest(
db, closer := newDBAdapter(t)
defer closer.Close()
ctx := context.Background()
c := newTestDB(db, driver)
u1 := user{Name: "User1", Age: 42}
@ -2463,7 +2499,6 @@ func TransactionTest(
db, closer := newDBAdapter(t)
defer closer.Close()
ctx := context.Background()
c := newTestDB(db, driver)
u1 := user{Name: "User1", Age: 42}
@ -2501,7 +2536,6 @@ func TransactionTest(
db, closer := newDBAdapter(t)
defer closer.Close()
ctx := context.Background()
c := newTestDB(db, driver)
cMock := mockTxBeginner{
@ -2536,7 +2570,6 @@ func TransactionTest(
db, closer := newDBAdapter(t)
defer closer.Close()
ctx := context.Background()
c := newTestDB(db, driver)
cMock := mockTxBeginner{
@ -2562,7 +2595,6 @@ func TransactionTest(
db, closer := newDBAdapter(t)
defer closer.Close()
ctx := context.Background()
c := newTestDB(db, driver)
cMock := mockTxBeginner{
@ -2588,7 +2620,6 @@ func TransactionTest(
db, closer := newDBAdapter(t)
defer closer.Close()
ctx := context.Background()
c := newTestDB(db, driver)
c.db = mockDBAdapter{}
@ -2598,6 +2629,21 @@ func TransactionTest(
})
tt.AssertErrContains(t, err, "KSQL", "can't start transaction", "DBAdapter", "TxBeginner")
})
t.Run("should report error context.Canceled the context has been canceled", func(t *testing.T) {
db, closer := newDBAdapter(t)
defer closer.Close()
ctx, cancel := context.WithCancel(ctx)
cancel()
c := newTestDB(db, driver)
err := c.Transaction(ctx, func(db Provider) error {
return nil
})
tt.AssertEqual(t, errors.Is(err, context.Canceled), true)
})
})
}