Fixed recovery unit tests

This commit is contained in:
Manu Mtz.-Almeida 2016-01-26 19:28:41 +01:00
parent ffb5c0412a
commit 1183c1ca8f
2 changed files with 3 additions and 2 deletions

View File

@ -37,7 +37,7 @@ func RecoveryWithWriter(out io.Writer) HandlerFunc {
if logger != nil { if logger != nil {
stack := stack(3) stack := stack(3)
httprequest, _ := httputil.DumpRequest(c.Request, false) httprequest, _ := httputil.DumpRequest(c.Request, false)
logger.Printf("[Recovery] panic recovered:\n%s\n%s\n%s %s", string(httprequest), err, stack, reset) logger.Printf("[Recovery] panic recovered:\n%s\n%s\n%s%s", string(httprequest), err, stack, reset)
} }
c.AbortWithStatus(500) c.AbortWithStatus(500)
} }

View File

@ -23,7 +23,8 @@ func TestPanicInHandler(t *testing.T) {
w := performRequest(router, "GET", "/recovery") w := performRequest(router, "GET", "/recovery")
// TEST // TEST
assert.Equal(t, w.Code, 500) assert.Equal(t, w.Code, 500)
assert.Contains(t, buffer.String(), "Panic recovery -> Oupps, Houston, we have a problem") assert.Contains(t, buffer.String(), "GET /recovery")
assert.Contains(t, buffer.String(), "Oupps, Houston, we have a problem")
assert.Contains(t, buffer.String(), "TestPanicInHandler") assert.Contains(t, buffer.String(), "TestPanicInHandler")
} }