allow tests to have custom backend configs

DropMorePackets · Feb 18, 2024 · 94d6979 · 94d6979
1 parent 0833c29
commit 94d6979
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/pkg/testutil/haproxy.go b/pkg/testutil/haproxy.go
@@ -38,7 +38,7 @@ backend backend
 
     {{ .CustomBackendConfig }}
 
-    http-request return status 200 content-type "text/plain" string "Hello World!\n"
+    {{ .BackendConfig }}
 
 backend e2e-spoa
     mode tcp
@@ -71,6 +71,7 @@ type HAProxyConfig struct {
 	EngineConfig         string
 	FrontendPort         string
 	CustomFrontendConfig string
+	BackendConfig        string
 	CustomBackendConfig  string
 }
 
@@ -103,6 +104,12 @@ func WithHAProxy(cfg HAProxyConfig, f func(t *testing.T)) func(t *testing.T) {
 			cfg.EngineConfig = haproxyEngineConfig
 		}
 
+		if cfg.BackendConfig == "" {
+			cfg.BackendConfig = `
+http-request return status 200 content-type "text/plain" string "Hello World!\n"
+`
+		}
+
 		engineConfigFile := TempFile(t, "e2e.cfg", cfg.EngineConfig)
 		tcfg.EngineConfig = engineConfigFile
 		defer os.Remove(engineConfigFile)