diff options
| author |  Tobi Smethurst <31960611+tsmethurst@users.noreply.github.com> | 2021-08-12 21:03:24 +0200 |
|---|---|---|
| committer |  GitHub <noreply@github.com> | 2021-08-12 21:03:24 +0200 |
| commit | 98263a7de64269898a2f81207e38943b5c8e8653 (patch) | |
| tree | 743c90f109a6c5d27832d1dcef2388d939f0f77a /testrig/db.go | |
| parent | Text duplication fix (#137) (diff) | |
| download | gotosocial-98263a7de64269898a2f81207e38943b5c8e8653.tar.xz | |
Grand test fixup (#138)
* start fixing up tests
* fix up tests + automate with drone
* fiddle with linting
* messing about with drone.yml
* some more fiddling
* hmmm
* add cache
* add vendor directory
* verbose
* ci updates
* update some little things
* update sig
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) |