From 543f100dfed80dff0b5993c184047f220a60ff04 Mon Sep 17 00:00:00 2001 From: "mayur.popli" Date: Tue, 25 Mar 2025 03:10:35 +0530 Subject: [PATCH] fix: error coverage --- pkg/response/response.go | 12 ++++++++++++ pkg/response/response_test.go | 7 +++++++ 2 files changed, 19 insertions(+) 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 +}