package cache import ( "testing" "git.frostfs.info/TrueCloudLab/frostfs-contract/frostfsid/client" "git.frostfs.info/TrueCloudLab/frostfs-s3-gw/api/data" "git.frostfs.info/TrueCloudLab/frostfs-s3-gw/creds/accessbox" cidtest "git.frostfs.info/TrueCloudLab/frostfs-sdk-go/container/id/test" "git.frostfs.info/TrueCloudLab/frostfs-sdk-go/object" oidtest "git.frostfs.info/TrueCloudLab/frostfs-sdk-go/object/id/test" "github.com/nspcc-dev/neo-go/pkg/crypto/keys" "github.com/nspcc-dev/neo-go/pkg/util" "github.com/stretchr/testify/require" "go.uber.org/zap" "go.uber.org/zap/zaptest/observer" ) func TestAccessBoxCacheType(t *testing.T) { logger, observedLog := getObservedLogger() cache := NewAccessBoxCache(DefaultAccessBoxConfig(logger)) addr := oidtest.Address() box := &accessbox.Box{} var attrs []object.Attribute err := cache.Put(addr, box, attrs) require.NoError(t, err) val := cache.Get(addr) require.Equal(t, box, val.Box) require.Equal(t, attrs, val.Attributes) require.Equal(t, 0, observedLog.Len()) err = cache.cache.Set(addr, "tmp") require.NoError(t, err) assertInvalidCacheEntry(t, cache.Get(addr), observedLog) } func TestBucketsCacheType(t *testing.T) { logger, observedLog := getObservedLogger() cache := NewBucketCache(DefaultBucketConfig(logger)) bktInfo := &data.BucketInfo{Name: "bucket"} err := cache.Put(bktInfo) require.NoError(t, err) val := cache.Get("", bktInfo.Name) require.Equal(t, bktInfo, val) require.Equal(t, 0, observedLog.Len()) err = cache.cache.Set(bktInfo.Name+"."+bktInfo.Zone, "tmp") require.NoError(t, err) assertInvalidCacheEntry(t, cache.Get(bktInfo.Zone, bktInfo.Name), observedLog) } func TestObjectNamesCacheType(t *testing.T) { logger, observedLog := getObservedLogger() cache := NewObjectsNameCache(DefaultObjectsNameConfig(logger)) key := "name" addr := oidtest.Address() err := cache.Put(key, addr) require.NoError(t, err) val := cache.Get(key) require.Equal(t, addr, *val) require.Equal(t, 0, observedLog.Len()) err = cache.cache.Set(key, "tmp") require.NoError(t, err) assertInvalidCacheEntry(t, cache.Get(key), observedLog) } func TestObjectCacheType(t *testing.T) { logger, observedLog := getObservedLogger() cache := New(DefaultObjectsConfig(logger)) addr := oidtest.Address() extObjInfo := &data.ExtendedObjectInfo{ ObjectInfo: &data.ObjectInfo{ ID: addr.Object(), CID: addr.Container(), }, NodeVersion: &data.NodeVersion{ BaseNodeVersion: data.BaseNodeVersion{ FilePath: "obj", Size: 50, }, IsUnversioned: true, }, IsLatest: true, } err := cache.PutObject(extObjInfo) require.NoError(t, err) val := cache.GetObject(addr) require.Equal(t, extObjInfo, val) require.Equal(t, 0, observedLog.Len()) err = cache.cache.Set(addr, "tmp") require.NoError(t, err) assertInvalidCacheEntry(t, cache.GetObject(addr), observedLog) } func TestObjectsListCacheType(t *testing.T) { logger, observedLog := getObservedLogger() cache := NewObjectsListCache(DefaultObjectsListConfig(logger)) cnrID := cidtest.ID() key := ObjectsListKey{cid: cnrID, prefix: "obj"} versions := []*data.NodeVersion{{BaseNodeVersion: data.BaseNodeVersion{OID: oidtest.ID()}}} err := cache.PutVersions(key, versions) require.NoError(t, err) val := cache.GetVersions(key) require.Equal(t, versions, val) require.Equal(t, 0, observedLog.Len()) err = cache.cache.Set(key, "tmp") require.NoError(t, err) assertInvalidCacheEntry(t, cache.GetVersions(key), observedLog) err = cache.cache.Set("key", "tmp") require.NoError(t, err) cache.CleanCacheEntriesContainingObject(key.prefix, cnrID) require.Equal(t, 2, observedLog.Len()) require.Equal(t, observedLog.All()[1].Message, "invalid cache key type") } func TestObjectInfoCacheType(t *testing.T) { logger, observedLog := getObservedLogger() cache := NewSystemCache(DefaultSystemConfig(logger)) key := "key" objInfo := &data.ObjectInfo{Name: key} err := cache.PutObject(key, objInfo) require.NoError(t, err) val := cache.GetObject(key) require.Equal(t, objInfo, val) require.Equal(t, 0, observedLog.Len()) err = cache.cache.Set(key, "tmp") require.NoError(t, err) assertInvalidCacheEntry(t, cache.GetObject(key), observedLog) } func TestCORsCacheType(t *testing.T) { logger, observedLog := getObservedLogger() cache := NewSystemCache(DefaultSystemConfig(logger)) key := "key" cors := &data.CORSConfiguration{} err := cache.PutCORS(key, cors) require.NoError(t, err) val := cache.GetCORS(key) require.Equal(t, cors, val) require.Equal(t, 0, observedLog.Len()) err = cache.cache.Set(key, "tmp") require.NoError(t, err) assertInvalidCacheEntry(t, cache.GetCORS(key), observedLog) } func TestSettingsCacheType(t *testing.T) { logger, observedLog := getObservedLogger() cache := NewSystemCache(DefaultSystemConfig(logger)) key := "key" settings := &data.BucketSettings{Versioning: data.VersioningEnabled} err := cache.PutSettings(key, settings) require.NoError(t, err) val := cache.GetSettings(key) require.Equal(t, settings, val) require.Equal(t, 0, observedLog.Len()) err = cache.cache.Set(key, "tmp") require.NoError(t, err) assertInvalidCacheEntry(t, cache.GetSettings(key), observedLog) } func TestNotificationConfigurationCacheType(t *testing.T) { logger, observedLog := getObservedLogger() cache := NewSystemCache(DefaultSystemConfig(logger)) key := "key" notificationConfig := &data.NotificationConfiguration{} err := cache.PutNotificationConfiguration(key, notificationConfig) require.NoError(t, err) val := cache.GetNotificationConfiguration(key) require.Equal(t, notificationConfig, val) require.Equal(t, 0, observedLog.Len()) err = cache.cache.Set(key, "tmp") require.NoError(t, err) assertInvalidCacheEntry(t, cache.GetNotificationConfiguration(key), observedLog) } func TestFrostFSIDSubjectCacheType(t *testing.T) { logger, observedLog := getObservedLogger() cache := NewFrostfsIDCache(DefaultFrostfsIDConfig(logger)) key, err := util.Uint160DecodeStringLE("4ea976429703418ef00fc4912a409b6a0b973034") require.NoError(t, err) value := &client.SubjectExtended{} err = cache.PutSubject(key, value) require.NoError(t, err) val := cache.GetSubject(key) require.Equal(t, value, val) require.Equal(t, 0, observedLog.Len()) err = cache.cache.Set(key, "tmp") require.NoError(t, err) assertInvalidCacheEntry(t, cache.GetSubject(key), observedLog) } func TestFrostFSIDUserKeyCacheType(t *testing.T) { logger, observedLog := getObservedLogger() cache := NewFrostfsIDCache(DefaultFrostfsIDConfig(logger)) ns, name := "ns", "name" value, err := keys.NewPrivateKey() require.NoError(t, err) err = cache.PutUserKey(ns, name, value.PublicKey()) require.NoError(t, err) val := cache.GetUserKey(ns, name) require.Equal(t, value.PublicKey(), val) require.Equal(t, 0, observedLog.Len()) err = cache.cache.Set(ns+"/"+name, "tmp") require.NoError(t, err) assertInvalidCacheEntry(t, cache.GetUserKey(ns, name), observedLog) } func assertInvalidCacheEntry(t *testing.T, val interface{}, observedLog *observer.ObservedLogs) { require.Nil(t, val) require.Equal(t, 1, observedLog.Len()) require.Equal(t, observedLog.All()[0].Message, "invalid cache entry type") } func getObservedLogger() (*zap.Logger, *observer.ObservedLogs) { loggerCore, observedLog := observer.New(zap.WarnLevel) return zap.New(loggerCore), observedLog }