WIP

2026-03-15 20:21:47 +08:00
parent 45d94cf391
commit c8b48dc248
4 changed files with 140 additions and 0 deletions
@@ -0,0 +1,42 @@
+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,
+	})
+}