106 lines
2.8 KiB
Go
106 lines
2.8 KiB
Go
package requestlog_test
|
|
|
|
import (
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/go-chi/chi/v5"
|
|
"github.com/stretchr/testify/require"
|
|
"go.uber.org/zap"
|
|
"go.uber.org/zap/zaptest/observer"
|
|
|
|
"gitlab.com/slon/shad-go/middleware/requestlog"
|
|
)
|
|
|
|
type oneShotHandler struct {
|
|
inner http.HandlerFunc
|
|
|
|
t *testing.T
|
|
used bool
|
|
}
|
|
|
|
func (h *oneShotHandler) ServeHTTP(w http.ResponseWriter, r *http.Request) {
|
|
if h.used {
|
|
h.t.Errorf("handler used twice")
|
|
return
|
|
}
|
|
|
|
h.used = true
|
|
h.inner(w, r)
|
|
}
|
|
|
|
func TestRequestLog(t *testing.T) {
|
|
core, obs := observer.New(zap.DebugLevel)
|
|
|
|
m := chi.NewRouter()
|
|
m.Use(requestlog.Log(zap.New(core)))
|
|
|
|
m.Get("/simple", (&oneShotHandler{
|
|
inner: func(w http.ResponseWriter, r *http.Request) {
|
|
w.WriteHeader(http.StatusOK)
|
|
},
|
|
t: t,
|
|
}).ServeHTTP)
|
|
|
|
m.Post("/post", func(w http.ResponseWriter, r *http.Request) {
|
|
w.WriteHeader(http.StatusOK)
|
|
})
|
|
m.Get("/forbidden", func(w http.ResponseWriter, r *http.Request) {
|
|
w.WriteHeader(http.StatusForbidden)
|
|
})
|
|
|
|
m.Get("/slow", func(w http.ResponseWriter, r *http.Request) {
|
|
time.Sleep(time.Millisecond * 200)
|
|
w.WriteHeader(http.StatusOK)
|
|
})
|
|
m.Get("/forgetful", func(w http.ResponseWriter, r *http.Request) {
|
|
// TODO
|
|
})
|
|
m.Get("/buggy", func(w http.ResponseWriter, r *http.Request) {
|
|
panic("bug")
|
|
})
|
|
|
|
m.ServeHTTP(httptest.NewRecorder(), httptest.NewRequest("GET", "/simple", nil))
|
|
m.ServeHTTP(httptest.NewRecorder(), httptest.NewRequest("POST", "/post", nil))
|
|
m.ServeHTTP(httptest.NewRecorder(), httptest.NewRequest("GET", "/forbidden", nil))
|
|
m.ServeHTTP(httptest.NewRecorder(), httptest.NewRequest("GET", "/slow", nil))
|
|
m.ServeHTTP(httptest.NewRecorder(), httptest.NewRequest("GET", "/forgetful", nil))
|
|
|
|
require.Panics(t, func() {
|
|
m.ServeHTTP(httptest.NewRecorder(), httptest.NewRequest("GET", "/buggy", nil))
|
|
})
|
|
|
|
checkEntries := func(path string, panic bool, code int) {
|
|
entries := obs.FilterField(zap.String("path", path)).All()
|
|
|
|
require.Len(t, entries, 2)
|
|
require.Equal(t, "request started", entries[0].Message)
|
|
require.Contains(t, entries[0].ContextMap(), "request_id")
|
|
|
|
var requestID zap.Field
|
|
for _, f := range entries[0].Context {
|
|
if f.Key == "request_id" {
|
|
requestID = f
|
|
}
|
|
}
|
|
|
|
if !panic {
|
|
require.Equal(t, "request finished", entries[1].Message)
|
|
require.Contains(t, entries[1].Context, zap.Int("status_code", code))
|
|
require.Contains(t, entries[1].ContextMap(), "duration")
|
|
require.Contains(t, entries[1].Context, requestID)
|
|
} else {
|
|
require.Equal(t, "request panicked", entries[1].Message)
|
|
require.Contains(t, entries[1].Context, requestID)
|
|
}
|
|
}
|
|
|
|
checkEntries("/simple", false, http.StatusOK)
|
|
checkEntries("/post", false, http.StatusOK)
|
|
checkEntries("/forbidden", false, http.StatusForbidden)
|
|
checkEntries("/slow", false, http.StatusOK)
|
|
checkEntries("/forgetful", false, http.StatusOK)
|
|
checkEntries("/buggy", true, 0)
|
|
}
|