mirror of
https://github.com/zap-proto/fiber.git
synced 2026-10-30 15:06:23 +00:00
43 lines
1.1 KiB
Go
43 lines
1.1 KiB
Go
package cache
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/gofiber/utils/v2"
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func Test_manager_get(t *testing.T) {
|
|
t.Parallel()
|
|
cacheManager := newManager(nil, true)
|
|
t.Run("Item not found in cache", func(t *testing.T) {
|
|
t.Parallel()
|
|
it, err := cacheManager.get(context.Background(), utils.UUIDv4())
|
|
require.ErrorIs(t, err, errCacheMiss)
|
|
assert.Nil(t, it)
|
|
})
|
|
t.Run("Item found in cache", func(t *testing.T) {
|
|
t.Parallel()
|
|
id := utils.UUIDv4()
|
|
cacheItem := cacheManager.acquire()
|
|
cacheItem.body = []byte("test-body")
|
|
require.NoError(t, cacheManager.set(context.Background(), id, cacheItem, 10*time.Second))
|
|
it, err := cacheManager.get(context.Background(), id)
|
|
require.NoError(t, err)
|
|
assert.NotNil(t, it)
|
|
})
|
|
}
|
|
|
|
func Test_manager_logKey(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
redactedManager := newManager(nil, true)
|
|
assert.Equal(t, redactedKey, redactedManager.logKey("secret"))
|
|
|
|
plainManager := newManager(nil, false)
|
|
assert.Equal(t, "secret", plainManager.logKey("secret"))
|
|
}
|