From 41f2669ebcc24dbbef7dcca705a4cd75b7c43f28 Mon Sep 17 00:00:00 2001
From: "Alireza (Pure)" <alireza1377eftekhari@gmail.com>
Date: Mon, 2 Jan 2023 07:08:53 +0330
Subject: [PATCH] console logger HTTP status bug fixed and the corresponding
 unit test added (#3453)

---
 response_writer.go      |  1 +
 response_writer_test.go | 18 ++++++++++++++++++
 2 files changed, 19 insertions(+)

diff --git a/response_writer.go b/response_writer.go
index 43e828d..5cec1f6 100644
--- a/response_writer.go
+++ b/response_writer.go
@@ -61,6 +61,7 @@ func (w *responseWriter) WriteHeader(code int) {
 	if code > 0 && w.status != code {
 		if w.Written() {
 			debugPrint("[WARNING] Headers were already written. Wanted to override status code %d with %d", w.status, code)
+			return
 		}
 		w.status = code
 	}
diff --git a/response_writer_test.go b/response_writer_test.go
index 57d163c..6fa5ec7 100644
--- a/response_writer_test.go
+++ b/response_writer_test.go
@@ -132,3 +132,21 @@ func TestResponseWriterFlush(t *testing.T) {
 	assert.NoError(t, err)
 	assert.Equal(t, http.StatusInternalServerError, resp.StatusCode)
 }
+
+func TestResponseWriterStatusCode(t *testing.T) {
+	testWriter := httptest.NewRecorder()
+	writer := &responseWriter{}
+	writer.reset(testWriter)
+	w := ResponseWriter(writer)
+
+	w.WriteHeader(http.StatusOK)
+	w.WriteHeaderNow()
+
+	assert.Equal(t, http.StatusOK, w.Status())
+	assert.True(t, w.Written())
+
+	w.WriteHeader(http.StatusUnauthorized)
+
+	// status must be 200 although we tried to change it
+	assert.Equal(t, http.StatusOK, w.Status())
+}