ff30f6e0a4
Signed-off-by: dependabot[bot] <support@github.com> Co-authored-by: dependabot[bot] <49699333+dependabot[bot]@users.noreply.github.com> Co-authored-by: github-actions <41898282+github-actions[bot]@users.noreply.github.com> Co-authored-by: Cool Developer <cool199966@outlook.com> Co-authored-by: marbar3778 <marbar3778@yahoo.com> Co-authored-by: Julien Robert <julien@rbrt.fr>
102 lines
2.9 KiB
Go
102 lines
2.9 KiB
Go
package cache_test
|
|
|
|
import (
|
|
"fmt"
|
|
"testing"
|
|
|
|
dbm "github.com/cosmos/cosmos-db"
|
|
"github.com/cosmos/iavl"
|
|
"github.com/stretchr/testify/require"
|
|
|
|
"cosmossdk.io/log"
|
|
"cosmossdk.io/store/cache"
|
|
"cosmossdk.io/store/cachekv"
|
|
iavlstore "cosmossdk.io/store/iavl"
|
|
"cosmossdk.io/store/types"
|
|
"cosmossdk.io/store/wrapper"
|
|
)
|
|
|
|
func TestGetOrSetStoreCache(t *testing.T) {
|
|
db := wrapper.NewDBWrapper(dbm.NewMemDB())
|
|
mngr := cache.NewCommitKVStoreCacheManager(cache.DefaultCommitKVStoreCacheSize)
|
|
|
|
sKey := types.NewKVStoreKey("test")
|
|
tree := iavl.NewMutableTree(db, 100, false, log.NewNopLogger())
|
|
store := iavlstore.UnsafeNewStore(tree)
|
|
store2 := mngr.GetStoreCache(sKey, store)
|
|
|
|
require.NotNil(t, store2)
|
|
require.Equal(t, store2, mngr.GetStoreCache(sKey, store))
|
|
}
|
|
|
|
func TestUnwrap(t *testing.T) {
|
|
db := wrapper.NewDBWrapper(dbm.NewMemDB())
|
|
mngr := cache.NewCommitKVStoreCacheManager(cache.DefaultCommitKVStoreCacheSize)
|
|
|
|
sKey := types.NewKVStoreKey("test")
|
|
tree := iavl.NewMutableTree(db, 100, false, log.NewNopLogger())
|
|
store := iavlstore.UnsafeNewStore(tree)
|
|
_ = mngr.GetStoreCache(sKey, store)
|
|
|
|
require.Equal(t, store, mngr.Unwrap(sKey))
|
|
require.Nil(t, mngr.Unwrap(types.NewKVStoreKey("test2")))
|
|
}
|
|
|
|
func TestStoreCache(t *testing.T) {
|
|
db := wrapper.NewDBWrapper(dbm.NewMemDB())
|
|
mngr := cache.NewCommitKVStoreCacheManager(cache.DefaultCommitKVStoreCacheSize)
|
|
|
|
sKey := types.NewKVStoreKey("test")
|
|
tree := iavl.NewMutableTree(db, 100, false, log.NewNopLogger())
|
|
store := iavlstore.UnsafeNewStore(tree)
|
|
kvStore := mngr.GetStoreCache(sKey, store)
|
|
|
|
for i := uint(0); i < cache.DefaultCommitKVStoreCacheSize*2; i++ {
|
|
key := []byte(fmt.Sprintf("key_%d", i))
|
|
value := []byte(fmt.Sprintf("value_%d", i))
|
|
|
|
kvStore.Set(key, value)
|
|
|
|
res := kvStore.Get(key)
|
|
require.Equal(t, res, value)
|
|
require.Equal(t, res, store.Get(key))
|
|
|
|
kvStore.Delete(key)
|
|
|
|
require.Nil(t, kvStore.Get(key))
|
|
require.Nil(t, store.Get(key))
|
|
}
|
|
}
|
|
|
|
func TestReset(t *testing.T) {
|
|
db := wrapper.NewDBWrapper(dbm.NewMemDB())
|
|
mngr := cache.NewCommitKVStoreCacheManager(cache.DefaultCommitKVStoreCacheSize)
|
|
|
|
sKey := types.NewKVStoreKey("test")
|
|
tree := iavl.NewMutableTree(db, 100, false, log.NewNopLogger())
|
|
store := iavlstore.UnsafeNewStore(tree)
|
|
store2 := mngr.GetStoreCache(sKey, store)
|
|
|
|
require.NotNil(t, store2)
|
|
require.Equal(t, store2, mngr.GetStoreCache(sKey, store))
|
|
|
|
// reset and check if the cache is gone
|
|
mngr.Reset()
|
|
require.Nil(t, mngr.Unwrap(sKey))
|
|
|
|
// check if the cache is recreated
|
|
require.Equal(t, store2, mngr.GetStoreCache(sKey, store))
|
|
}
|
|
|
|
func TestCacheWrap(t *testing.T) {
|
|
db := wrapper.NewDBWrapper(dbm.NewMemDB())
|
|
mngr := cache.NewCommitKVStoreCacheManager(cache.DefaultCommitKVStoreCacheSize)
|
|
|
|
sKey := types.NewKVStoreKey("test")
|
|
tree := iavl.NewMutableTree(db, 100, false, log.NewNopLogger())
|
|
store := iavlstore.UnsafeNewStore(tree)
|
|
|
|
cacheWrapper := mngr.GetStoreCache(sKey, store).CacheWrap()
|
|
require.IsType(t, &cachekv.Store{}, cacheWrapper)
|
|
}
|