From 3b3de52e66ae4c434c4cc0b4ed621843a0d241f8 Mon Sep 17 00:00:00 2001
From: JIeJaitt <498938874@qq.com>
Date: Wed, 2 Apr 2025 17:42:30 +0800
Subject: [PATCH] =?UTF-8?q?=F0=9F=A9=B9=20Fix:=20Handle=20error=20in=20res?=
 =?UTF-8?q?ponse=20writing=20for=20CSRF=20proxy=20middleware=20test?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 middleware/csrf/csrf.go      | 4 +---
 middleware/csrf/csrf_test.go | 5 +++--
 2 files changed, 4 insertions(+), 5 deletions(-)

diff --git a/middleware/csrf/csrf.go b/middleware/csrf/csrf.go
index c2484d30..aab36685 100644
--- a/middleware/csrf/csrf.go
+++ b/middleware/csrf/csrf.go
@@ -86,9 +86,7 @@ func New(config ...Config) fiber.Handler {
 
 	// Return new handler
 	return func(c fiber.Ctx) error {
-		var (
-			err error
-		)
+		var err error
 
 		// Don't execute middleware if Next returns true
 		if cfg.Next != nil && cfg.Next(c) {
diff --git a/middleware/csrf/csrf_test.go b/middleware/csrf/csrf_test.go
index fbc03c38..8355d2e5 100644
--- a/middleware/csrf/csrf_test.go
+++ b/middleware/csrf/csrf_test.go
@@ -1574,9 +1574,10 @@ func Test_CSRF_With_Proxy_Middleware(t *testing.T) {
 	t.Parallel()
 
 	// 1. Create a target server that the proxy will forward to
-	targetServer := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+	targetServer := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, _ *http.Request) {
 		w.WriteHeader(http.StatusOK)
-		_, _ = w.Write([]byte("Hello from target"))
+		_, err := w.Write([]byte("Hello from target"))
+		require.NoError(t, err)
 	}))
 	defer targetServer.Close()