apollo-backend/internal/testhelper/pgxpool.go
André Medeiros f9b9c595cf Better testing (#62)
* some tests

* more tests

* tidy up go.mod

* more tests

* add postgres

* beep

* again

* Set up schema

* fix device test
2022-05-07 12:37:21 -04:00

34 lines
561 B
Go

package testhelper
import (
"context"
"os"
"testing"
"github.com/jackc/pgx/v4"
"github.com/stretchr/testify/require"
)
func NewTestPgxConn(t *testing.T) *pgx.Conn {
t.Helper()
ctx := context.Background()
connString := os.Getenv("DATABASE_URL")
if connString == "" {
t.Skipf("skipping due to missing environment variable %v", "DATABASE_URL")
}
config, err := pgx.ParseConfig(connString)
require.NoError(t, err)
conn, err := pgx.ConnectConfig(ctx, config)
require.NoError(t, err)
t.Cleanup(func() {
conn.Close(ctx)
})
return conn
}