feat: post setting response status code

handler.go

@@ -14,6 +14,7 @@ func Handler(handler EasyHandler) MuxHandler {
 		response := NewResponse(responseWriter)
 
 		receivedResponse := handler(request, response)
+		responseWriter.WriteHeader(receivedResponse.GetStatus())
 		for key, value := range receivedResponse.GetHeaders() {
 			responseWriter.Header().Set(key, value)
 		}

response.go

@@ -6,8 +6,9 @@ import (
 )
 
 type Response struct {
-	responseWriter http.ResponseWriter
 	headers        map[string]string
+	responseWriter http.ResponseWriter
+	status         int
 }
 
 func NewResponse(responseWriter http.ResponseWriter) Response {
@@ -47,10 +48,18 @@ func (response *Response) GetHeaderOrDefaultValue(key string, defaultValue strin
 }
 
 func (response *Response) SetStatus(status int) *Response {
-	response.responseWriter.WriteHeader(status)
+	response.status = status
 	return response
 }
 
+func (response *Response) GetStatus() int {
+	if response.status == 0 {
+		response.status = http.StatusOK
+	}
+
+	return response.status
+}
+
 func (response *Response) Json(data interface{}) *Response {
 	response.SetHeader("Content-Type", "application/json")
 

response_test.go

@@ -25,7 +25,13 @@ func TestResponse_SetStatus(t *testing.T) {
 	response.SetResponseWriter(responseWriter).
 		SetStatus(http.StatusCreated)
 
-	assert.Equal(t, http.StatusCreated, responseWriter.Code)
+	assert.Equal(t, http.StatusCreated, response.GetStatus())
+}
+
+func TestResponse_GetStatus_IfNoStatusDefined(t *testing.T) {
+	response := Response{}
+
+	assert.Equal(t, http.StatusOK, response.GetStatus())
 }
 
 func TestResponse_Json(t *testing.T) {