Michael McGuinness
bbe6f4188e
Feature/tests * improvetests * generation * simplifiedtesting * simplifiedtesting * longfile
91 lines
2.2 KiB
Go
91 lines
2.2 KiB
Go
package test
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"strconv"
|
|
"testing"
|
|
"time"
|
|
|
|
db "queryorchestration/internal/database"
|
|
"queryorchestration/internal/serviceconfig"
|
|
|
|
"github.com/docker/go-connections/nat"
|
|
"github.com/stretchr/testify/require"
|
|
"github.com/testcontainers/testcontainers-go"
|
|
"github.com/testcontainers/testcontainers-go/wait"
|
|
)
|
|
|
|
type CreateDatabaseConfig struct {
|
|
RunMigrations bool
|
|
}
|
|
|
|
const (
|
|
dbAlias = "postgres"
|
|
dbPort = 5432
|
|
)
|
|
|
|
func CreateDB(t testing.TB, ctx context.Context, cfg serviceconfig.ConfigProvider, network string, dcfg *CreateDatabaseConfig) testcontainers.Container {
|
|
port, err := nat.NewPort("tcp", strconv.Itoa(dbPort))
|
|
require.NoError(t, err)
|
|
|
|
name := NormaliseAlias(fmt.Sprintf("queryorchestration_%s", t.Name()))
|
|
pass := "pass"
|
|
user := "postgres"
|
|
|
|
req := testcontainers.ContainerRequest{
|
|
Image: "postgres:17.2-alpine3.21",
|
|
Name: "postgres_test_queryorchestration",
|
|
Env: map[string]string{
|
|
"POSTGRES_USER": user,
|
|
"POSTGRES_PASSWORD": pass,
|
|
},
|
|
Cmd: []string{"postgres", "-c", "max_connections=1000"},
|
|
ExposedPorts: []string{port.Port()},
|
|
WaitingFor: wait.ForAll(
|
|
wait.ForExposedPort(),
|
|
wait.ForListeningPort(port),
|
|
wait.ForLog("database system is ready to accept connections"),
|
|
wait.ForSQL(port, "postgres", func(host string, port nat.Port) string {
|
|
return fmt.Sprintf("postgres://%s:%s@%s:%d/postgres?sslmode=disable",
|
|
user, pass, host, port.Int())
|
|
}).
|
|
WithStartupTimeout(25*time.Second).
|
|
WithPollInterval(10*time.Second),
|
|
),
|
|
Networks: []string{network},
|
|
NetworkAliases: map[string][]string{
|
|
network: {dbAlias},
|
|
},
|
|
}
|
|
|
|
container, err := testcontainers.GenericContainer(ctx, testcontainers.GenericContainerRequest{
|
|
ContainerRequest: req,
|
|
Started: true,
|
|
Reuse: true,
|
|
})
|
|
require.NoError(t, err)
|
|
|
|
host, err := container.Host(ctx)
|
|
require.NoError(t, err)
|
|
mappedPort, err := container.MappedPort(ctx, port)
|
|
require.NoError(t, err)
|
|
|
|
cfg.SetDBUser(user)
|
|
cfg.SetDBSecret(pass)
|
|
cfg.SetDBName(name)
|
|
cfg.SetDBNoSSL(true)
|
|
cfg.SetDBHost(host)
|
|
cfg.SetDBPort(mappedPort.Int())
|
|
|
|
if dcfg.RunMigrations {
|
|
err := db.RunMigrations(ctx, cfg)
|
|
require.NoError(t, err)
|
|
|
|
err = cfg.SetDBPool(ctx)
|
|
require.NoError(t, err)
|
|
}
|
|
|
|
return container
|
|
}
|