Added test for recovery

This commit is contained in:
Sasha Myasoedov 2014-08-08 15:50:52 +03:00
parent 6abe841c1f
commit fcd997e083

44
recovery_test.go Normal file
View File

@ -0,0 +1,44 @@
package gin
import (
"bytes"
"log"
"net/http"
"net/http/httptest"
"os"
"testing"
)
// TestPanicInHandler assert that panic has been recovered.
func TestPanicInHandler(t *testing.T) {
req, _ := http.NewRequest("GET", "/", nil)
w := httptest.NewRecorder()
// Disable panic logs for testing
log.SetOutput(bytes.NewBuffer(nil))
r := Default()
r.GET("/", func(_ *Context) {
panic("Oupps, Houston, we have a problem")
})
r.ServeHTTP(w, req)
// restore logging
log.SetOutput(os.Stderr)
if w.Code != 500 {
t.Errorf("Response code should be Internal Server Error, was: %s", w.Code)
}
bodyAsString := w.Body.String()
// fixme:
if bodyAsString != "" {
t.Errorf("Response body should be empty, was %s", bodyAsString)
}
//fixme:
if len(w.HeaderMap) != 0 {
t.Errorf("No headers should be provided, was %s", w.HeaderMap)
}
}