From 98263a7de64269898a2f81207e38943b5c8e8653 Mon Sep 17 00:00:00 2001
From: Tobi Smethurst <31960611+tsmethurst@users.noreply.github.com>
Date: Thu, 12 Aug 2021 21:03:24 +0200
Subject: 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
---
 testrig/db.go | 11 ++++++++++-
 1 file changed, 10 insertions(+), 1 deletion(-)

(limited to 'testrig/db.go')

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)
-- 
cgit v1.2.3