TestCacheStorageOperationsObserveRequestContext() — fiber Function Reference

Source Code

middleware/cache/cache_test.go lines 411–483

func TestCacheStorageOperationsObserveRequestContext(t *testing.T) {
	t.Parallel()

	storage := newContextRecorderStorage()
	app := fiber.New()

	app.Use(func(c fiber.Ctx) error {
		ctxLabel := string(c.Request().Header.Peek("X-Context"))
		if ctxLabel == "" {
			return c.Next()
		}

		canceled := string(c.Request().Header.Peek("X-Cancel")) == "true"
		if canceled {
			c.SetContext(canceledContextWithMarker(ctxLabel))
		} else {
			c.SetContext(contextWithMarker(ctxLabel))
		}
		return c.Next()
	})

	app.Use(New(Config{Storage: storage, Expiration: time.Minute}))

	app.Get("/cache", func(c fiber.Ctx) error {
		return c.SendString("payload")
	})

	firstReq := httptest.NewRequest(fiber.MethodGet, "/cache", http.NoBody)
	firstReq.Header.Set("X-Context", "store")
	firstReq.Header.Set("X-Cancel", "true")

	resp, err := app.Test(firstReq)
	require.NoError(t, err)
	require.Equal(t, fiber.StatusOK, resp.StatusCode)

	secondReq := httptest.NewRequest(fiber.MethodGet, "/cache", http.NoBody)
	secondReq.Header.Set("X-Context", "fetch")
	secondReq.Header.Set("X-Cancel", "true")

	resp, err = app.Test(secondReq)
	require.NoError(t, err)
	require.Equal(t, fiber.StatusOK, resp.StatusCode)

	setRecords := storage.recordedSets()
	require.Len(t, setRecords, 2)
	for _, rec := range setRecords {
		require.Contains(t, []string{"/cache_GET", "/cache_GET_body"}, rec.key)
		require.Equal(t, "store", rec.value)
		require.True(t, rec.canceled)
	}

	getRecords := storage.recordedGets()
	require.NotEmpty(t, getRecords)

	var fetchEntry, fetchBody bool
	for _, rec := range getRecords {
		if rec.value != "fetch" {
			continue
		}

		if rec.key == "/cache_GET" {
			require.True(t, rec.canceled)
			fetchEntry = true
		}
		if rec.key == "/cache_GET_body" {
			require.True(t, rec.canceled)
			fetchBody = true
		}
	}

	require.True(t, fetchEntry, "expected cached entry retrieval to observe request context")
	require.True(t, fetchBody, "expected cached body retrieval to observe request context")
}