add testcases

redirect_test.go

@@ -7,7 +7,11 @@ package fiber
 import (
 	"bytes"
 	"encoding/hex"
+	"encoding/json"
+	"io"
 	"mime/multipart"
+	"net/http"
+	"net/http/httptest"
 	"net/url"
 	"testing"
 
@@ -477,6 +481,160 @@ func Test_Redirect_parseAndClearFlashMessages(t *testing.T) {
 	require.Empty(t, c.Redirect().messages)
 }
 
+// Test_Redirect_parseAndClearFlashMessages_InvalidHex tests the case where hex decoding fails
+func Test_Redirect_parseAndClearFlashMessages_InvalidHex(t *testing.T) {
+	t.Parallel()
+
+	app := New()
+
+	// Setup request and response
+	c := app.AcquireCtx(&fasthttp.RequestCtx{}).(*DefaultCtx) //nolint:errcheck,forcetypeassert // not needed
+	defer app.ReleaseCtx(c)
+
+	// Create redirect instance
+	r := AcquireRedirect()
+	r.c = c
+
+	// Set invalid hex value in flash cookie
+	c.Request().Header.SetCookie(FlashCookieName, "not-a-valid-hex-string")
+
+	// Call parseAndClearFlashMessages
+	r.parseAndClearFlashMessages()
+
+	// Verify that no flash messages are processed (should be empty)
+	require.Empty(t, r.messages)
+
+	// Release redirect
+	ReleaseRedirect(r)
+}
+
+func Test_Redirect_CompleteFlowWithFlashMessages(t *testing.T) {
+	t.Parallel()
+
+	app := New()
+
+	// First handler that sets flash messages and redirects
+	app.Get("/source", func(c Ctx) error {
+		// Redirect to the target handler
+		return c.Redirect().With("string_message", "Hello, World!").
+			With("number_message", "12345").
+			With("bool_message", "true").
+			To("/target")
+	})
+
+	// Second handler that receives and processes flash messages
+	app.Get("/target", func(c Ctx) error {
+		// Get all flash messages and return them as a JSON response
+		return c.JSON(Map{
+			"string_message": c.Redirect().Message("string_message").Value,
+			"number_message": c.Redirect().Message("number_message").Value,
+			"bool_message":   c.Redirect().Message("bool_message").Value,
+		})
+	})
+
+	// Step 1: Make the initial request to the source route
+	req := httptest.NewRequest(MethodGet, "/source", nil)
+	resp, err := app.Test(req)
+	require.NoError(t, err)
+	require.Equal(t, StatusSeeOther, resp.StatusCode)
+	require.Equal(t, "/target", resp.Header.Get(HeaderLocation))
+
+	// Verify and get the cookie from the response
+	cookies := resp.Cookies()
+	var flashCookie *http.Cookie
+	for _, cookie := range cookies {
+		if cookie.Name == "fiber_flash" {
+			flashCookie = cookie
+			break
+		}
+	}
+	require.NotNil(t, flashCookie, "Flash cookie should be set")
+
+	// Step 2: Make the second request to the target route with the cookie
+	req = httptest.NewRequest(MethodGet, "/target", nil)
+	req.Header.Set("Cookie", flashCookie.Name+"="+flashCookie.Value)
+	resp, err = app.Test(req)
+	require.NoError(t, err)
+	require.Equal(t, StatusOK, resp.StatusCode)
+
+	// Parse the JSON response and verify flash messages
+	body, err := io.ReadAll(resp.Body)
+	require.NoError(t, err)
+
+	var result map[string]any
+	err = json.Unmarshal(body, &result)
+	require.NoError(t, err)
+
+	// Verify all flash messages were received correctly
+	require.Equal(t, "Hello, World!", result["string_message"])
+	require.Equal(t, "12345", result["number_message"]) // JSON numbers are float64
+	require.Equal(t, "true", result["bool_message"])
+}
+
+func Test_Redirect_FlashMessagesWithSpecialChars(t *testing.T) {
+	t.Parallel()
+
+	app := New()
+
+	// Handler that sets flash messages with special characters and redirects
+	app.Get("/special-source", func(c Ctx) error {
+		// Create a large message to test encoding of larger data
+		return c.Redirect().With("null_bytes", "Contains\x00null\x00bytes").
+			With("control_chars", "Contains\r\ncontrol\tcharacters").
+			With("unicode", "Unicode: 你好世界").
+			With("emoji", "Emoji: 🔥🚀😊").
+			To("/special-target")
+	})
+
+	// Target handler that receives the flash messages
+	app.Get("/special-target", func(c Ctx) error {
+		return c.JSON(Map{
+			"null_bytes":    c.Redirect().Message("null_bytes").Value,
+			"control_chars": c.Redirect().Message("control_chars").Value,
+			"unicode":       c.Redirect().Message("unicode").Value,
+			"emoji":         c.Redirect().Message("emoji").Value,
+		})
+	})
+
+	// Step 1: Make the initial request
+	req := httptest.NewRequest(MethodGet, "/special-source", nil)
+	resp, err := app.Test(req)
+	require.NoError(t, err)
+	require.Equal(t, StatusSeeOther, resp.StatusCode)
+	require.Equal(t, "/special-target", resp.Header.Get(HeaderLocation))
+
+	// Get the flash cookie
+	var flashCookie *http.Cookie
+	for _, cookie := range resp.Cookies() {
+		if cookie.Name == "fiber_flash" {
+			flashCookie = cookie
+			break
+		}
+	}
+	require.NotNil(t, flashCookie, "Flash cookie should be set")
+
+	// Step 2: Make the second request with the cookie
+	req = httptest.NewRequest(MethodGet, "/special-target", nil)
+	req.Header.Set("Cookie", flashCookie.Name+"="+flashCookie.Value)
+	resp, err = app.Test(req)
+	require.NoError(t, err)
+	require.Equal(t, StatusOK, resp.StatusCode)
+
+	// Parse and verify the response
+	body, err := io.ReadAll(resp.Body)
+	require.NoError(t, err)
+
+	var result map[string]any
+	err = json.Unmarshal(body, &result)
+	require.NoError(t, err)
+
+	// Verify special character handling
+	require.Equal(t, "Contains\x00null\x00bytes", result["null_bytes"])
+	require.Equal(t, "Contains\r\ncontrol\tcharacters", result["control_chars"])
+	require.Equal(t, "Unicode: 你好世界", result["unicode"])
+	require.Equal(t, "Emoji: 🔥🚀😊", result["emoji"])
+}
+
 // go test -v -run=^$ -bench=Benchmark_Redirect_Route -benchmem -count=4
 func Benchmark_Redirect_Route(b *testing.B) {
 	app := New()