78 lines
2.5 KiB
Go
78 lines
2.5 KiB
Go
package util_test
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
"net/http"
|
|
"testing"
|
|
|
|
"allaboutapps.dev/aw/go-starter/internal/api"
|
|
"allaboutapps.dev/aw/go-starter/internal/api/middleware"
|
|
"allaboutapps.dev/aw/go-starter/internal/test"
|
|
"allaboutapps.dev/aw/go-starter/internal/util"
|
|
"github.com/labstack/echo/v4"
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestCacheControl(t *testing.T) {
|
|
test.WithTestServer(t, func(s *api.Server) {
|
|
path := "/testing-1c2cad5f-7545-4177-9bfe-8dc7ed368b33"
|
|
|
|
s.Echo.GET(path, func(c echo.Context) error {
|
|
cache := util.CacheControlDirectiveFromContext(c.Request().Context())
|
|
|
|
switch {
|
|
case cache.HasDirective(util.CacheControlDirectiveNoCache) && cache.HasDirective(util.CacheControlDirectiveNoStore):
|
|
return c.JSON(http.StatusOK, "no-cache,no-store")
|
|
case cache.HasDirective(util.CacheControlDirectiveNoCache):
|
|
return c.JSON(http.StatusOK, "no-cache")
|
|
case cache.HasDirective(util.CacheControlDirectiveNoStore):
|
|
return c.JSON(http.StatusOK, "no-store")
|
|
}
|
|
|
|
return c.NoContent(http.StatusNoContent)
|
|
}, middleware.CacheControl())
|
|
|
|
cacheControlNoCache := util.CacheControlDirectiveNoCache
|
|
cacheControlNoStore := util.CacheControlDirectiveNoStore
|
|
|
|
header := http.Header{}
|
|
header.Set(util.HTTPHeaderCacheControl, fmt.Sprintf("%s,%s", cacheControlNoStore.String(), cacheControlNoCache.String()))
|
|
|
|
res := test.PerformRequest(t, s, "GET", path, nil, header)
|
|
require.Equal(t, http.StatusOK, res.Result().StatusCode)
|
|
|
|
var resp string
|
|
err := json.NewDecoder(res.Result().Body).Decode(&resp)
|
|
require.NoError(t, err)
|
|
assert.Equal(t, "no-cache,no-store", resp)
|
|
|
|
header.Set(util.HTTPHeaderCacheControl, cacheControlNoCache.String())
|
|
|
|
res = test.PerformRequest(t, s, "GET", path, nil, header)
|
|
require.Equal(t, http.StatusOK, res.Result().StatusCode)
|
|
|
|
err = json.NewDecoder(res.Result().Body).Decode(&resp)
|
|
require.NoError(t, err)
|
|
assert.Equal(t, "no-cache", resp)
|
|
|
|
header.Set(util.HTTPHeaderCacheControl, cacheControlNoStore.String())
|
|
|
|
res = test.PerformRequest(t, s, "GET", path, nil, header)
|
|
require.Equal(t, http.StatusOK, res.Result().StatusCode)
|
|
|
|
err = json.NewDecoder(res.Result().Body).Decode(&resp)
|
|
require.NoError(t, err)
|
|
assert.Equal(t, "no-store", resp)
|
|
|
|
res = test.PerformRequest(t, s, "GET", path, nil, nil)
|
|
assert.Equal(t, http.StatusNoContent, res.Result().StatusCode)
|
|
|
|
header.Set(util.HTTPHeaderCacheControl, "gunther")
|
|
|
|
res = test.PerformRequest(t, s, "GET", path, nil, header)
|
|
assert.Equal(t, http.StatusNoContent, res.Result().StatusCode)
|
|
})
|
|
}
|