Test_Redirect_CompleteFlowWithFlashMessages() — fiber Function Reference

Source Code

redirect_test.go lines 575–638

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", http.NoBody)
	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", 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 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"])
}