diff --git a/coderd/database/dbtestutil/postgres.go b/coderd/database/dbtestutil/postgres.go
index e5aa4b14de83b..1ab80569dedb1 100644
--- a/coderd/database/dbtestutil/postgres.go
+++ b/coderd/database/dbtestutil/postgres.go
@@ -138,6 +138,7 @@ func initDefaultConnection(t TBSubset) error {
type OpenOptions struct {
DBFrom *string
+ LogDSN bool
}
type OpenOption func(*OpenOptions)
@@ -150,9 +151,18 @@ func WithDBFrom(dbFrom string) OpenOption {
}
}
+// WithLogDSN sets whether the DSN should be logged during testing.
+// This provides an ergonomic way to connect to test databases during debugging.
+func WithLogDSN(logDSN bool) OpenOption {
+ return func(o *OpenOptions) {
+ o.LogDSN = logDSN
+ }
+}
+
// TBSubset is a subset of the testing.TB interface.
// It allows to use dbtestutil.Open outside of tests.
type TBSubset interface {
+ Name() string
Cleanup(func())
Helper()
Logf(format string, args ...any)
@@ -227,6 +237,11 @@ func Open(t TBSubset, opts ...OpenOption) (string, error) {
Port: port,
DBName: dbName,
}.DSN()
+
+ // Optionally log the DSN to help connect to the test database.
+ if openOptions.LogDSN {
+ _, _ = fmt.Fprintf(os.Stderr, "Connect to the database for %s using: psql '%s'\n", t.Name(), dsn)
+ }
return dsn, nil
}
diff --git a/coderd/database/gen/dump/main.go b/coderd/database/gen/dump/main.go
index f99b69bdaef93..1d84339eecce9 100644
--- a/coderd/database/gen/dump/main.go
+++ b/coderd/database/gen/dump/main.go
@@ -19,6 +19,10 @@ type mockTB struct {
cleanup []func()
}
+func (*mockTB) Name() string {
+ return "mockTB"
+}
+
func (t *mockTB) Cleanup(f func()) {
t.cleanup = append(t.cleanup, f)
}
pFad - Phonifier reborn
Pfad - The Proxy pFad of © 2024 Garber Painting. All rights reserved.
Note: This service is not intended for secure transactions such as banking, social media, email, or purchasing. Use at your own risk. We assume no liability whatsoever for broken pages.
Alternative Proxies:
Alternative Proxy
pFad Proxy
pFad v3 Proxy
pFad v4 Proxy