Backend: Close db connection after running tests

Signed-off-by: Michael Mayer <michael@liquidbytes.net>

internal/api/api_test.go

@@ -3,11 +3,14 @@ package api
 import (
 	"net/http"
 	"net/http/httptest"
+	"os"
 	"strings"
+	"testing"
 
 	"github.com/gin-gonic/gin"
 	"github.com/photoprism/photoprism/internal/config"
 	"github.com/photoprism/photoprism/internal/service"
+	"github.com/sirupsen/logrus"
 )
 
 // NewApiTest returns new API test helper
@@ -37,3 +40,16 @@ func PerformRequestWithBody(r http.Handler, method, path, body string) *httptest
 	r.ServeHTTP(w, req)
 	return w
 }
+
+func TestMain(m *testing.M) {
+	log = logrus.StandardLogger()
+	log.SetLevel(logrus.DebugLevel)
+
+	c := config.TestConfig()
+
+	code := m.Run()
+
+	_ = c.CloseDb()
+
+	os.Exit(code)
+}

internal/config/config_test.go

@@ -1,13 +1,28 @@
 package config
 
 import (
+	"os"
 	"strings"
 	"testing"
 
 	"github.com/photoprism/photoprism/pkg/fs"
+	"github.com/sirupsen/logrus"
 	"github.com/stretchr/testify/assert"
 )
 
+func TestMain(m *testing.M) {
+	log = logrus.StandardLogger()
+	log.SetLevel(logrus.DebugLevel)
+
+	c := TestConfig()
+
+	code := m.Run()
+
+	_ = c.CloseDb()
+
+	os.Exit(code)
+}
+
 func TestNewConfig(t *testing.T) {
 	ctx := CliTestContext()
 

internal/photoprism/photoprism_test.go

@@ -4,7 +4,7 @@ import (
 	"os"
 	"testing"
 
-	"github.com/photoprism/photoprism/internal/entity"
+	"github.com/photoprism/photoprism/internal/config"
 	"github.com/sirupsen/logrus"
 )
 
@@ -12,13 +12,11 @@ func TestMain(m *testing.M) {
 	log = logrus.StandardLogger()
 	log.SetLevel(logrus.DebugLevel)
 
-	db := entity.InitTestDb(os.Getenv("PHOTOPRISM_TEST_DSN"))
+	c := config.TestConfig()
 
 	code := m.Run()
 
-	if db != nil {
-		db.Close()
-	}
+	_ = c.CloseDb()
 
 	os.Exit(code)
 }

internal/service/service_test.go

@@ -21,6 +21,8 @@ func TestMain(m *testing.M) {
 
 	code := m.Run()
 
+	_ = c.CloseDb()
+
 	os.Exit(code)
 }