mirror of
https://github.com/grafana/grafana.git
synced 2025-07-30 03:12:59 +08:00
701f6d5436
* UserService: update callers to use the UserService instead of calling sqlstore directly There is one major change hiding in this PR. UserService.Delete originally called a number of services to delete user-related records. I moved everything except the actual call to the user table, and moved those into the API. This was done to avoid dependencies cycles; many of our services depend on the user service, so the user service itself should have as few dependencies as possible.
88 lines
2.5 KiB
Go
88 lines
2.5 KiB
Go
package middleware
|
|
|
|
import (
|
|
"path/filepath"
|
|
"strings"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
|
|
"github.com/grafana/grafana/pkg/infra/remotecache"
|
|
"github.com/grafana/grafana/pkg/models"
|
|
"github.com/grafana/grafana/pkg/services/auth"
|
|
"github.com/grafana/grafana/pkg/setting"
|
|
"github.com/grafana/grafana/pkg/web"
|
|
)
|
|
|
|
func TestRecoveryMiddleware(t *testing.T) {
|
|
t.Run("Given an API route that panics", func(t *testing.T) {
|
|
apiURL := "/api/whatever"
|
|
recoveryScenario(t, "recovery middleware should return JSON", apiURL, func(t *testing.T, sc *scenarioContext) {
|
|
sc.handlerFunc = panicHandler
|
|
sc.fakeReq("GET", apiURL).exec()
|
|
sc.req.Header.Set("content-type", "application/json")
|
|
|
|
assert.Equal(t, 500, sc.resp.Code)
|
|
assert.Equal(t, "Internal Server Error - Check the Grafana server logs for the detailed error message.", sc.respJson["message"])
|
|
assert.True(t, strings.HasPrefix(sc.respJson["error"].(string), "Server Error"))
|
|
})
|
|
})
|
|
|
|
t.Run("Given a non-API route that panics", func(t *testing.T) {
|
|
apiURL := "/whatever"
|
|
recoveryScenario(t, "recovery middleware should return html", apiURL, func(t *testing.T, sc *scenarioContext) {
|
|
sc.handlerFunc = panicHandler
|
|
sc.fakeReq("GET", apiURL).exec()
|
|
|
|
assert.Equal(t, 500, sc.resp.Code)
|
|
assert.Equal(t, "text/html; charset=UTF-8", sc.resp.Header().Get("content-type"))
|
|
assert.Contains(t, sc.resp.Body.String(), "<title>Grafana - Error</title>")
|
|
})
|
|
})
|
|
}
|
|
|
|
func panicHandler(c *models.ReqContext) {
|
|
panic("Handler has panicked")
|
|
}
|
|
|
|
func recoveryScenario(t *testing.T, desc string, url string, fn scenarioFunc) {
|
|
t.Run(desc, func(t *testing.T) {
|
|
cfg := setting.NewCfg()
|
|
cfg.ErrTemplateName = "error-template"
|
|
sc := &scenarioContext{
|
|
t: t,
|
|
url: url,
|
|
cfg: cfg,
|
|
}
|
|
|
|
viewsPath, err := filepath.Abs("../../public/views")
|
|
require.NoError(t, err)
|
|
|
|
sc.m = web.New()
|
|
sc.m.UseMiddleware(Recovery(cfg))
|
|
|
|
sc.m.Use(AddDefaultResponseHeaders(cfg))
|
|
sc.m.UseMiddleware(web.Renderer(viewsPath, "[[", "]]"))
|
|
|
|
sc.userAuthTokenService = auth.NewFakeUserAuthTokenService()
|
|
sc.remoteCacheService = remotecache.NewFakeStore(t)
|
|
|
|
contextHandler := getContextHandler(t, nil, nil, nil, nil, nil)
|
|
sc.m.Use(contextHandler.Middleware)
|
|
// mock out gc goroutine
|
|
sc.m.Use(OrgRedirect(cfg, sc.userService))
|
|
|
|
sc.defaultHandler = func(c *models.ReqContext) {
|
|
sc.context = c
|
|
if sc.handlerFunc != nil {
|
|
sc.handlerFunc(sc.context)
|
|
}
|
|
}
|
|
|
|
sc.m.Get(url, sc.defaultHandler)
|
|
|
|
fn(t, sc)
|
|
})
|
|
}
|