Add more unit tests

This commit is contained in:
Jamie Curnow
2023-08-02 15:32:16 +10:00
parent 155e09407f
commit 9ac0e8c943
7 changed files with 363 additions and 2 deletions

View File

@ -13,8 +13,7 @@ func PrettyPrint(next http.Handler) http.Handler {
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
prettyStr := r.URL.Query().Get("pretty")
if prettyStr == "1" || prettyStr == "true" {
ctx := r.Context()
ctx = context.WithValue(ctx, c.PrettyPrintCtxKey, true)
ctx := context.WithValue(r.Context(), c.PrettyPrintCtxKey, true)
next.ServeHTTP(w, r.WithContext(ctx))
} else {
next.ServeHTTP(w, r)