7
7
"database/sql"
8
8
"fmt"
9
9
"github.com/golang-migrate/migrate/v4"
10
+ "log"
10
11
"strings"
11
12
"testing"
12
13
)
@@ -38,20 +39,22 @@ var (
38
39
func isReady (ctx context.Context , c dktest.ContainerInfo ) bool {
39
40
ip , port , err := c .Port (defaultPort )
40
41
if err != nil {
41
- fmt .Println ("port error:" , err )
42
+ log .Println ("port error:" , err )
42
43
return false
43
44
}
44
45
45
46
db , err := sql .Open ("postgres" , fmt .Sprintf ("postgres://root@%v:%v?sslmode=disable" , ip , port ))
46
47
if err != nil {
47
- fmt .Println ("open error:" , err )
48
+ log .Println ("open error:" , err )
48
49
return false
49
50
}
50
51
if err := db .PingContext (ctx ); err != nil {
51
- fmt .Println ("ping error:" , err )
52
+ log .Println ("ping error:" , err )
52
53
return false
53
54
}
54
- db .Close ()
55
+ if err := db .Close (); err != nil {
56
+ log .Println ("close error:" , err )
57
+ }
55
58
return true
56
59
}
57
60
@@ -68,7 +71,11 @@ func createDB(t *testing.T, c dktest.ContainerInfo) {
68
71
if err = db .Ping (); err != nil {
69
72
t .Fatal (err )
70
73
}
71
- defer db .Close ()
74
+ defer func () {
75
+ if err := db .Close (); err != nil {
76
+ t .Error (err )
77
+ }
78
+ }()
72
79
73
80
if _ , err = db .Exec ("CREATE DATABASE migrate" ); err != nil {
74
81
t .Fatal (err )
@@ -88,7 +95,7 @@ func Test(t *testing.T) {
88
95
c := & CockroachDb {}
89
96
d , err := c .Open (addr )
90
97
if err != nil {
91
- t .Fatalf ( "%v" , err )
98
+ t .Fatal ( err )
92
99
}
93
100
dt .Test (t , d , []byte ("SELECT 1" ))
94
101
})
@@ -107,12 +114,12 @@ func TestMigrate(t *testing.T) {
107
114
c := & CockroachDb {}
108
115
d , err := c .Open (addr )
109
116
if err != nil {
110
- t .Fatalf ( "%v" , err )
117
+ t .Fatal ( err )
111
118
}
112
119
113
120
m , err := migrate .NewWithDatabaseInstance ("file://./examples/migrations" , "migrate" , d )
114
121
if err != nil {
115
- t .Fatalf ( "%v" , err )
122
+ t .Fatal ( err )
116
123
}
117
124
dt .TestMigrate (t , m , []byte ("SELECT 1" ))
118
125
})
@@ -131,7 +138,7 @@ func TestMultiStatement(t *testing.T) {
131
138
c := & CockroachDb {}
132
139
d , err := c .Open (addr )
133
140
if err != nil {
134
- t .Fatalf ( "%v" , err )
141
+ t .Fatal ( err )
135
142
}
136
143
if err := d .Run (strings .NewReader ("CREATE TABLE foo (foo text); CREATE TABLE bar (bar text);" )); err != nil {
137
144
t .Fatalf ("expected err to be nil, got %v" , err )
@@ -161,7 +168,7 @@ func TestFilterCustomQuery(t *testing.T) {
161
168
c := & CockroachDb {}
162
169
_ , err = c .Open (addr )
163
170
if err != nil {
164
- t .Fatalf ( "%v" , err )
171
+ t .Fatal ( err )
165
172
}
166
173
})
167
174
}
0 commit comments