diff --git a/pkg/response/response.go b/pkg/response/response.go index bb89236..03beeef 100644 --- a/pkg/response/response.go +++ b/pkg/response/response.go @@ -17,6 +17,18 @@ const ( InvalidRequestErrorType ErrorType = "INVALID_REQUEST" ) +type errorResponseWriter struct{} + +func (e *errorResponseWriter) Header() http.Header { + return http.Header{} +} + +func (e *errorResponseWriter) Write([]byte) (int, error) { + return 0, errors.New("write error") +} + +func (e *errorResponseWriter) WriteHeader(statusCode int) {} + func SendAck(w http.ResponseWriter) { resp := &model.Response{ Message: model.Message{ diff --git a/pkg/response/response_test.go b/pkg/response/response_test.go index 2544bb9..437d779 100644 --- a/pkg/response/response_test.go +++ b/pkg/response/response_test.go @@ -165,3 +165,10 @@ func compareJSON(expected, actual map[string]interface{}) bool { actualBytes, _ := json.Marshal(actual) return bytes.Equal(expectedBytes, actualBytes) } + + +func TestSendAck_WriteError(t *testing.T) { + w := &errorResponseWriter{} + SendAck(w) + // No need to assert, just ensure it doesn't panic +}