diff --git a/internal/appliance/upgrades/BUILD.bazel b/internal/appliance/upgrades/BUILD.bazel index 2fe635e7253..8abee4ce101 100644 --- a/internal/appliance/upgrades/BUILD.bazel +++ b/internal/appliance/upgrades/BUILD.bazel @@ -23,5 +23,8 @@ go_test( srcs = ["util_test.go"], embed = [":upgrades"], tags = ["requires-network"], - deps = ["//internal/observation"], + deps = [ + "//internal/database/dbtest", + "//internal/observation", + ], ) diff --git a/internal/appliance/upgrades/util_test.go b/internal/appliance/upgrades/util_test.go index 429a3b43fb0..6aa20e20435 100644 --- a/internal/appliance/upgrades/util_test.go +++ b/internal/appliance/upgrades/util_test.go @@ -4,6 +4,7 @@ import ( "os" "testing" + "github.com/sourcegraph/sourcegraph/internal/database/dbtest" "github.com/sourcegraph/sourcegraph/internal/observation" ) @@ -95,8 +96,22 @@ func TestCheckConnection_Ping(t *testing.T) { for _, test := range tests { t.Run(test.name, func(t *testing.T) { + db := dbtest.NewDB(t) + defer db.Close() + + var dbName string + 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("PGDATABASE", dbName) + t.Setenv("PGSSLMODE", "disable") + t.Setenv("PGTZ", "UTC") dsns, err := getApplianceDSNs() if err != nil {