From fcd997e08362409aca1d83c6fbd54c33b8857b40 Mon Sep 17 00:00:00 2001
From: Sasha Myasoedov <msoedov@gmail.com>
Date: Fri, 8 Aug 2014 15:50:52 +0300
Subject: [PATCH] Added test for recovery

---
 recovery_test.go | 44 ++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 44 insertions(+)
 create mode 100644 recovery_test.go

diff --git a/recovery_test.go b/recovery_test.go
new file mode 100644
index 00000000..097700ce
--- /dev/null
+++ b/recovery_test.go
@@ -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)
+	}
+
+}