Michael McGuinness
52 lines
1.2 KiB
Go
52 lines
1.2 KiB
Go
package test
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
queryservice "queryorchestration/api/queryService"
|
|
"queryorchestration/internal/serviceconfig"
|
|
"testing"
|
|
|
|
"github.com/docker/go-connections/nat"
|
|
"github.com/stretchr/testify/require"
|
|
"github.com/testcontainers/testcontainers-go"
|
|
)
|
|
|
|
type Service = string
|
|
|
|
const (
|
|
QueryService = queryservice.Name
|
|
)
|
|
|
|
type ServiceConfig struct {
|
|
Name Service
|
|
Cfg serviceconfig.ConfigProvider
|
|
Network *testcontainers.DockerNetwork
|
|
Env map[string]string
|
|
}
|
|
|
|
func CreateService(t testing.TB, ctx context.Context, config *ServiceConfig) (*Container, func()) {
|
|
port, err := nat.NewPort("tcp", "8080")
|
|
require.NoError(t, err)
|
|
|
|
container, cleanup := createContainer(t, ctx, &containerConfig{
|
|
Cfg: config.Cfg,
|
|
Name: config.Name,
|
|
Network: config.Network,
|
|
Env: config.Env,
|
|
WaitForMsg: "⇨ http server started on [::]:8080",
|
|
})
|
|
|
|
host, err := container.Host(ctx)
|
|
require.NoError(t, err)
|
|
mappedPort, err := container.MappedPort(ctx, port)
|
|
require.NoError(t, err)
|
|
|
|
address := fmt.Sprintf("http://%s:%s", host, mappedPort.Port())
|
|
|
|
return &Container{
|
|
URI: address,
|
|
Container: container,
|
|
}, cleanup
|
|
}
|