Added NotAuthorized function, which gives a 403 response, and the respective tests. Also added additional logging for route/response code testing

Refactored to include Unauthorized and Forbidden methods, instead of NotAuthorized

Squash previous two commits, added Forbidden and Unauthorized methods

Author: pbdeuchler

web.go

@@ -65,6 +65,16 @@ func (ctx *Context) NotFound(message string) {
     ctx.ResponseWriter.Write([]byte(message))
 }
 
+//Unauthorized writes a 401 HTTP response
+func (ctx *Context) Unauthorized() {
+    ctx.ResponseWriter.WriteHeader(401)
+}
+
+//Forbidden writes a 403 HTTP response
+func (ctx *Context) Forbidden() {
+    ctx.ResponseWriter.WriteHeader(403)
+}
+
 // ContentType sets the Content-Type header for an HTTP response.
 // For example, ctx.ContentType("json") sets the content-type to "application/json"
 // If the supplied value contains a slash (/) it is set as the Content-Type

web_test.go

@@ -145,6 +145,12 @@ func init() {
 
     Get("/error/notfound/(.*)", func(ctx *Context, message string) { ctx.NotFound(message) })
 
+    Get("/error/unauthorized", func(ctx *Context) { ctx.Unauthorized() })
+    Post("/error/unauthorized", func(ctx *Context) { ctx.Unauthorized() })
+
+    Get("/error/forbidden", func(ctx *Context) { ctx.Forbidden() })
+    Post("/error/forbidden", func(ctx *Context) { ctx.Forbidden() })
+
     Post("/posterror/code/(.*)/(.*)", func(ctx *Context, code string, message string) string {
         n, _ := strconv.Atoi(code)
         ctx.Abort(n, message)
@@ -224,6 +230,10 @@ var tests = []Test{
     //long url
     {"GET", "/echo/" + strings.Repeat("0123456789", 100), nil, "", 200, strings.Repeat("0123456789", 100)},
     {"GET", "/writetest", nil, "", 200, "hello"},
+    {"GET", "/error/unauthorized", nil, "", 401, ""},
+    {"POST", "/error/unauthorized", nil, "", 401, ""},
+    {"GET", "/error/forbidden", nil, "", 403, ""},
+    {"POST", "/error/forbidden", nil, "", 403, ""},
     {"GET", "/error/notfound/notfound", nil, "", 404, "notfound"},
     {"GET", "/doesnotexist", nil, "", 404, "Page not found"},
     {"POST", "/doesnotexist", nil, "", 404, "Page not found"},
@@ -278,10 +288,10 @@ func TestRouting(t *testing.T) {
         resp := getTestResponse(test.method, test.path, test.body, test.headers, nil)
 
         if resp.statusCode != test.expectedStatus {
-            t.Fatalf("expected status %d got %d", test.expectedStatus, resp.statusCode)
+            t.Fatalf("%v(%v) expected status %d got %d", test.method, test.path, test.expectedStatus, resp.statusCode)
         }
         if resp.body != test.expectedBody {
-            t.Fatalf("expected %q got %q", test.expectedBody, resp.body)
+            t.Fatalf("%v(%v) expected %q got %q", test.method, test.path, test.expectedBody, resp.body)
         }
         if cl, ok := resp.headers["Content-Length"]; ok {
             clExp, _ := strconv.Atoi(cl[0])