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
|
package glsql
import (
"testing"
"github.com/stretchr/testify/require"
)
func TestOpenDB(t *testing.T) {
dbCfg := GetDBConfig(t, "postgres")
t.Run("failed to ping because of incorrect config", func(t *testing.T) {
badCfg := dbCfg
badCfg.Host = "not-existing.com"
_, err := OpenDB(badCfg)
require.Error(t, err, "opening of DB with incorrect configuration must fail")
})
t.Run("connected with proper config", func(t *testing.T) {
db, err := OpenDB(dbCfg)
require.NoError(t, err, "opening of DB with correct configuration must not fail")
require.NoError(t, db.Close())
})
}
func TestUint64Provider(t *testing.T) {
var provider Uint64Provider
dst1 := provider.To()
require.Equal(t, []interface{}{new(uint64)}, dst1, "must be a single value holder")
val1 := dst1[0].(*uint64)
*val1 = uint64(100)
dst2 := provider.To()
require.Equal(t, []interface{}{new(uint64)}, dst2, "must be a single value holder")
val2 := dst2[0].(*uint64)
*val2 = uint64(200)
require.Equal(t, []uint64{100, 200}, provider.Values())
dst3 := provider.To()
val3 := dst3[0].(*uint64)
*val3 = uint64(300)
require.Equal(t, []uint64{100, 200, 300}, provider.Values())
}
func TestScanAll(t *testing.T) {
t.Parallel()
db := NewDB(t)
var ids Uint64Provider
notEmptyRows, err := db.Query("SELECT id FROM (VALUES (1), (200), (300500)) AS t(id)")
require.NoError(t, err)
require.NoError(t, ScanAll(notEmptyRows, &ids))
require.Equal(t, []uint64{1, 200, 300500}, ids.Values())
var nothing Uint64Provider
emptyRows, err := db.Query("SELECT id FROM (VALUES (1), (200), (300500)) AS t(id) WHERE id < 0")
require.NoError(t, err)
require.NoError(t, ScanAll(emptyRows, ¬hing))
require.Equal(t, ([]uint64)(nil), nothing.Values())
}
|
