forked from TrueCloudLab/frostfs-s3-gw
54 lines
1.1 KiB
Go
54 lines
1.1 KiB
Go
package cache
|
|
|
|
import (
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/nspcc-dev/neofs-s3-gw/api/data"
|
|
oid "github.com/nspcc-dev/neofs-sdk-go/object/id"
|
|
objecttest "github.com/nspcc-dev/neofs-sdk-go/object/test"
|
|
"github.com/stretchr/testify/require"
|
|
"go.uber.org/zap"
|
|
)
|
|
|
|
func getTestConfig() *Config {
|
|
return &Config{
|
|
Size: 10,
|
|
Lifetime: 5 * time.Second,
|
|
Logger: zap.NewExample(),
|
|
}
|
|
}
|
|
|
|
func TestCache(t *testing.T) {
|
|
obj := objecttest.Object()
|
|
objID, _ := obj.ID()
|
|
cnrID, _ := obj.ContainerID()
|
|
|
|
var addr oid.Address
|
|
addr.SetContainer(cnrID)
|
|
addr.SetObject(objID)
|
|
|
|
objInfo := &data.ObjectInfo{
|
|
ID: addr.Object(),
|
|
CID: addr.Container(),
|
|
}
|
|
|
|
t.Run("check get", func(t *testing.T) {
|
|
cache := New(getTestConfig())
|
|
err := cache.PutObject(objInfo)
|
|
require.NoError(t, err)
|
|
|
|
actual := cache.GetObject(addr)
|
|
require.Equal(t, objInfo, actual)
|
|
})
|
|
|
|
t.Run("check delete", func(t *testing.T) {
|
|
cache := New(getTestConfig())
|
|
err := cache.PutObject(objInfo)
|
|
require.NoError(t, err)
|
|
|
|
cache.Delete(addr)
|
|
actual := cache.GetObject(addr)
|
|
require.Nil(t, actual)
|
|
})
|
|
}
|