test/http2: Add custom-response endpoint for flexible e2e testing

Add a `/custom-response` endpoint that returns a configurable response via the `CUSTOM_RESPONSE` environment variable.
frobware · Dec 10, 2024 · 7520610 · 7520610
1 parent 8be1749
commit 7520610
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/test/http2/server.go b/test/http2/server.go
@@ -24,6 +24,7 @@ func lookupEnv(key, defaultVal string) string {
 func Serve() {
 	crtFile := lookupEnv("TLS_CRT", defaultTLSCrt)
 	keyFile := lookupEnv("TLS_KEY", defaultTLSKey)
+	customResponse := lookupEnv("CUSTOM_RESPONSE", "custom response")
 
 	http.HandleFunc("/", func(w http.ResponseWriter, req *http.Request) {
 		fmt.Fprint(w, req.Proto)
@@ -33,6 +34,10 @@ func Serve() {
 		fmt.Fprint(w, "ready")
 	})
 
+	http.HandleFunc("/custom-response", func(w http.ResponseWriter, req *http.Request) {
+		fmt.Fprint(w, customResponse)
+	})
+
 	go func() {
 		port := lookupEnv("HTTP_PORT", defaultHTTPPort)
 		log.Printf("Listening on port %v\n", port)