internal/praefect/datastore/init_test.go


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27

// +build postgres

package datastore

import (
	"log"
	"os"
	"testing"

	"gitlab.com/gitlab-org/gitaly/v14/internal/praefect/config"
	"gitlab.com/gitlab-org/gitaly/v14/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)
}

const databaseName = "datastore"

func getDB(t testing.TB) glsql.DB { return glsql.GetDB(t, databaseName) }

func getDBConfig(t testing.TB) config.DB { return glsql.GetDBConfig(t, databaseName) }