mirror of
https://github.com/grafana/grafana.git
synced 2024-11-29 20:24:18 -06:00
82 lines
2.2 KiB
Go
82 lines
2.2 KiB
Go
package middleware
|
|
|
|
import (
|
|
"path/filepath"
|
|
"testing"
|
|
|
|
ms "github.com/go-macaron/session"
|
|
"github.com/grafana/grafana/pkg/bus"
|
|
m "github.com/grafana/grafana/pkg/models"
|
|
"github.com/grafana/grafana/pkg/services/session"
|
|
. "github.com/smartystreets/goconvey/convey"
|
|
"gopkg.in/macaron.v1"
|
|
)
|
|
|
|
func TestRecoveryMiddleware(t *testing.T) {
|
|
Convey("Given an api route that panics", t, func() {
|
|
apiURL := "/api/whatever"
|
|
recoveryScenario("recovery middleware should return json", apiURL, func(sc *scenarioContext) {
|
|
sc.handlerFunc = PanicHandler
|
|
sc.fakeReq("GET", apiURL).exec()
|
|
sc.req.Header.Add("content-type", "application/json")
|
|
|
|
So(sc.resp.Code, ShouldEqual, 500)
|
|
So(sc.respJson["message"], ShouldStartWith, "Internal Server Error - Check the Grafana server logs for the detailed error message.")
|
|
So(sc.respJson["error"], ShouldStartWith, "Server Error")
|
|
})
|
|
})
|
|
|
|
Convey("Given a non-api route that panics", t, func() {
|
|
apiURL := "/whatever"
|
|
recoveryScenario("recovery middleware should return html", apiURL, func(sc *scenarioContext) {
|
|
sc.handlerFunc = PanicHandler
|
|
sc.fakeReq("GET", apiURL).exec()
|
|
|
|
So(sc.resp.Code, ShouldEqual, 500)
|
|
So(sc.resp.Header().Get("content-type"), ShouldEqual, "text/html; charset=UTF-8")
|
|
So(sc.resp.Body.String(), ShouldContainSubstring, "<title>Grafana - Error</title>")
|
|
})
|
|
})
|
|
}
|
|
|
|
func PanicHandler(c *m.ReqContext) {
|
|
panic("Handler has panicked")
|
|
}
|
|
|
|
func recoveryScenario(desc string, url string, fn scenarioFunc) {
|
|
Convey(desc, func() {
|
|
defer bus.ClearBusHandlers()
|
|
|
|
sc := &scenarioContext{
|
|
url: url,
|
|
}
|
|
viewsPath, _ := filepath.Abs("../../public/views")
|
|
|
|
sc.m = macaron.New()
|
|
sc.m.Use(Recovery())
|
|
|
|
sc.m.Use(macaron.Renderer(macaron.RenderOptions{
|
|
Directory: viewsPath,
|
|
Delims: macaron.Delims{Left: "[[", Right: "]]"},
|
|
}))
|
|
|
|
sc.m.Use(GetContextHandler())
|
|
// mock out gc goroutine
|
|
session.StartSessionGC = func() {}
|
|
sc.m.Use(Sessioner(&ms.Options{}, 0))
|
|
sc.m.Use(OrgRedirect())
|
|
sc.m.Use(AddDefaultResponseHeaders())
|
|
|
|
sc.defaultHandler = func(c *m.ReqContext) {
|
|
sc.context = c
|
|
if sc.handlerFunc != nil {
|
|
sc.handlerFunc(sc.context)
|
|
}
|
|
}
|
|
|
|
sc.m.Get(url, sc.defaultHandler)
|
|
|
|
fn(sc)
|
|
})
|
|
}
|