Function Test_CacheClampsInvalidStoredDate

middleware/cache/cache_test.go:2674–2723

(t *testing.T)

}

func Test_CacheClampsInvalidStoredDate(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
		}

		it.date = uint64(math.MaxInt64) + 1024
		updated, err := it.MarshalMsg(nil)
		if err != nil {
			return val
		}

		return updated
	})

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

	app.Get("/", func(c fiber.Ctx) error {
		c.Set(fiber.HeaderCacheControl, "public, max-age=60")
		return c.SendString("ok")
	})

	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"))

	parsedDate, err := http.ParseTime(resp.Header.Get(fiber.HeaderDate))
	require.NoError(t, err)
	require.WithinDuration(t, time.Now(), parsedDate, time.Minute)

	ageVal, err := strconv.Atoi(resp.Header.Get(fiber.HeaderAge))
	require.NoError(t, err)
	require.Less(t, ageVal, 60)
	require.GreaterOrEqual(t, ageVal, 0)
}

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