03bca7b791
Co-authored-by: cool-developer <51834436+cool-develope@users.noreply.github.com> Co-authored-by: yihuang <huang@crypto.com>
85 lines
2.0 KiB
Go
85 lines
2.0 KiB
Go
package commitment
|
|
|
|
import (
|
|
"testing"
|
|
|
|
dbm "github.com/cosmos/cosmos-db"
|
|
"github.com/stretchr/testify/require"
|
|
|
|
"cosmossdk.io/log"
|
|
"cosmossdk.io/store/v2"
|
|
"cosmossdk.io/store/v2/commitment/iavl"
|
|
)
|
|
|
|
func generateTree(treeType string) store.Tree {
|
|
if treeType == "iavl" {
|
|
cfg := iavl.DefaultConfig()
|
|
db := dbm.NewMemDB()
|
|
tree := iavl.NewIavlTree(db, log.NewNopLogger(), cfg)
|
|
|
|
return tree
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func TestIavlTree(t *testing.T) {
|
|
// generate a new tree
|
|
tree := generateTree("iavl")
|
|
require.NotNil(t, tree)
|
|
|
|
initVersion := tree.GetLatestVersion()
|
|
require.Equal(t, uint64(0), initVersion)
|
|
|
|
// write a batch of version 1
|
|
cs1 := store.NewChangeset()
|
|
cs1.Add([]byte("key1"), []byte("value1"))
|
|
cs1.Add([]byte("key2"), []byte("value2"))
|
|
cs1.Add([]byte("key3"), []byte("value3"))
|
|
|
|
err := tree.WriteBatch(cs1)
|
|
require.NoError(t, err)
|
|
|
|
workingHash := tree.WorkingHash()
|
|
require.NotNil(t, workingHash)
|
|
require.Equal(t, uint64(0), tree.GetLatestVersion())
|
|
|
|
// commit the batch
|
|
commitHash, err := tree.Commit()
|
|
require.NoError(t, err)
|
|
require.Equal(t, workingHash, commitHash)
|
|
require.Equal(t, uint64(1), tree.GetLatestVersion())
|
|
version1Hash := tree.WorkingHash()
|
|
|
|
// write a batch of version 2
|
|
cs2 := store.NewChangeset()
|
|
cs2.Add([]byte("key4"), []byte("value4"))
|
|
cs2.Add([]byte("key5"), []byte("value5"))
|
|
cs2.Add([]byte("key6"), []byte("value6"))
|
|
cs2.Add([]byte("key1"), nil) // delete key1
|
|
err = tree.WriteBatch(cs2)
|
|
require.NoError(t, err)
|
|
workingHash = tree.WorkingHash()
|
|
require.NotNil(t, workingHash)
|
|
commitHash, err = tree.Commit()
|
|
require.NoError(t, err)
|
|
require.Equal(t, workingHash, commitHash)
|
|
|
|
// get proof for key1
|
|
proof, err := tree.GetProof(1, []byte("key1"))
|
|
require.NoError(t, err)
|
|
require.NotNil(t, proof.GetExist())
|
|
|
|
proof, err = tree.GetProof(2, []byte("key1"))
|
|
require.NoError(t, err)
|
|
require.NotNil(t, proof.GetNonexist())
|
|
|
|
// load version 1
|
|
err = tree.LoadVersion(1)
|
|
require.NoError(t, err)
|
|
require.Equal(t, version1Hash, tree.WorkingHash())
|
|
|
|
// close the db
|
|
require.NoError(t, tree.Close())
|
|
}
|