Test_Redirect_FlashMessagesWithSpecialChars() — fiber Function Reference

Source Code

redirect_test.go lines 640–704

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", http.NoBody)
	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", http.NoBody)
	req.Header.Set("Cookie", flashCookie.Name+"="+flashCookie.Value)
	resp, err = app.Test(req)
	require.NoError(t, err)
	require.Equal(t, StatusOK, resp.StatusCode)

	assertFlashCookieCleared(t, resp.Header.Get(HeaderSetCookie))

	// 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"])
}