Adding tests for coverage

Signed-off-by: brunodmartins <[email protected]>

Author: brunodmartins

Diff for: middleware/cache/cache_test.go

@@ -697,9 +697,8 @@ func Test_CustomCacheHeader(t *testing.T) {
 	require.Equal(t, cacheMiss, resp.Header.Get("Cache-Status"))
 }
 
-func Test_CacheInvalidation(t *testing.T) {
+func Test_CacheInvalidation_byRequest(t *testing.T) {
 	t.Parallel()
-
 	t.Run("Invalidation by requests", func(t *testing.T) {
 		t.Parallel()
 		app := fiber.New()
@@ -732,7 +731,10 @@ func Test_CacheInvalidation(t *testing.T) {
 		require.NoError(t, err)
 		require.NotEqual(t, body, bodyInvalidate)
 	})
+}
 
+func Test_CacheInvalidation_noCacheEntry(t *testing.T) {
+	t.Parallel()
 	t.Run("Cache Invalidator should not be called if no cache entry exist ", func(t *testing.T) {
 		t.Parallel()
 		app := fiber.New()
@@ -751,6 +753,71 @@ func Test_CacheInvalidation(t *testing.T) {
 	})
 }
 
+func Test_CacheInvalidation_removeFromHeap(t *testing.T) {
+	t.Parallel()
+	t.Run("Invalidate and remove from the heap", func(t *testing.T) {
+		t.Parallel()
+		app := fiber.New()
+		app.Use(New(Config{
+			CacheControl: true,
+			CacheInvalidator: func(c fiber.Ctx) bool {
+				return fiber.Query[bool](c, "invalidate")
+			},
+			MaxBytes: 10 * 1024 * 1024,
+		}))
+
+		app.Get("/", func(c fiber.Ctx) error {
+			return c.SendString(time.Now().String())
+		})
+
+		resp, err := app.Test(httptest.NewRequest(fiber.MethodGet, "/", nil))
+		require.NoError(t, err)
+		body, err := io.ReadAll(resp.Body)
+		require.NoError(t, err)
+
+		respCached, err := app.Test(httptest.NewRequest(fiber.MethodGet, "/", nil))
+		require.NoError(t, err)
+		bodyCached, err := io.ReadAll(respCached.Body)
+		require.NoError(t, err)
+		require.True(t, bytes.Equal(body, bodyCached))
+		require.NotEmpty(t, respCached.Header.Get(fiber.HeaderCacheControl))
+
+		respInvalidate, err := app.Test(httptest.NewRequest(fiber.MethodGet, "/?invalidate=true", nil))
+		require.NoError(t, err)
+		bodyInvalidate, err := io.ReadAll(respInvalidate.Body)
+		require.NoError(t, err)
+		require.NotEqual(t, body, bodyInvalidate)
+	})
+}
+
+func Test_CacheStorage_CustomHeaders(t *testing.T) {
+	t.Parallel()
+	app := fiber.New()
+	app.Use(New(Config{
+		CacheControl: true,
+		Storage:      memory.New(),
+		MaxBytes:     10 * 1024 * 1024,
+	}))
+
+	app.Get("/", func(c fiber.Ctx) error {
+		c.Response().Header.Set("Content-Type", "text/xml")
+		c.Response().Header.Set("Content-Encoding", "utf8")
+		return c.Send([]byte("<xml><value>Test</value></xml>"))
+	})
+
+	resp, err := app.Test(httptest.NewRequest(fiber.MethodGet, "/", nil))
+	require.NoError(t, err)
+	body, err := io.ReadAll(resp.Body)
+	require.NoError(t, err)
+
+	respCached, err := app.Test(httptest.NewRequest(fiber.MethodGet, "/", nil))
+	require.NoError(t, err)
+	bodyCached, err := io.ReadAll(respCached.Body)
+	require.NoError(t, err)
+	require.True(t, bytes.Equal(body, bodyCached))
+	require.NotEmpty(t, respCached.Header.Get(fiber.HeaderCacheControl))
+}
+
 // Because time points are updated once every X milliseconds, entries in tests can often have
 // equal expiration times and thus be in an random order. This closure hands out increasing
 // time intervals to maintain strong ascending order of expiration