Test_Cache_MaxBytes_DeletionFailureRestoresTracking() — fiber Function Reference

Source Code

middleware/cache/cache_test.go lines 3856–3912

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

	storage := newFailingCacheStorage()

	app := fiber.New()
	app.Use(New(Config{
		MaxBytes:   4,
		Expiration: 1 * time.Hour,
		Storage:    storage,
	}))

	app.Get("/:name", func(c fiber.Ctx) error {
		return c.SendString("data")
	})

	// Seed the cache with a single entry
	rsp, err := app.Test(httptest.NewRequest(fiber.MethodGet, "/first", http.NoBody))
	require.NoError(t, err)
	require.Equal(t, cacheMiss, rsp.Header.Get("X-Cache"))

	var storedKeys []string
	storage.mu.Lock()
	for key := range storage.data {
		storedKeys = append(storedKeys, key)
		if strings.Contains(key, "/first") {
			storage.errs["del|"+key] = errors.New("delete failed")
		}
	}
	storage.mu.Unlock()
	t.Logf("stored keys after first cache: %v", storedKeys)

	// Next request triggers eviction; deletion failure should surface an error
	rsp, err = app.Test(httptest.NewRequest(fiber.MethodGet, "/second", http.NoBody))
	require.NoError(t, err)
	body, err := io.ReadAll(rsp.Body)
	require.NoError(t, err)
	require.Equal(t, fiber.StatusInternalServerError, rsp.StatusCode)
	require.Contains(t, string(body), "failed to delete key")
	require.NoError(t, rsp.Body.Close())
	var remainingKeys []string
	storage.mu.RLock()
	for key := range storage.data {
		remainingKeys = append(remainingKeys, key)
	}
	storage.mu.RUnlock()
	t.Logf("stored keys after deletion failure: %v", remainingKeys)
	storage.mu.Lock()
	storage.errs = make(map[string]error)
	storage.mu.Unlock()

	// Another request should succeed and be cacheable after restoring heap tracking
	rsp, err = app.Test(httptest.NewRequest(fiber.MethodGet, "/third", http.NoBody))
	require.NoError(t, err)
	require.Equal(t, cacheMiss, rsp.Header.Get("X-Cache"))
	require.NoError(t, rsp.Body.Close())
}