Michael McGuinness
90353d8161
Doc Clean Structure * outline * isdocclean * placeholder for clean log * versionplustesting * versionplustesting * testspassed
49 lines
1.0 KiB
Go
49 lines
1.0 KiB
Go
package test
|
|
|
|
import (
|
|
"context"
|
|
docinitrunner "queryorchestration/api/docInitRunner"
|
|
queryrunner "queryorchestration/api/queryRunner"
|
|
"testing"
|
|
|
|
"queryorchestration/internal/serviceconfig"
|
|
|
|
"github.com/testcontainers/testcontainers-go"
|
|
)
|
|
|
|
type Runner = string
|
|
|
|
const (
|
|
DocInitRunner = docinitrunner.Name
|
|
DocCleanRunner = docinitrunner.Name
|
|
QueryRunner = queryrunner.Name
|
|
)
|
|
|
|
type RunnerConfig struct {
|
|
Name Runner
|
|
QueueURL string
|
|
Cfg serviceconfig.ConfigProvider
|
|
Network *testcontainers.DockerNetwork
|
|
Env map[string]string
|
|
}
|
|
|
|
func CreateRunner(t *testing.T, ctx context.Context, config *RunnerConfig) (*Container, func()) {
|
|
if config.Env == nil {
|
|
config.Env = map[string]string{}
|
|
}
|
|
config.Env["QUEUE_URL"] = config.QueueURL
|
|
|
|
c, cleanup := createContainer(t, ctx, &containerConfig{
|
|
Network: config.Network,
|
|
Name: config.Name,
|
|
Cfg: config.Cfg,
|
|
Env: config.Env,
|
|
WaitForMsg: "Listening to queue",
|
|
})
|
|
|
|
return &Container{
|
|
Container: c,
|
|
URI: config.QueueURL,
|
|
}, cleanup
|
|
}
|