query-orchestration/internal/test/service.go

package test

import (
	"context"
	"fmt"
	"testing"

	queryservice "queryorchestration/api/queryService"
	"queryorchestration/internal/serviceconfig"

	"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
}