74
75
// NewPostgreSQLConfig returns a new MySQL config for test
76
>
func NewPostgreSQLConfig(pluginName string, connectAttrs map[string]string) *config.SQL {
postgresql_test_util.go
77
>
return &config.SQL{
78
>
User: testPostgreSQLUser,
79
>
Password: testPostgreSQLPassword,
80
>
ConnectAddr: net.JoinHostPort(
81
>
environment.GetPostgreSQLAddress(),
82
>
strconv.Itoa(environment.GetPostgreSQLPort()),
83
>
),
84
>
ConnectProtocol: testPostgreSQLConnectionProtocol,
85
>
PluginName: pluginName,
86
>
DatabaseName: testPostgreSQLDatabaseNamePrefix + shuffle.String(testPostgreSQLDatabaseNameSuffix),
87
>
ConnectAttributes: connectAttrs,
88
>
}
89
>
}
90
92
>
adminCfg := *cfg
93
>
// NOTE need to connect with empty name to create new database
94
>
adminCfg.DatabaseName = ""
95
>
96
>
db, err := sql.NewSQLAdminDB(sqlplugin.DbKindUnknown, &adminCfg, resolver.NewNoopResolver(), log.NewTestLogger(), metrics.NoopMetricsHandler)
97
>
if err != nil {
98
t.Fatalf("unable to create PostgreSQL admin DB: %v", err)
99
}
101
103
>
if err != nil {
104
t.Fatalf("unable to create PostgreSQL database: %v", err)
105
}
106
}
107
109
>
db, err := sql.NewSQLAdminDB(sqlplugin.DbKindUnknown, cfg, resolver.NewNoopResolver(), log.NewTestLogger(), metrics.NoopMetricsHandler)
110
>
if err != nil {
111
t.Fatalf("unable to create PostgreSQL admin DB: %v", err)
112
}
114
116
>
if err != nil {
117
t.Fatal(err)
118
}
119
121
>
if err != nil {
122
t.Fatal(err)
123
}
124
126
>
if err = db.Exec(stmt); err != nil {
127
t.Fatal(err)
128
}
129
}
130
132
>
if err != nil {
133
t.Fatal(err)
134
}
135
137
>
if err != nil {
138
t.Fatal(err)
139
}
140
142
>
if err = db.Exec(stmt); err != nil {
143
t.Fatal(err)
144
}