Peter Kurfer
d70ba748f5
- merge packages to get a more concise layout because plugins are no more and therefore there's not a lot to be exported - fix test logger - rework config parsing to be easier and more transparent - remove unnecessary APIs because dynamic endpoint handling is rather a won't implement
64 lines
1.6 KiB
Go
64 lines
1.6 KiB
Go
package integration
|
|
|
|
import (
|
|
"context"
|
|
"path/filepath"
|
|
"runtime"
|
|
"strings"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/docker/go-connections/nat"
|
|
"github.com/testcontainers/testcontainers-go"
|
|
"github.com/testcontainers/testcontainers-go/wait"
|
|
)
|
|
|
|
func SetupINetMockContainer(ctx context.Context, tb testing.TB, exposedPorts ...string) (imContainer testcontainers.Container, err error) {
|
|
_, fileName, _, _ := runtime.Caller(0)
|
|
|
|
var repoRoot string
|
|
if repoRoot, err = filepath.Abs(filepath.Join(filepath.Dir(fileName), "..", "..", "..")); err != nil {
|
|
return
|
|
}
|
|
|
|
var waitStrategies []wait.Strategy
|
|
|
|
var tcpPortPresent = false
|
|
for _, port := range exposedPorts {
|
|
if strings.Contains(port, "tcp") {
|
|
tcpPortPresent = true
|
|
waitStrategies = append(waitStrategies, wait.ForListeningPort(nat.Port(port)))
|
|
}
|
|
}
|
|
|
|
if !tcpPortPresent {
|
|
exposedPorts = append(exposedPorts, "80/tcp")
|
|
waitStrategies = append(waitStrategies, wait.ForListeningPort("80/tcp"))
|
|
}
|
|
|
|
req := testcontainers.ContainerRequest{
|
|
FromDockerfile: testcontainers.FromDockerfile{
|
|
Context: repoRoot,
|
|
Dockerfile: filepath.Join("./", "testdata", "integration.dockerfile"),
|
|
},
|
|
ExposedPorts: exposedPorts,
|
|
WaitingFor: wait.ForAll(waitStrategies...),
|
|
}
|
|
|
|
imContainer, err = testcontainers.GenericContainer(ctx, testcontainers.GenericContainerRequest{
|
|
ContainerRequest: req,
|
|
Started: true,
|
|
})
|
|
|
|
if err != nil {
|
|
return
|
|
}
|
|
|
|
tb.Cleanup(func() {
|
|
shutdownCtx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
|
|
defer cancel()
|
|
_ = imContainer.Terminate(shutdownCtx)
|
|
})
|
|
|
|
return
|
|
}
|