🩹 Fix: Handle error in response writing for CSRF proxy middleware test

pull/3390/head
JIeJaitt 2025-04-02 17:42:30 +08:00
parent f202f4d362
commit 3b3de52e66
2 changed files with 4 additions and 5 deletions

View File

@ -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) {

View File

@ -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()