Provide a way to return config from the test database (#503)

google · May 31, 2020 · 70e20df · 70e20df
1 parent 208bab6
commit 70e20df
Showing 1 changed file with 15 additions and 1 deletion.
diff --git a/internal/database/database_util.go b/internal/database/database_util.go
@@ -43,7 +43,7 @@ import (
 //
 // All database tests can be skipped by running `go test -short` or by setting
 // the `SKIP_DATABASE_TESTS` environment variable.
-func NewTestDatabase(tb testing.TB) *DB {
+func NewTestDatabaseWithConfig(tb testing.TB) (*DB, *Config) {
  tb.Helper()
 
  if testing.Short() {
@@ -126,6 +126,20 @@ func NewTestDatabase(tb testing.TB) *DB {
  db.Close(context.Background())
  })
 
+ return db, &Config{
+ Name: dbname,
+ User: username,
+ Host: container.GetBoundIP("5432/tcp"),
+ Port: container.GetPort("5432/tcp"),
+ SSLMode: "disable",
+ Password: password,
+ }
+}
+
+func NewTestDatabase(tb testing.TB) *DB {
+ tb.Helper()
+
+ db, _ := NewTestDatabaseWithConfig(tb)
  return db
 }