diff options
Diffstat (limited to 'testrig/db.go')
| -rw-r--r-- | testrig/db.go | 11 |
1 files changed, 10 insertions, 1 deletions
diff --git a/testrig/db.go b/testrig/db.go index f34f7936b..659a74ca2 100644 --- a/testrig/db.go +++ b/testrig/db.go @@ -20,6 +20,7 @@ package testrig import ( "context" + "os" "github.com/sirupsen/logrus" "github.com/superseriousbusiness/gotosocial/internal/db" @@ -52,9 +53,17 @@ var testModels []interface{} = []interface{}{ &oauth.Client{}, } -// NewTestDB returns a new initialized, empty database for testing +// NewTestDB returns a new initialized, empty database for testing. +// +// If the environment variable GTS_DB_ADDRESS is set, it will take that +// value as the database address instead. func NewTestDB() db.DB { config := NewTestConfig() + alternateAddress := os.Getenv("GTS_DB_ADDRESS") + if alternateAddress != "" { + config.DBConfig.Address = alternateAddress + } + l := logrus.New() l.SetLevel(logrus.TraceLevel) testDB, err := pg.NewPostgresService(context.Background(), config, l) |