Test_CacheHeuristicFreshnessAddsWarning113AfterThreshold() — fiber Function Reference

Source Code

middleware/cache/cache_test.go lines 2399–2465

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

	storage := newMutatingStorage(func(key string, val []byte) []byte {
		if strings.HasSuffix(key, "_body") {
			return val
		}

		var it item
		if _, err := it.UnmarshalMsg(val); err != nil {
			return val
		}

		oldDate := time.Now().Add(-25 * time.Hour).UTC()
		sec := oldDate.Unix()
		if sec < 0 {
			sec = 0
		}
		it.date = uint64(sec) //nolint:gosec // safe: sec is clamped to non-negative range

		future := time.Now().Add(48 * time.Hour).UTC()
		expSec := future.Unix()
		if expSec < 0 {
			expSec = 0
		}
		it.exp = uint64(expSec) //nolint:gosec // safe: expSec is clamped to non-negative range
		it.ttl = uint64((48 * time.Hour) / time.Second)

		updated, err := it.MarshalMsg(nil)
		if err != nil {
			return val
		}

		return updated
	})

	app := fiber.New()
	app.Use(New(Config{
		Expiration: 2 * time.Second,
		Storage:    storage,
	}))

	var count int
	app.Get("/", func(c fiber.Ctx) error {
		count++
		return c.SendString("body" + strconv.Itoa(count))
	})

	resp, err := app.Test(httptest.NewRequest(fiber.MethodGet, "/", http.NoBody))
	require.NoError(t, err)
	require.Equal(t, cacheMiss, resp.Header.Get("X-Cache"))

	resp, err = app.Test(httptest.NewRequest(fiber.MethodGet, "/", http.NoBody))
	require.NoError(t, err)
	require.Equal(t, cacheHit, resp.Header.Get("X-Cache"))

	warnings := resp.Header.Values(fiber.HeaderWarning)
	require.NotEmpty(t, warnings)
	found := false
	for _, w := range warnings {
		if strings.Contains(w, "113") {
			found = true
			break
		}
	}
	require.True(t, found, "warning 113 not found in %v", warnings)
}