mirror of https://github.com/go-gost/gost.git
43 lines
1.4 KiB
Go
43 lines
1.4 KiB
Go
package e2e
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/testcontainers/testcontainers-go"
|
|
"github.com/testcontainers/testcontainers-go/wait"
|
|
)
|
|
|
|
func RunEchoContainer(ctx context.Context, networkName string) (testcontainers.Container, error) {
|
|
req := testcontainers.ContainerRequest{
|
|
Image: "hashicorp/http-echo:latest",
|
|
Networks: []string{networkName},
|
|
Cmd: []string{"-text=hello-gost", "-listen=:5678"},
|
|
ExposedPorts: []string{"5678/tcp"},
|
|
WaitingFor: wait.ForHTTP("/").WithPort("5678/tcp"),
|
|
}
|
|
return testcontainers.GenericContainer(ctx, testcontainers.GenericContainerRequest{
|
|
ContainerRequest: req,
|
|
Started: true,
|
|
})
|
|
}
|
|
|
|
func RunGostContainer(ctx context.Context, networkName, yamlPath string) (testcontainers.Container, error) {
|
|
req := testcontainers.ContainerRequest{
|
|
FromDockerfile: testcontainers.FromDockerfile{
|
|
Context: ".",
|
|
Dockerfile: "Dockerfile",
|
|
KeepImage: true,
|
|
},
|
|
Networks: []string{networkName},
|
|
Files: []testcontainers.ContainerFile{
|
|
{HostFilePath: "/tmp/gost-test-bin", ContainerFilePath: "/bin/gost", FileMode: 0755},
|
|
{HostFilePath: yamlPath, ContainerFilePath: "/config.yaml", FileMode: 0644},
|
|
},
|
|
Cmd: []string{"/bin/gost", "-C", "/config.yaml"},
|
|
}
|
|
return testcontainers.GenericContainer(ctx, testcontainers.GenericContainerRequest{
|
|
ContainerRequest: req,
|
|
Started: true,
|
|
})
|
|
}
|