feat(store/v2): add the commitment store (#18526)

This commit is contained in:
cool-developer
2023-11-22 14:17:48 +00:00
committed by GitHub
parent 6ca0b2d1aa
commit fc4ce6c778
21 changed files with 349 additions and 164 deletions
+15 -8
View File
@@ -9,10 +9,13 @@ import (
"cosmossdk.io/log"
"cosmossdk.io/store/v2"
"cosmossdk.io/store/v2/commitment"
"cosmossdk.io/store/v2/commitment/iavl"
"cosmossdk.io/store/v2/storage/sqlite"
)
const defaultStoreKey = "default"
type PruningTestSuite struct {
suite.Suite
@@ -34,7 +37,9 @@ func (s *PruningTestSuite) SetupTest() {
ss, err := sqlite.New(s.T().TempDir())
s.Require().NoError(err)
sc := iavl.NewIavlTree(dbm.NewMemDB(), log.NewNopLogger(), iavl.DefaultConfig())
tree := iavl.NewIavlTree(dbm.NewMemDB(), log.NewNopLogger(), iavl.DefaultConfig())
sc, err := commitment.NewCommitStore(map[string]commitment.Tree{"default": tree}, logger)
s.Require().NoError(err)
s.manager = NewManager(logger, ss, sc)
s.ss = ss
@@ -57,9 +62,11 @@ func (s *PruningTestSuite) TestPruning() {
for i := uint64(0); i < latestVersion; i++ {
version := i + 1
cs := store.NewChangeset()
cs.Add([]byte("key"), []byte(fmt.Sprintf("value%d", version)))
cs := store.NewChangeset(map[string]store.KVPairs{defaultStoreKey: {}})
cs.AddKVPair(defaultStoreKey, store.KVPair{
Key: []byte("key"),
Value: []byte(fmt.Sprintf("value%d", version)),
})
err := s.sc.WriteBatch(cs)
s.Require().NoError(err)
@@ -75,22 +82,22 @@ func (s *PruningTestSuite) TestPruning() {
s.manager.Stop()
// check the store for the version 96
val, err := s.ss.Get("", latestVersion-4, []byte("key"))
val, err := s.ss.Get(defaultStoreKey, latestVersion-4, []byte("key"))
s.Require().NoError(err)
s.Require().Equal([]byte("value96"), val)
// check the store for the version 50
val, err = s.ss.Get("", 50, []byte("key"))
val, err = s.ss.Get(defaultStoreKey, 50, []byte("key"))
s.Require().Error(err)
s.Require().Nil(val)
// check the commitment for the version 96
proof, err := s.sc.GetProof(latestVersion-4, []byte("key"))
proof, err := s.sc.GetProof(defaultStoreKey, latestVersion-4, []byte("key"))
s.Require().NoError(err)
s.Require().NotNil(proof.GetExist())
// check the commitment for the version 95
proof, err = s.sc.GetProof(latestVersion-5, []byte("key"))
proof, err = s.sc.GetProof(defaultStoreKey, latestVersion-5, []byte("key"))
s.Require().Error(err)
s.Require().Nil(proof)
}