Set json content type in API GW response when using ctx.JSON() (#10)

JSainsburyPLC · Jan 13, 2021 · e2d63e5 · e2d63e5
1 parent 034c615
commit e2d63e5
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/apigw_handler.go b/apigw_handler.go
@@ -95,6 +95,11 @@ func (c *APIGatewayProxyContext) JSON(statusCode int, body interface{}) error {
 		if err != nil {
 			return err
 		}
+
+		if c.Response.Headers == nil {
+			c.Response.Headers = make(map[string]string)
+		}
+		c.Response.Headers["Content-Type"] = "application/json"
 	}
 	c.Response.StatusCode = statusCode
 	c.Response.Body = string(b)

diff --git a/apigw_handler_test.go b/apigw_handler_test.go
@@ -110,6 +110,7 @@ func TestAPIGatewayProxyContext_JSON(t *testing.T) {
 			},
 			expectedRes: events.APIGatewayProxyResponse{
 				StatusCode: 200,
+				Headers:    map[string]string{"Content-Type": "application/json"},
 				Body:       `{"name":"one","status":"ok"}`,
 			},
 		},