-
Notifications
You must be signed in to change notification settings - Fork 16
/
datastore_setup.go
108 lines (90 loc) · 2.58 KB
/
datastore_setup.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
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
package tests
import (
"database/sql"
"fmt"
"testing"
"time"
"github.com/HewlettPackard/galadriel/pkg/server/db"
"github.com/HewlettPackard/galadriel/pkg/server/db/postgres"
"github.com/HewlettPackard/galadriel/pkg/server/db/sqlite"
"github.com/ory/dockertest/v3"
"github.com/ory/dockertest/v3/docker"
"github.com/sirupsen/logrus"
"github.com/stretchr/testify/require"
)
const (
postgresImage = "15-alpine"
user = "test_user"
password = "test_password"
dbname = "test_db"
)
func setupSQLiteDatastore(t *testing.T) db.Datastore {
// Use an in-memory database
dsn := ":memory:"
datastore, err := sqlite.NewDatastore(dsn, logrus.New())
require.NoError(t, err)
t.Cleanup(func() {
t.Log("Closing SQLite database")
err = datastore.Close()
require.NoError(t, err)
})
return datastore
}
func setupPostgresDatastore(t *testing.T) db.Datastore {
conn := startPostgresDB(t)
datastore, err := postgres.NewDatastore(conn, logrus.New())
require.NoError(t, err)
t.Cleanup(func() {
t.Log("Closing Postgres database")
err = datastore.Close()
require.NoError(t, err)
})
return datastore
}
// starts a postgres DB in a docker container and returns the connection string
func startPostgresDB(tb testing.TB) string {
pool, err := dockertest.NewPool("")
require.NoError(tb, err)
err = pool.Client.Ping()
require.NoError(tb, err)
// pulls an image, creates a container based on it and runs it
resource, err := pool.RunWithOptions(&dockertest.RunOptions{
Repository: "postgres",
Tag: postgresImage,
Env: []string{
"POSTGRES_PASSWORD=" + password,
"POSTGRES_USER=" + user,
"POSTGRES_DB=" + dbname,
},
}, func(config *docker.HostConfig) {
// set AutoRemove to true so that stopped container goes away by itself
config.AutoRemove = true
config.RestartPolicy = docker.RestartPolicy{Name: "no"}
})
require.NoError(tb, err)
hostAndPort := resource.GetHostPort("5432/tcp")
databaseURL := fmt.Sprintf("postgres://%s:%s@%s/%s?sslmode=disable", user, password, hostAndPort, dbname)
tb.Logf("Connecting to a test database on url: %s", databaseURL)
// wait until db in container is ready using exponential backoff-retry
pool.MaxWait = 60 * time.Second
if err = pool.Retry(func() error {
db, err := sql.Open("postgres", databaseURL)
if err != nil {
return err
}
defer func() {
cerr := db.Close()
if err != nil {
err = cerr
}
}()
return db.Ping()
}); err != nil {
tb.Fatalf("Could not connect to docker: %s", err)
}
tb.Cleanup(func() {
err = pool.Purge(resource)
require.NoError(tb, err)
})
return databaseURL
}