1package negroni 2 3import ( 4 "bytes" 5 "log" 6 "net/http" 7 "net/http/httptest" 8 "testing" 9) 10 11func Test_Logger(t *testing.T) { 12 buff := bytes.NewBufferString("") 13 recorder := httptest.NewRecorder() 14 15 l := NewLogger() 16 l.Logger = log.New(buff, "[negroni] ", 0) 17 18 n := New() 19 // replace log for testing 20 n.Use(l) 21 n.UseHandler(http.HandlerFunc(func(rw http.ResponseWriter, r *http.Request) { 22 rw.WriteHeader(http.StatusNotFound) 23 })) 24 25 req, err := http.NewRequest("GET", "http://localhost:3000/foobar", nil) 26 if err != nil { 27 t.Error(err) 28 } 29 30 n.ServeHTTP(recorder, req) 31 expect(t, recorder.Code, http.StatusNotFound) 32 refute(t, len(buff.String()), 0) 33} 34