Michael McGuinness
fee71e7740
Feature/postprocessing * tests * passtest * fixshorttests * mosttests * improvingbasedockerfile * testspeeds * testing * host * canparallel * clean * passfullsuite * singlepagemax * test * findfeatures * findstables * tbls * tablestoo * tablestoo * lateraltests * tableloc * cleanup * inlinetable * childids * cleanup * tests
108 lines
3.0 KiB
Go
108 lines
3.0 KiB
Go
package test
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"io"
|
|
"net/http"
|
|
"strconv"
|
|
"testing"
|
|
"time"
|
|
|
|
"queryorchestration/internal/serviceconfig"
|
|
"queryorchestration/internal/serviceconfig/objectstore"
|
|
|
|
"github.com/docker/go-connections/nat"
|
|
"github.com/stretchr/testify/require"
|
|
"github.com/testcontainers/testcontainers-go"
|
|
"github.com/testcontainers/testcontainers-go/wait"
|
|
)
|
|
|
|
type AWSContainerConfig struct {
|
|
Container testcontainers.Container
|
|
ExternalEndpoint string
|
|
}
|
|
|
|
type AWSConfigProvider interface {
|
|
serviceconfig.ConfigProvider
|
|
objectstore.ConfigProvider
|
|
}
|
|
|
|
const (
|
|
awsAlias = "localstack"
|
|
awsPort = 4566
|
|
)
|
|
|
|
func CreateAWSContainer(t testing.TB, ctx context.Context, cfg AWSConfigProvider, network string) (*AWSContainerConfig, func()) {
|
|
alias := NormaliseAlias(fmt.Sprintf("%s_%s", awsAlias, t.Name()))
|
|
|
|
port, err := nat.NewPort("tcp", strconv.FormatInt(int64(awsPort), 10))
|
|
require.NoError(t, err)
|
|
|
|
req := testcontainers.ContainerRequest{
|
|
Image: "localstack/localstack:4.1.0",
|
|
Env: map[string]string{
|
|
"AWS_ACCESS_KEY_ID": cfg.GetAWSKeyID(),
|
|
"AWS_SECRET_ACCESS_KEY": cfg.GetAWSSecretKey(),
|
|
"AWS_SESSION_TOKEN": cfg.GetAWSSessionToken(),
|
|
"AWS_REGION": cfg.GetAWSRegion(),
|
|
"SERVICES": "s3,sqs,cloudwatch,logs",
|
|
"SKIP_SSL_CERT_DOWNLOAD": "1",
|
|
"LOCALSTACK_HOST": alias,
|
|
"SQS_ENDPOINT_STRATEGY": "path",
|
|
"EAGER_SERVICE_LOADING": "1",
|
|
"DEBUG": "1",
|
|
"LS_LOG": "trace",
|
|
"SQS_CLOUDWATCH_METRICS_REPORT_INTERVAL": "5",
|
|
},
|
|
ExposedPorts: []string{port.Port()},
|
|
WaitingFor: wait.ForAll(
|
|
wait.ForListeningPort(port),
|
|
wait.ForLog("Ready."),
|
|
wait.ForHTTP("/_localstack/health").
|
|
WithPort(port).
|
|
WithStartupTimeout(50*time.Second).
|
|
WithPollInterval(5*time.Second).
|
|
WithResponseMatcher(func(body io.Reader) bool {
|
|
return true
|
|
}).
|
|
WithStatusCodeMatcher(func(statusCode int) bool {
|
|
return statusCode == http.StatusOK
|
|
}),
|
|
),
|
|
Networks: []string{network},
|
|
NetworkAliases: map[string][]string{
|
|
network: {alias},
|
|
},
|
|
}
|
|
|
|
container, err := testcontainers.GenericContainer(ctx, testcontainers.GenericContainerRequest{
|
|
ContainerRequest: req,
|
|
Started: true,
|
|
})
|
|
require.NoError(t, err)
|
|
|
|
host, err := container.Host(ctx)
|
|
require.NoError(t, err)
|
|
mappedPort, err := container.MappedPort(ctx, port)
|
|
require.NoError(t, err)
|
|
|
|
extEndpoint := fmt.Sprintf("http://%s:%s", host, mappedPort.Port())
|
|
cfg.SetAWSEndpoint(extEndpoint)
|
|
cfg.SetSQSEndpoint(extEndpoint)
|
|
cfg.SetS3Endpoint(extEndpoint)
|
|
t.Setenv("AWS_ENDPOINT_URL", extEndpoint)
|
|
t.Setenv("AWS_ENDPOINT_URL_SQS", extEndpoint)
|
|
t.Setenv("AWS_ENDPOINT_URL_S3", extEndpoint)
|
|
|
|
return &AWSContainerConfig{
|
|
Container: container,
|
|
ExternalEndpoint: extEndpoint,
|
|
}, func() {
|
|
err := container.Terminate(ctx)
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
}
|
|
}
|