allow tests to have custom spop engine configs

DropMorePackets · Feb 17, 2024 · 0833c29 · 0833c29
1 parent ac05ebd
commit 0833c29
Showing 1 changed file with 7 additions and 3 deletions.
diff --git a/pkg/testutil/haproxy.go b/pkg/testutil/haproxy.go
@@ -68,6 +68,7 @@ spoe-message e2e-res
 
 type HAProxyConfig struct {
 	EngineAddr           string
+	EngineConfig         string
 	FrontendPort         string
 	CustomFrontendConfig string
 	CustomBackendConfig  string
@@ -92,14 +93,17 @@ func WithHAProxy(cfg HAProxyConfig, f func(t *testing.T)) func(t *testing.T) {
 		type tmplCfg struct {
 			HAProxyConfig
 
-			EngineConfig string
-			StatsSocket  string
+			StatsSocket string
 		}
 		var tcfg tmplCfg
 		tcfg.HAProxyConfig = cfg
 		tcfg.StatsSocket = fmt.Sprintf("%s/stats%s.sock", os.TempDir(), tcfg.FrontendPort)
 
-		engineConfigFile := TempFile(t, "e2e.cfg", haproxyEngineConfig)
+		if cfg.EngineConfig == "" {
+			cfg.EngineConfig = haproxyEngineConfig
+		}
+
+		engineConfigFile := TempFile(t, "e2e.cfg", cfg.EngineConfig)
 		tcfg.EngineConfig = engineConfigFile
 		defer os.Remove(engineConfigFile)