Esempio n. 1
0
func Test_recovery_production(t *testing.T) {
	// Setup server & test server
	ts := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
		request := createRequestContext(r, w)
		defer recovery(request, false)
		panic(util.Status404())
	}))
	defer ts.Close()

	response, _ := http.Post(ts.URL, "application/x-www-form-urlencoded", strings.NewReader("key=value"))
	if response.StatusCode != 404 {
		t.Errorf(test.ExpectedNumberButFoundNumber, 404, response.StatusCode)
	} else {
		data, _ := ioutil.ReadAll(response.Body)
		response.Body.Close()
		fmt.Println(string(data))
	}
}
Esempio n. 2
0
////////////////////////////////////////////////////////////////////////////////////////////////////
// ServeHTTP handle HTTP request and HTTP response.
func (s *Server) ServeHTTP(w http.ResponseWriter, r *http.Request) {
	request := createRequestContext(r, w)
	oauth := createOAuthContext(request)

	// Handle error
	defer recovery(request, s.sandbox)

	/* Condition validation: validate request method */
	if !methodsValidation.MatchString(request.Method) {
		panic(util.Status405())
	}

	// Should redirect request to static folder or not?
	if request.Method == Get && len(Cfg.StaticFolders) > 0 {
		for prefix, folder := range Cfg.StaticFolders {
			if path := request.Path; strings.HasPrefix(path, prefix) {
				path = strings.Replace(path, prefix, folder, 1)
				if file, err := os.Open(path); err == nil {
					defer file.Close()

					if info, _ := file.Stat(); !info.IsDir() {
						http.ServeContent(w, r, path, info.ModTime(), file)
						return
					}
				}
				panic(util.Status404())
			}
		}
	}

	// Find route to handle request
	if route, pathParams := s.router.matchRoute(request, oauth); route != nil {
		if pathParams != nil {
			request.PathParams = pathParams
		}
		route.invokeHandler(request, oauth)
	} else {
		panic(util.Status503())
	}
}