Added test cases for ServeHTTP method

Author: roshangade

api_test.go

@@ -6,6 +6,9 @@
 package rest
 
 import (
+	"io/ioutil"
+	"net/http"
+	"net/http/httptest"
 	"testing"
 )
 
@@ -26,6 +29,12 @@ func validateRoute(fun string, method string, url string, t *testing.T) {
 	}
 }
 
+func TestAPI_Route(t *testing.T) {
+	a.Route("GET", "/greeting", handle)
+
+	validateRoute("Route", "GET", "/greeting", t)
+}
+
 func TestAPI_Use(t *testing.T) {
 	a.Use(handle)
 
@@ -97,3 +106,30 @@ func TestAPI_Exception(t *testing.T) {
 		t.Error("API: Exception is not working properly")
 	}
 }
+
+func TestAPI_ServeHTTP(t *testing.T) {
+	var _api API
+
+	_api.Get("/", func(ctx *Context) {
+		ctx.JSON(`{"message": "Hello World!"}`)
+	})
+
+	dummy := httptest.NewServer(_api)
+	defer dummy.Close()
+
+	res, err := http.Get(dummy.URL)
+
+	if err != nil {
+		t.Error("ServeHTTP error")
+	}
+
+	greeting, err := ioutil.ReadAll(res.Body)
+	res.Body.Close()
+	if err != nil {
+		t.Error("ServeHTTP error")
+	}
+
+	if string(greeting) != `{"message": "Hello World!"}` {
+		t.Error("Response does not match")
+	}
+}