60 lines
1.3 KiB
Go
60 lines
1.3 KiB
Go
|
package postgresql
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"database/sql"
|
||
|
|
"fmt"
|
||
|
|
"github.com/hashicorp/vault/helper/testhelpers/docker"
|
||
|
|
"net/url"
|
||
|
|
"os"
|
||
|
|
"testing"
|
||
|
|
)
|
||
|
|
|
||
|
|
func PrepareTestContainer(t *testing.T, version string) (func(), string) {
|
||
|
|
if os.Getenv("PG_URL") != "" {
|
||
|
|
return func() {}, os.Getenv("PG_URL")
|
||
|
|
}
|
||
|
|
|
||
|
|
if version == "" {
|
||
|
|
version = "11"
|
||
|
|
}
|
||
|
|
runner, err := docker.NewServiceRunner(docker.RunOptions{
|
||
|
|
ImageRepo: "postgres",
|
||
|
|
ImageTag: version,
|
||
|
|
Env: []string{"POSTGRES_PASSWORD=secret", "POSTGRES_DB=database"},
|
||
|
|
Ports: []string{"5432/tcp"},
|
||
|
|
})
|
||
|
|
if err != nil {
|
||
|
|
t.Fatalf("Could not start docker Postgres: %s", err)
|
||
|
|
}
|
||
|
|
|
||
|
|
svc, err := runner.StartService(context.Background(), connectPostgres)
|
||
|
|
if err != nil {
|
||
|
|
t.Fatalf("Could not start docker Postgres: %s", err)
|
||
|
|
}
|
||
|
|
|
||
|
|
return svc.Cleanup, svc.Config.URL().String()
|
||
|
|
}
|
||
|
|
|
||
|
|
func connectPostgres(ctx context.Context, host string, port int) (docker.ServiceConfig, error) {
|
||
|
|
u := url.URL{
|
||
|
|
Scheme: "postgres",
|
||
|
|
User: url.UserPassword("postgres", "secret"),
|
||
|
|
Host: fmt.Sprintf("%s:%d", host, port),
|
||
|
|
Path: "postgres",
|
||
|
|
RawQuery: "sslmode=disable",
|
||
|
|
}
|
||
|
|
|
||
|
|
db, err := sql.Open("postgres", u.String())
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
defer db.Close()
|
||
|
|
|
||
|
|
err = db.Ping()
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
return docker.NewServiceURL(u), nil
|
||
|
|
}
|