From f2176c3100d032cff4e093503ca0aba95954308c Mon Sep 17 00:00:00 2001
From: Sasha Myasoedov <msoedov@gmail.com>
Date: Mon, 11 Aug 2014 13:37:35 +0300
Subject: [PATCH] Adjusted tests for recovery.

---
 recovery_test.go | 42 +++++++++++++++++++++++++++++++++++++++---
 1 file changed, 39 insertions(+), 3 deletions(-)

diff --git a/recovery_test.go b/recovery_test.go
index 097700ce..7383a1df 100644
--- a/recovery_test.go
+++ b/recovery_test.go
@@ -11,14 +11,14 @@ import (
 
 // TestPanicInHandler assert that panic has been recovered.
 func TestPanicInHandler(t *testing.T) {
-	req, _ := http.NewRequest("GET", "/", nil)
+	req, _ := http.NewRequest("GET", "/recovery", nil)
 	w := httptest.NewRecorder()
 
 	// Disable panic logs for testing
 	log.SetOutput(bytes.NewBuffer(nil))
 
 	r := Default()
-	r.GET("/", func(_ *Context) {
+	r.GET("/recovery", func(_ *Context) {
 		panic("Oupps, Houston, we have a problem")
 	})
 
@@ -32,7 +32,43 @@ func TestPanicInHandler(t *testing.T) {
 	}
 	bodyAsString := w.Body.String()
 
-	//	fixme:
+	//fixme: no message provided?
+	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)
+	}
+
+}
+
+// TestPanicWithAbort assert that panic has been recovered even if context.Abort was used.
+func TestPanicWithAbort(t *testing.T) {
+	req, _ := http.NewRequest("GET", "/recovery", nil)
+	w := httptest.NewRecorder()
+
+	// Disable panic logs for testing
+	log.SetOutput(bytes.NewBuffer(nil))
+
+	r := Default()
+	r.GET("/recovery", func(c *Context) {
+		c.Abort(400)
+		panic("Oupps, Houston, we have a problem")
+	})
+
+	r.ServeHTTP(w, req)
+
+	// restore logging
+	log.SetOutput(os.Stderr)
+
+	// fixme: why not 500?
+	if w.Code != 400 {
+		t.Errorf("Response code should be Bad request, was: %s", w.Code)
+	}
+	bodyAsString := w.Body.String()
+
+	//fixme: no message provided?
 	if bodyAsString != "" {
 		t.Errorf("Response body should be empty, was  %s", bodyAsString)
 	}