- Updated unit test cases

- Changed unhandled error to runtime error (go like keyword)
- Not found route will execute all preSend and postSend tasks

Author: roshangade

api.go

@@ -153,8 +153,8 @@ func (api *API) UnhandledException(handle Handler) {
 
 // error variables to handle expected errors
 var (
-	ErrCodeNotFound          = "URL_NOT_FOUND"
-	ErrCodeUncaughtException = "UNCAUGHT_EXCEPTION"
+	ErrCodeNotFound     = "URL_NOT_FOUND"
+	ErrCodeRuntimeError = "RUNTIME_ERROR"
 )
 
 // It's required handle for http module.
@@ -175,7 +175,7 @@ func (api *API) ServeHTTP(res http.ResponseWriter, req *http.Request) {
 		err := recover()
 		if err != nil {
 			if !ctx.end {
-				ctx.code = ErrCodeUncaughtException
+				ctx.code = ErrCodeRuntimeError
 				ctx.err = fmt.Errorf("%v", err)
 				ctx.unhandledException()
 			}

api_test.go

@@ -90,7 +90,7 @@ func TestAPI_Patch(t *testing.T) {
 	validateRoute("Patch", "PATCH", "/:uid", t)
 }
 
-func TestAPI_Exception(t *testing.T) {
+func TestAPI_OnError(t *testing.T) {
 	a.OnError("UID_NOT_FOUND", handle)
 
 	flag := true

context.go

@@ -216,17 +216,16 @@ func (ctx *Context) unhandledException() {
 
 	// NOT FOUND handler
 	if ctx.code == ErrCodeNotFound {
-		http.NotFound(ctx.Response, ctx.Request)
-		return
+		ctx.Status(http.StatusNotFound)
 	}
 
 	if ctx.code != "" || ctx.err != nil {
-		msg := ctx.code
+		msg := "Error Code: " + ctx.code
 		if ctx.err != nil {
-			msg += "\nError: " + ctx.err.Error()
+			msg += "\nError Message: " + ctx.err.Error()
 		}
 		ctx.SetHeader("Content-Type", "text/plain;charset=UTF-8")
-		if ctx.status > 400 {
+		if ctx.status < 400 {
 			ctx.Status(http.StatusInternalServerError)
 		}
 		ctx.Write([]byte(msg))
@@ -238,11 +237,9 @@ func (ctx *Context) recover() {
 	err := recover()
 	if err != nil {
 		//TODO: debugger mode
-		log.Println("Unhandled Error: ", err)
+		log.Println("Runtime Error: ", err)
 		if !ctx.requestSent {
-			ctx.Response.WriteHeader(http.StatusInternalServerError)
-			ctx.Response.Header().Set("Content-Type", "text/plain;charset=UTF-8")
-			_, _ = ctx.Response.Write([]byte("Internal Server Error"))
+			http.Error(ctx.Response, "Internal Server Error", http.StatusInternalServerError)
 		}
 	}
 }

examples/server.go

@@ -32,7 +32,7 @@ func main() {
 		s := time.Now().UnixNano()
 		ctx.PreSend(func() {
 			x := time.Now().UnixNano() - s
-			ctx.SetHeader("X-Runtime", strconv.FormatInt(x/int64(time.Millisecond), 10))
+			ctx.SetHeader("X-Runtime", strconv.FormatInt(x/int64(time.Microsecond), 10))
 		})
 	})
 
@@ -52,7 +52,11 @@ func main() {
 
 	fmt.Println("Starting server.")
 
-	err := http.ListenAndServe(":8080", api)
+	server := http.Server{
+		Addr:    ":8080",
+		Handler: api,
+	}
 
+	err := server.ListenAndServe()
 	fmt.Println(err)
 }

go.mod

@@ -1,3 +1,3 @@
 module github.com/go-rs/rest-api-framework
 
-go 1.12
+go 1.13

namespace_test.go

@@ -88,7 +88,7 @@ func TestNamespace_Patch(t *testing.T) {
 	validateNsRoute("Patch", "PATCH", "/test/:uid", t)
 }
 
-func TestNamespace_Exception(t *testing.T) {
+func TestNamespace_OnError(t *testing.T) {
 	ns.OnError("UID_NOT_FOUND", handle)
 
 	flag := true

render/text_test.go

@@ -20,7 +20,7 @@ func TestText_Write(t *testing.T) {
 		t.Error("Render text is not valid")
 	}
 
-	if w.Header().Get("Content-Type") != "text/plain" {
+	if w.Header().Get("Content-Type") != "text/plain;charset=UTF-8" {
 		t.Error("Content-Type Header is not set.")
 	}
 }