Allowing to define Response headers. Fixes #1 (#10)

* Allowing to define Response headers. Fixes #1

* Allowing array of values for Response headers. Related with #1

Author: joanlopez

README.md

@@ -70,7 +70,11 @@ imposters/create_gopher.json
     },
     "response": {
         "status": 200,
-        "content_type": "application/json",
+        "headers": {
+            "Content-Type": [
+                "application/json"
+            ]
+        },
         "bodyFile": "responses/create_gopher_response.json"
     }
 }

handler.go

@@ -29,7 +29,8 @@ func ImposterHandler(imposter Imposter) http.HandlerFunc {
 			return
 		}
 
-		w.Header().Set("Content-Type", imposter.Response.ContentType)
+
+		writeHeaders(imposter, w)
 		w.WriteHeader(imposter.Response.Status)
 		writeBody(imposter, w)
 	}
@@ -100,6 +101,18 @@ func compareHeaderValues(a, b []string) bool {
 	return true
 }
 
+func writeHeaders(imposter Imposter, w http.ResponseWriter) {
+	if imposter.Response.Headers == nil {
+		return
+	}
+
+	for key, values := range *imposter.Response.Headers {
+		for _, val := range values {
+			w.Header().Add(key, val)
+		}
+	}
+}
+
 func writeBody(imposter Imposter, w http.ResponseWriter) {
 	wb := []byte(imposter.Response.Body)
 

handler_test.go

@@ -45,9 +45,9 @@ func TestImposterHandler(t *testing.T) {
 		expectedBody string
 		statusCode   int
 	}{
-		{"valid imposter with body", Imposter{Request: validRequest, Response: Response{Status: http.StatusOK, ContentType: "application/json", Body: body}}, body, http.StatusOK},
-		{"valid imposter with bodyFile", Imposter{Request: validRequest, Response: Response{Status: http.StatusOK, ContentType: "application/json", BodyFile: &bodyFile}}, string(expectedBodyFileData), http.StatusOK},
-		{"valid imposter with not exists bodyFile", Imposter{Request: validRequest, Response: Response{Status: http.StatusOK, ContentType: "application/json", BodyFile: &bodyFileFake}}, "", http.StatusOK},
+		{"valid imposter with body", Imposter{Request: validRequest, Response: Response{Status: http.StatusOK, Headers: &http.Header{"Content-Type": []string{"application/json"}}, Body: body}}, body, http.StatusOK},
+		{"valid imposter with bodyFile", Imposter{Request: validRequest, Response: Response{Status: http.StatusOK, Headers: &http.Header{"Content-Type": []string{"application/json"}}, BodyFile: &bodyFile}}, string(expectedBodyFileData), http.StatusOK},
+		{"valid imposter with not exists bodyFile", Imposter{Request: validRequest, Response: Response{Status: http.StatusOK, Headers: &http.Header{"Content-Type": []string{"application/json"}}, BodyFile: &bodyFileFake}}, "", http.StatusOK},
 	}
 
 	for _, tt := range dataTest {

imposter.go

@@ -21,5 +21,5 @@ type Response struct {
 	Status      int     `json:"status"`
 	Body        string  `json:"body"`
 	BodyFile    *string `json:"bodyFile"`
-	ContentType string  `json:"content_type"`
+	Headers    *http.Header `json:"headers"`
 }