Added check in JSON render for int, float64 & bool

Added unit test cases for render functions

Author: roshangade

render/json.go

@@ -29,7 +29,10 @@ var (
  * JSON Write
  */
 func (j JSON) Write(w http.ResponseWriter) (data []byte, err error) {
-	if reflect.TypeOf(j.Body).String() == "string" {
+	_type := reflect.TypeOf(j.Body).String()
+	if _type == "int" || _type == "float64" || _type == "bool" {
+		err = invalidJson
+	} else if _type == "string" {
 		data, err = json.RawMessage(j.Body.(string)).MarshalJSON()
 	} else {
 		data, err = json.Marshal(j.Body)

render/json_test.go

@@ -0,0 +1,75 @@
+/*!
+ * rest-api-framework
+ * Copyright(c) 2019 Roshan Gade
+ * MIT Licensed
+ */
+package render
+
+import (
+	"net/http/httptest"
+	"testing"
+)
+
+func TestJSON_Write1(t *testing.T) {
+	json := JSON{
+		Body: "Hello World",
+	}
+	w := httptest.NewRecorder()
+	_, err := json.Write(w)
+
+	if err != invalidJson {
+		t.Error("Should not render text")
+	}
+
+	if w.Header().Get("Content-Type") != "" {
+		t.Error("Content-Type Header is not set.")
+	}
+}
+
+func TestJSON_Write2(t *testing.T) {
+	json := JSON{
+		Body: "{\"Message\":\"Hello World\"}",
+	}
+	w := httptest.NewRecorder()
+	_, err := json.Write(w)
+
+	if err != nil {
+		t.Error("Should not throw an error")
+	}
+
+	if w.Header().Get("Content-Type") != "application/json" {
+		t.Error("Content-Type Header is not set.")
+	}
+}
+
+func TestJSON_Write3(t *testing.T) {
+	json := JSON{
+		Body: make(map[string]string),
+	}
+	w := httptest.NewRecorder()
+	_, err := json.Write(w)
+
+	if err != nil {
+		t.Error("Should not throw an error")
+	}
+
+	if w.Header().Get("Content-Type") != "application/json" {
+		t.Error("Content-Type Header is not set.")
+	}
+}
+
+func TestJSON_Write4(t *testing.T) {
+	json := JSON{
+		Body: true,
+	}
+	w := httptest.NewRecorder()
+	_, err := json.Write(w)
+
+	if err != invalidJson {
+		t.Error("Should not throw an error")
+	}
+
+	if w.Header().Get("Content-Type") != "" {
+		t.Error("Content-Type Header is not set.")
+	}
+}

render/text_test.go

@@ -0,0 +1,27 @@
+/*!
+ * rest-api-framework
+ * Copyright(c) 2019 Roshan Gade
+ * MIT Licensed
+ */
+package render
+
+import (
+	"net/http/httptest"
+	"testing"
+)
+
+func TestText_Write(t *testing.T) {
+	txt := Text{
+		Body: "Hello World",
+	}
+	w := httptest.NewRecorder()
+	body, err := txt.Write(w)
+
+	if err != nil || string(body) != txt.Body {
+		t.Error("Render text is not valid")
+	}
+
+	if w.Header().Get("Content-Type") != "text/plain" {
+		t.Error("Content-Type Header is not set.")
+	}
+}

version.txt

@@ -1 +1 @@
-0.0.1-alpha.2
+0.0.1-alpha.3