e2e: add connect-to flag smoke test

saucelabs · Sep 11, 2024 · b03e94a · b03e94a
1 parent dd6c0f3
commit b03e94a
Show file tree

Hide file tree

Showing 3 changed files with 27 additions and 0 deletions.
diff --git a/e2e/forwarder/service.go b/e2e/forwarder/service.go
@@ -177,6 +177,11 @@ func (s *Service) WithDNSTimeout(timeout time.Duration) *Service {
 	return s
 }
 
+func (s *Service) WithConnectTo(v string) compose.ServiceBuilder {
+	s.Environment["FORWARDER_CONNECT_TO"] = v
+	return s
+}
+
 func (s *Service) WithHTTPDialTimeout(timeout time.Duration) *Service {
 	s.Environment["FORWARDER_HTTP_DIAL_TIMEOUT"] = timeout.String()
 	return s

diff --git a/e2e/setups.go b/e2e/setups.go
@@ -40,6 +40,7 @@ func AllSetups() []setup.Setup {
 	SetupFlagResponseHeader(l)
 	SetupFlagConnectHeader(l)
 	SetupFlagDNSServer(l)
+	SetupFlagConnectTo(l)
 	SetupFlagInsecure(l)
 	SetupFlagMITMCACert(l)
 	SetupFlagMITMGenCA(l)
@@ -294,6 +295,23 @@ func SetupFlagDNSServer(l *setupList) {
 	}
 }
 
+func SetupFlagConnectTo(l *setupList) {
+	l.Add(
+		setup.Setup{
+			Name: "flag-connect-to",
+			Compose: compose.NewBuilder().
+				AddService(
+					forwarder.HttpbinService()).
+				AddService(
+					forwarder.ProxyService().
+						WithConnectTo("foo::httpbin:8080")).
+				MustBuild(),
+			Run: "^TestFlagConnectTo$",
+		},
+	)
+
+}
+
 func SetupFlagInsecure(l *setupList) {
 	l.Add(
 		setup.Setup{

diff --git a/e2e/tests/flag_test.go b/e2e/tests/flag_test.go
@@ -136,6 +136,10 @@ func TestFlagDNSServer(t *testing.T) {
 	})
 }
 
+func TestFlagConnectTo(t *testing.T) {
+	newClient(t, serviceScheme("HTTPBIN_PROTOCOL") + "://foo:123").GET("/status/200").ExpectStatus(http.StatusOK)
+}
+
 func TestFlagInsecure(t *testing.T) {
 	t.Run("true", func(t *testing.T) {
 		newClient(t, httpbin).GET("/status/200").ExpectStatus(http.StatusOK)