mirror of https://github.com/go-gost/gost.git
test(e2e): add ingress module e2e test suite
Verifies hostname→endpoint routing at the reverse-proxy tunnel entrypoint. A public gost runs a tunnel handler with an ingress mapping example.local→tunnel-UUID and an HTTP entrypoint; an internal client binds a reverse tunnel forwarding to the echo server. A request with a mapped Host is routed through the tunnel to the echo server, while an unmapped Host matches no ingress rule and is rejected.master
parent
9355607ba7
commit
1fd69ac704
|
|
@ -0,0 +1,90 @@
|
||||||
|
package e2e
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"fmt"
|
||||||
|
"io"
|
||||||
|
"testing"
|
||||||
|
|
||||||
|
"github.com/stretchr/testify/suite"
|
||||||
|
"github.com/testcontainers/testcontainers-go"
|
||||||
|
)
|
||||||
|
|
||||||
|
// IngressSuite verifies hostname→endpoint routing at the reverse-proxy tunnel
|
||||||
|
// entrypoint. A public gost runs a tunnel handler with an ingress mapping
|
||||||
|
// example.local→tunnel-UUID and an HTTP entrypoint on :8420. An internal client
|
||||||
|
// binds a reverse tunnel (tunnel.id=UUID) that forwards to the echo server.
|
||||||
|
//
|
||||||
|
// A request to the entrypoint with Host:example.local is routed via the ingress
|
||||||
|
// table to the tunnel and reaches the echo server ("hello-gost"); a request
|
||||||
|
// with an unmapped Host matches no ingress rule and is rejected with no route.
|
||||||
|
type IngressSuite struct {
|
||||||
|
suite.Suite
|
||||||
|
ctx context.Context
|
||||||
|
echoC testcontainers.Container
|
||||||
|
serverC testcontainers.Container
|
||||||
|
clientC testcontainers.Container
|
||||||
|
}
|
||||||
|
|
||||||
|
func (s *IngressSuite) SetupSuite() {
|
||||||
|
s.ctx = context.Background()
|
||||||
|
|
||||||
|
echoC, err := RunEchoContainer(s.ctx, SharedNetworkName)
|
||||||
|
s.Require().NoError(err)
|
||||||
|
s.echoC = echoC
|
||||||
|
|
||||||
|
serverC, err := RunGostContainerWithOptions(s.ctx, SharedNetworkName,
|
||||||
|
"testdata/ingress/server.yaml", []string{"gost-server"}, []string{"8420/tcp"})
|
||||||
|
s.Require().NoError(err)
|
||||||
|
s.serverC = serverC
|
||||||
|
|
||||||
|
clientC, err := RunGostContainerWithOptions(s.ctx, SharedNetworkName,
|
||||||
|
"testdata/ingress/client.yaml", []string{"gost-client"}, []string{"8423/tcp"})
|
||||||
|
s.Require().NoError(err)
|
||||||
|
s.clientC = clientC
|
||||||
|
}
|
||||||
|
|
||||||
|
func (s *IngressSuite) TearDownSuite() {
|
||||||
|
for _, c := range []testcontainers.Container{s.clientC, s.serverC, s.echoC} {
|
||||||
|
if c != nil {
|
||||||
|
c.Terminate(s.ctx)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// request sends an HTTP GET to the server entrypoint with the given Host
|
||||||
|
// header. When waitTunnel is true it retries until the reverse tunnel is bound
|
||||||
|
// (the mapped-host happy path); otherwise it makes a single attempt (the
|
||||||
|
// unmapped-host case, which should never reach the echo server). Returns the
|
||||||
|
// response body.
|
||||||
|
func (s *IngressSuite) request(host string, waitTunnel bool) string {
|
||||||
|
loop := "for i in $(seq 1 30); do "
|
||||||
|
if !waitTunnel {
|
||||||
|
loop = "for i in 1; do "
|
||||||
|
}
|
||||||
|
cmd := []string{
|
||||||
|
"sh", "-c",
|
||||||
|
fmt.Sprintf("%sbody=$(curl -s -H 'Host: %s' http://gost-server:8420/); echo \"$body\" | grep -q hello-gost && { echo \"$body\"; exit 0; }; sleep 1; done; echo \"$body\"", loop, host),
|
||||||
|
}
|
||||||
|
_, out, err := s.echoC.Exec(s.ctx, cmd)
|
||||||
|
s.Require().NoError(err)
|
||||||
|
body, err := io.ReadAll(out)
|
||||||
|
s.Require().NoError(err)
|
||||||
|
return string(body)
|
||||||
|
}
|
||||||
|
|
||||||
|
// TestMappedHostname verifies that a Host matching an ingress rule is routed
|
||||||
|
// through the tunnel to the echo server.
|
||||||
|
func (s *IngressSuite) TestMappedHostname() {
|
||||||
|
s.Require().Contains(s.request("example.local", true), "hello-gost")
|
||||||
|
}
|
||||||
|
|
||||||
|
// TestUnmappedHostname verifies that a Host with no ingress rule is rejected
|
||||||
|
// (no route to host) and never reaches the echo server.
|
||||||
|
func (s *IngressSuite) TestUnmappedHostname() {
|
||||||
|
s.Require().NotContains(s.request("nomapped.local", false), "hello-gost")
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestIngressSuite(t *testing.T) {
|
||||||
|
suite.Run(t, new(IngressSuite))
|
||||||
|
}
|
||||||
|
|
@ -0,0 +1,37 @@
|
||||||
|
# Internal client behind NAT: binds a reverse tunnel (tunnel.id matches the
|
||||||
|
# server's ingress endpoint) and forwards tunneled connections to the echo
|
||||||
|
# server. The tunnel server is reached via chain-0's tunnel connector.
|
||||||
|
#
|
||||||
|
# service-1 is a dummy TCP listener on a fixed port; the rtcp reverse-tunnel
|
||||||
|
# listener binds remotely through the chain and opens no local port, so this
|
||||||
|
# gives the e2e harness a stable port to wait on.
|
||||||
|
services:
|
||||||
|
- name: service-0
|
||||||
|
addr: ":0"
|
||||||
|
handler:
|
||||||
|
type: rtcp
|
||||||
|
listener:
|
||||||
|
type: rtcp
|
||||||
|
chain: chain-0
|
||||||
|
forwarder:
|
||||||
|
nodes:
|
||||||
|
- addr: tcp-echo:5678
|
||||||
|
|
||||||
|
- name: service-1
|
||||||
|
addr: ":8423"
|
||||||
|
handler:
|
||||||
|
type: http
|
||||||
|
listener:
|
||||||
|
type: tcp
|
||||||
|
|
||||||
|
chains:
|
||||||
|
- name: chain-0
|
||||||
|
hops:
|
||||||
|
- nodes:
|
||||||
|
- addr: gost-server:8421
|
||||||
|
connector:
|
||||||
|
type: tunnel
|
||||||
|
metadata:
|
||||||
|
tunnel.id: 6be39003-f4fa-49e4-a6ef-1997684d160e
|
||||||
|
dialer:
|
||||||
|
type: tcp
|
||||||
|
|
@ -0,0 +1,23 @@
|
||||||
|
# Public-facing gost: a tunnel handler with an ingress table that maps the
|
||||||
|
# hostname example.local to a tunnel endpoint, plus an HTTP entrypoint on :8420.
|
||||||
|
# External requests to the entrypoint are routed by their Host header through
|
||||||
|
# the ingress table to the matching tunnel.
|
||||||
|
services:
|
||||||
|
- name: service-0
|
||||||
|
addr: ":8421"
|
||||||
|
handler:
|
||||||
|
type: tunnel
|
||||||
|
metadata:
|
||||||
|
entrypoint: ":8420"
|
||||||
|
ingress: ingress-0
|
||||||
|
listener:
|
||||||
|
type: tcp
|
||||||
|
|
||||||
|
ingresses:
|
||||||
|
- name: ingress-0
|
||||||
|
rules:
|
||||||
|
- hostname: example.local
|
||||||
|
endpoint: 6be39003-f4fa-49e4-a6ef-1997684d160e
|
||||||
|
|
||||||
|
log:
|
||||||
|
level: debug
|
||||||
Loading…
Reference in New Issue