snapshot of https://github.com/go-chi/chi.git (master, 2026-09-15; blobs > 131072B omitted)

middleware/logger_test.go

@@ -0,0 +1,51 @@
+package middleware
+
+import (
+	"bufio"
+	"bytes"
+	"net"
+	"net/http"
+	"net/http/httptest"
+	"testing"
+	"time"
+)
+
+type testLoggerWriter struct {
+	*httptest.ResponseRecorder
+}
+
+func (cw testLoggerWriter) Hijack() (net.Conn, *bufio.ReadWriter, error) {
+	return nil, nil, nil
+}
+
+func TestRequestLogger(t *testing.T) {
+	testHandler := http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+		_, ok := w.(http.Hijacker)
+		if !ok {
+			t.Errorf("http.Hijacker is unavailable on the writer. add the interface methods.")
+		}
+	})
+
+	r := httptest.NewRequest("GET", "/", nil)
+	w := testLoggerWriter{
+		ResponseRecorder: httptest.NewRecorder(),
+	}
+
+	handler := DefaultLogger(testHandler)
+	handler.ServeHTTP(w, r)
+}
+
+func TestRequestLoggerReadFrom(t *testing.T) {
+	data := []byte("file data")
+	testHandler := http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+		http.ServeContent(w, r, "file", time.Time{}, bytes.NewReader(data))
+	})
+
+	r := httptest.NewRequest("GET", "/", nil)
+	w := httptest.NewRecorder()
+
+	handler := DefaultLogger(testHandler)
+	handler.ServeHTTP(w, r)
+
+	assertEqual(t, data, w.Body.Bytes())
+}