handle for different password in CI DSN gen?

This commit is contained in:
Warren Gifford 2024-07-19 14:56:36 -07:00
parent 8872541c2a
commit 85c13639fc

View File

@ -99,16 +99,30 @@ func TestCheckConnection_Ping(t *testing.T) {
db := dbtest.NewDB(t)
defer db.Close()
var currentUser string
err := db.QueryRow("SELECT current_user").Scan(&currentUser)
if err != nil {
t.Errorf("unexpected error: %v", err)
t.FailNow()
}
url, err := dbtest.GetDSN()
if err != nil {
t.Errorf("unexpected error: %v", err)
t.FailNow()
}
password, _ := url.User.Password()
var dbName string
err := db.QueryRow("SELECT current_database()").Scan(&dbName)
err = db.QueryRow("SELECT current_database()").Scan(&dbName)
if err != nil {
t.Errorf("unexpected error: %v", err)
t.FailNow()
}
t.Setenv("CODEINTEL_PG_ALLOW_SINGLE_DB", "true")
t.Setenv("PGUSER", "sourcegraph")
t.Setenv("PGPASSWORD", "sourcegraph")
t.Setenv("PGUSER", currentUser)
t.Setenv("PGPASSWORD", password)
t.Setenv("PGDATABASE", dbName)
t.Setenv("PGSSLMODE", "disable")
t.Setenv("PGTZ", "UTC")