1 files changed, 22 insertions, 0 deletions
diff --git a/internal/praefect/datastore/init_test.go b/internal/praefect/datastore/init_test.go
new file mode 100644
index 000000000..bfbffa166
--- /dev/null
+++ b/internal/praefect/datastore/init_test.go
@@ -0,0 +1,22 @@
+// +build postgres
+
+package datastore
+
+import (
+	"log"
+	"os"
+	"testing"
+
+	"gitlab.com/gitlab-org/gitaly/internal/praefect/datastore/glsql"
+)
+
+func TestMain(m *testing.M) {
+	code := m.Run()
+	// Clean closes connection to database once all tests are done
+	if err := glsql.Clean(); err != nil {
+		log.Fatalln(err, "database disconnection failure")
+	}
+	os.Exit(code)
+}
+
+func getDB(t testing.TB) glsql.DB { return glsql.GetDB(t, "datastore") }