naett.c

commit 0805cf86711c09530fe8dd5678b5a9bcd965eda2
parent 22907c78be352fcbfa4842eee864c62355fcf903
Author: Erik Agsjö <erik.agsjo@gmail.com>
Date:   Sun, 12 Dec 2021 16:14:59 +0100

Trace test rig requests

Diffstat:
M
testrig/rig.go
 | 
17
+++++++++++++----

1 file changed, 13 insertions(+), 4 deletions(-)
diff --git a/testrig/rig.go b/testrig/rig.go
@@ -53,11 +53,20 @@ func runTest() error {
 	return nil
 }
 
+type Handler func(w http.ResponseWriter, r *http.Request)
+
+func trace(handler Handler) Handler {
+	return func(w http.ResponseWriter, r *http.Request) {
+		log.Printf("%v - %v", r.Method, r.URL)
+		handler(w, r)
+	}
+}
+
 func serve() {
-	http.HandleFunc("/get", testGETHandler)
-	http.HandleFunc("/post", testPOSTHandler)
-	http.HandleFunc("/redirect", testRedirectHandler)
-	http.HandleFunc("/redirected", redirectedHandler)
+	http.HandleFunc("/get", trace(testGETHandler))
+	http.HandleFunc("/post", trace(testPOSTHandler))
+	http.HandleFunc("/redirect", trace(testRedirectHandler))
+	http.HandleFunc("/redirected", trace(redirectedHandler))
 	log.Fatal(http.ListenAndServe(":4711", nil))
 }