-
Notifications
You must be signed in to change notification settings - Fork 11
/
driver_test.go
63 lines (56 loc) · 1.08 KB
/
driver_test.go
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
package bigquery
import (
"database/sql/driver"
"os"
"testing"
)
const mockConnectString = "bigquery://projectID/us/dataset1"
var (
testConnectionString string
)
func init() {
if os.Getenv(ConnectionStringEnvKey) != "" {
testConnectionString = os.Getenv(ConnectionStringEnvKey)
} else {
testConnectionString = mockConnectString
}
}
func TestDriver_Open(t *testing.T) {
type args struct {
connectionString string
}
tests := []struct {
name string
args args
wantC driver.Conn
wantErr bool
}{
{
name: "OK",
args: args{
connectionString: testConnectionString,
},
wantC: &Conn{
cfg: &Config{
Location: "us",
DatasetID: "dataset1",
ProjectID: "projectid",
},
},
wantErr: false,
},
}
for _, tt := range tests {
t.Run(tt.name, func(t *testing.T) {
d := &Driver{}
gotC, err := d.Open(tt.args.connectionString)
if (err != nil) != tt.wantErr {
t.Errorf("Open() error = %v, wantErr %v", err, tt.wantErr)
return
}
if gotC == nil {
t.Error("Open() failed to return a connection")
}
})
}
}