1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
|
package test
import (
"context"
"net/http"
"net/url"
"strconv"
"strings"
"testing"
"time"
xcontainer "github.com/docker/docker/api/types/container"
"github.com/docker/docker/api/types/network"
"github.com/stretchr/testify/require"
"github.com/testcontainers/testcontainers-go"
"github.com/testcontainers/testcontainers-go/wait"
"github.com/xlgmokha/x/pkg/env"
)
func normalizeIssuer(t *testing.T, issuer string) (string, int) {
for _, loopback := range []string{"0.0.0.0", "127.0.0.1", "[::]", "localhost"} {
issuer = strings.Replace(issuer, loopback, testcontainers.HostInternal, 1)
}
u, err := url.Parse(issuer)
require.NoError(t, err)
port, err := strconv.Atoi(u.Port())
require.NoError(t, err)
return issuer, port
}
func NewContainer(t *testing.T, ctx context.Context, envVars map[string]string) *testcontainers.DockerContainer {
require.Equal(t, http.StatusOK, HttpGet(t, ctx, envVars["OIDC_ISSUER"]+"/.well-known/openid-configuration").StatusCode)
container, err := testcontainers.Run(
ctx,
env.Fetch("IMAGE_TAG", "sparkled:invalid"),
containerOptions(t, envVars)...,
)
require.NoError(t, err)
return container
}
func containerOptions(t *testing.T, envVars map[string]string) []testcontainers.ContainerCustomizer {
issuer, port := normalizeIssuer(t, envVars["OIDC_ISSUER"])
envVars["OIDC_ISSUER"] = issuer
options := []testcontainers.ContainerCustomizer{
testcontainers.WithEnv(envVars),
testcontainers.WithLogConsumers(&LogConsumer{t: t}),
testcontainers.WithWaitStrategy(wait.ForLog("Listening on").WithStartupTimeout(10 * time.Second)),
}
if isGitlabCI() {
options = append(options, testcontainers.WithHostConfigModifier(func(cfg *xcontainer.HostConfig) {
cfg.NetworkMode = xcontainer.NetworkMode(network.NetworkHost)
}))
} else {
options = append(options, testcontainers.WithExposedPorts("8080/tcp", "9901/tcp", "10000/tcp"))
options = append(options, testcontainers.WithHostPortAccess(port))
}
return options
}
func isGitlabCI() bool {
return env.Fetch("CI", "") == "true"
}
|
