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
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
|
package query_test
import (
"context"
"database/sql"
"testing"
"github.com/stretchr/testify/assert"
"github.com/stretchr/testify/require"
"github.com/lxc/incus/v6/internal/server/db/query"
)
// Exercise possible failure modes.
func TestStrings_Error(t *testing.T) {
for _, c := range testStringsErrorCases {
t.Run(c.query, func(t *testing.T) {
tx := newTxForSlices(t)
values, err := query.SelectStrings(context.Background(), tx, c.query)
assert.EqualError(t, err, c.error)
assert.Nil(t, values)
})
}
}
var testStringsErrorCases = []struct {
query string
error string
}{
{"garbage", "near \"garbage\": syntax error"},
{"SELECT id, name FROM test", "sql: expected 2 destination arguments in Scan, not 1"},
}
// All values yield by the query are returned.
func TestStrings(t *testing.T) {
tx := newTxForSlices(t)
values, err := query.SelectStrings(context.Background(), tx, "SELECT name FROM test ORDER BY name")
assert.Nil(t, err)
assert.Equal(t, []string{"bar", "foo"}, values)
}
// Exercise possible failure modes.
func TestIntegers_Error(t *testing.T) {
for _, c := range testIntegersErrorCases {
t.Run(c.query, func(t *testing.T) {
tx := newTxForSlices(t)
values, err := query.SelectIntegers(context.Background(), tx, c.query)
assert.EqualError(t, err, c.error)
assert.Nil(t, values)
})
}
}
var testIntegersErrorCases = []struct {
query string
error string
}{
{"garbage", "near \"garbage\": syntax error"},
{"SELECT id, name FROM test", "sql: expected 2 destination arguments in Scan, not 1"},
}
// All values yield by the query are returned.
func TestIntegers(t *testing.T) {
tx := newTxForSlices(t)
values, err := query.SelectIntegers(context.Background(), tx, "SELECT id FROM test ORDER BY id")
assert.Nil(t, err)
assert.Equal(t, []int{0, 1}, values)
}
// Insert new rows in bulk.
func TestInsertStrings(t *testing.T) {
tx := newTxForSlices(t)
err := query.InsertStrings(tx, "INSERT INTO test(name) VALUES %s", []string{"xx", "yy"})
require.NoError(t, err)
values, err := query.SelectStrings(context.Background(), tx, "SELECT name FROM test ORDER BY name DESC LIMIT 2")
require.NoError(t, err)
assert.Equal(t, values, []string{"yy", "xx"})
}
// Return a new transaction against an in-memory SQLite database with a single
// test table populated with a few rows.
func newTxForSlices(t *testing.T) *sql.Tx {
db, err := sql.Open("sqlite3", ":memory:")
assert.NoError(t, err)
_, err = db.Exec("CREATE TABLE test (id INTEGER, name TEXT)")
assert.NoError(t, err)
_, err = db.Exec("INSERT INTO test VALUES (0, 'foo'), (1, 'bar')")
assert.NoError(t, err)
tx, err := db.Begin()
assert.NoError(t, err)
return tx
}
|
