From 134595dcc78e10d75d2326a4c6d55019e5fc567f Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Micha=C5=82=20Matczuk?= <michal.matczuk@saucelabs.com>
Date: Fri, 17 Nov 2023 12:26:43 +0100
Subject: [PATCH] e2e: make SetupFlagDirectDomains test all httpbin schemes

---
 e2e/setups.go | 33 +++++++++++++++++++--------------
 1 file changed, 19 insertions(+), 14 deletions(-)

diff --git a/e2e/setups.go b/e2e/setups.go
index 508a356a..91e3c2a0 100644
--- a/e2e/setups.go
+++ b/e2e/setups.go
@@ -401,20 +401,25 @@ func SetupFlagDenyDomains(l *setupList) {
 }
 
 func SetupFlagDirectDomains(l *setupList) {
-	l.Add(setup.Setup{
-		Name: "flag-direct-domains",
-		Compose: compose.NewBuilder().
-			AddService(
-				forwarder.HttpbinService()).
-			AddService(
-				forwarder.ProxyService().
-					WithUpstream(forwarder.UpstreamProxyServiceName, "http").
-					WithDirectDomains("httpbin")).
-			AddService(
-				forwarder.UpstreamProxyService()).
-			MustBuild(),
-		Run: "^TestFlagDirectDomains$",
-	})
+	for _, scheme := range forwarder.HttpbinSchemes {
+		l.Add(
+			setup.Setup{
+				Name: "flag-direct-domains-" + scheme,
+				Compose: compose.NewBuilder().
+					AddService(
+						forwarder.HttpbinService().
+							WithProtocol(scheme)).
+					AddService(
+						forwarder.ProxyService().
+							WithUpstream(forwarder.UpstreamProxyServiceName, "http").
+							WithDirectDomains("httpbin")).
+					AddService(
+						forwarder.UpstreamProxyService()).
+					MustBuild(),
+				Run: "^TestFlagDirectDomains$",
+			},
+		)
+	}
 }
 
 func SetupFlagRateLimit(l *setupList) {