Added ctx.NotFound

hoisie · hoisie · commit d5de75a89d99 · 2010-01-06T01:00:39.000-08:00
diff --git a/web.go b/web.go
@@ -55,12 +55,16 @@ func (ctx *Context) Abort(status int, body string) {
 }
 
 func (ctx *Context) Redirect(status int, url string) {
-    //note := "<a href=\"%v\">" + statusText[code] + "</a>.\n"
-
     ctx.SetHeader("Location", url, true)
     ctx.StartResponse(status)
-    ctx.WriteString("")
+    ctx.WriteString("Redirecting to: " + url)
+}
+
+func (ctx *Context) NotFound(message string) {
+    ctx.StartResponse(404)
+    ctx.WriteString(message)
 }
+
 //Sets a cookie -- duration is the amount of time in seconds. 0 = forever
 func (ctx *Context) SetCookie(name string, value string, age int64) {
     if age == 0 {
diff --git a/web_test.go b/web_test.go
@@ -111,6 +111,8 @@ func init() {
         return ""
     })
 
+    Get("/error/notfound/(.*)", func(ctx *Context, message string) { ctx.NotFound(message) })
+
     Post("/posterror/code/(.*)/(.*)", func(ctx *Context, code string, message string) string {
         n, _ := strconv.Atoi(code)
         ctx.Abort(n, message)
@@ -143,6 +145,7 @@ var tests = []Test{
     Test{"GET", "/echo/" + strings.Repeat("0123456789", 100), "", 200, strings.Repeat("0123456789", 100)},
 
     Test{"GET", "/writetest", "", 200, "hello"},
+    Test{"GET", "/error/notfound/notfound", "", 404, "notfound"},
     Test{"GET", "/doesnotexist", "", 404, "Page not found"},
     Test{"POST", "/doesnotexist", "", 404, "Page not found"},
     Test{"GET", "/error/code/500", "", 500, statusText[500]},
