1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
|
package query_test
import (
"context"
"database/sql"
"errors"
"testing"
"github.com/stretchr/testify/assert"
"github.com/stretchr/testify/require"
"github.com/lxc/incus/v6/internal/server/db/query"
)
// Any error happening when beginning the transaction will be propagated.
func TestTransaction_BeginError(t *testing.T) {
db := newDB(t)
err := db.Close()
require.NoError(t, err)
err = query.Transaction(context.TODO(), db, func(ctx context.Context, tx *sql.Tx) error { return nil })
assert.NotNil(t, err)
assert.Contains(t, err.Error(), "Failed to begin transaction")
}
// Any error happening when in the transaction function will cause a rollback.
func TestTransaction_FunctionError(t *testing.T) {
db := newDB(t)
err := query.Transaction(context.TODO(), db, func(ctx context.Context, tx *sql.Tx) error {
_, err := tx.Exec("CREATE TABLE test (id INTEGER)")
assert.NoError(t, err)
return errors.New("boom")
})
assert.EqualError(t, err, "boom")
tx, err := db.Begin()
assert.NoError(t, err)
tables, err := query.SelectStrings(context.Background(), tx, "SELECT name FROM sqlite_master WHERE type = 'table'")
assert.NoError(t, err)
assert.NotContains(t, tables, "test")
}
// Return a new in-memory SQLite database.
func newDB(t *testing.T) *sql.DB {
db, err := sql.Open("sqlite3", ":memory:")
assert.NoError(t, err)
return db
}
|
