func (s *S) TestRunDelayedHandlerWithHandler(c *check.C) {
recorder := httptest.NewRecorder()
request, err := http.NewRequest("GET", "/", nil)
c.Assert(err, check.IsNil)
h, log := doHandler()
context.SetDelayedHandler(request, h)
runDelayedHandler(recorder, request)
c.Assert(log.called, check.Equals, true)
}
func (r *delayedRouter) ServeHTTP(w http.ResponseWriter, req *http.Request) {
var match mux.RouteMatch
if !r.Match(req, &match) {
http.NotFound(w, req)
return
}
r.registerVars(req, match.Vars)
context.SetDelayedHandler(req, match.Handler)
}
func (s *S) TestAppLockMiddlewareDoesNothingForExcludedHandlers(c *check.C) {
recorder := httptest.NewRecorder()
request, err := http.NewRequest("POST", "/?:app=abc", nil)
c.Assert(err, check.IsNil)
finalHandler := http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {})
context.SetDelayedHandler(request, finalHandler)
h, log := doHandler()
m := &appLockMiddleware{
excludedHandlers: []http.Handler{finalHandler},
}
m.ServeHTTP(recorder, request, h)
c.Assert(log.called, check.Equals, true)
}