2019-02-01 17:03:09 -08:00
|
|
|
package rootmulti
|
2018-08-29 21:50:41 -07:00
|
|
|
|
|
|
|
import (
|
2018-08-31 15:22:37 -07:00
|
|
|
"testing"
|
|
|
|
|
2018-09-12 00:13:14 -07:00
|
|
|
"github.com/stretchr/testify/require"
|
2018-11-04 19:36:35 -08:00
|
|
|
abci "github.com/tendermint/tendermint/abci/types"
|
2019-08-02 06:20:39 -07:00
|
|
|
dbm "github.com/tendermint/tm-db"
|
2018-12-10 06:27:25 -08:00
|
|
|
|
2019-02-01 17:03:09 -08:00
|
|
|
"github.com/cosmos/cosmos-sdk/store/iavl"
|
2019-02-05 10:39:22 -08:00
|
|
|
"github.com/cosmos/cosmos-sdk/store/types"
|
2018-08-29 21:50:41 -07:00
|
|
|
)
|
|
|
|
|
2018-11-04 19:36:35 -08:00
|
|
|
func TestVerifyIAVLStoreQueryProof(t *testing.T) {
|
|
|
|
// Create main tree for testing.
|
|
|
|
db := dbm.NewMemDB()
|
2020-06-22 13:31:33 -07:00
|
|
|
iStore, err := iavl.LoadStore(db, types.CommitID{}, false)
|
2019-02-01 17:03:09 -08:00
|
|
|
store := iStore.(*iavl.Store)
|
2018-11-04 19:36:35 -08:00
|
|
|
require.Nil(t, err)
|
|
|
|
store.Set([]byte("MYKEY"), []byte("MYVALUE"))
|
|
|
|
cid := store.Commit()
|
|
|
|
|
|
|
|
// Get Proof
|
|
|
|
res := store.Query(abci.RequestQuery{
|
|
|
|
Path: "/key", // required path to get key/value+proof
|
|
|
|
Data: []byte("MYKEY"),
|
|
|
|
Prove: true,
|
2018-08-29 21:50:41 -07:00
|
|
|
})
|
2020-08-14 10:58:53 -07:00
|
|
|
require.NotNil(t, res.ProofOps)
|
2018-08-29 21:50:41 -07:00
|
|
|
|
2018-11-04 19:36:35 -08:00
|
|
|
// Verify proof.
|
|
|
|
prt := DefaultProofRuntime()
|
2020-08-14 10:58:53 -07:00
|
|
|
err = prt.VerifyValue(res.ProofOps, cid.Hash, "/MYKEY", []byte("MYVALUE"))
|
2018-11-04 19:36:35 -08:00
|
|
|
require.Nil(t, err)
|
2018-08-29 21:50:41 -07:00
|
|
|
|
2018-11-04 19:36:35 -08:00
|
|
|
// Verify (bad) proof.
|
2020-08-14 10:58:53 -07:00
|
|
|
err = prt.VerifyValue(res.ProofOps, cid.Hash, "/MYKEY_NOT", []byte("MYVALUE"))
|
2018-11-04 19:36:35 -08:00
|
|
|
require.NotNil(t, err)
|
2018-08-29 21:50:41 -07:00
|
|
|
|
2018-11-04 19:36:35 -08:00
|
|
|
// Verify (bad) proof.
|
2020-08-14 10:58:53 -07:00
|
|
|
err = prt.VerifyValue(res.ProofOps, cid.Hash, "/MYKEY/MYKEY", []byte("MYVALUE"))
|
2018-11-04 19:36:35 -08:00
|
|
|
require.NotNil(t, err)
|
2018-08-29 21:50:41 -07:00
|
|
|
|
2018-11-04 19:36:35 -08:00
|
|
|
// Verify (bad) proof.
|
2020-08-14 10:58:53 -07:00
|
|
|
err = prt.VerifyValue(res.ProofOps, cid.Hash, "MYKEY", []byte("MYVALUE"))
|
2018-11-04 19:36:35 -08:00
|
|
|
require.NotNil(t, err)
|
|
|
|
|
|
|
|
// Verify (bad) proof.
|
2020-08-14 10:58:53 -07:00
|
|
|
err = prt.VerifyValue(res.ProofOps, cid.Hash, "/MYKEY", []byte("MYVALUE_NOT"))
|
2018-11-04 19:36:35 -08:00
|
|
|
require.NotNil(t, err)
|
|
|
|
|
|
|
|
// Verify (bad) proof.
|
2020-08-14 10:58:53 -07:00
|
|
|
err = prt.VerifyValue(res.ProofOps, cid.Hash, "/MYKEY", []byte(nil))
|
2018-11-04 19:36:35 -08:00
|
|
|
require.NotNil(t, err)
|
|
|
|
}
|
2018-08-29 21:50:41 -07:00
|
|
|
|
2018-11-04 19:36:35 -08:00
|
|
|
func TestVerifyMultiStoreQueryProof(t *testing.T) {
|
|
|
|
// Create main tree for testing.
|
|
|
|
db := dbm.NewMemDB()
|
2019-02-01 17:03:09 -08:00
|
|
|
store := NewStore(db)
|
|
|
|
iavlStoreKey := types.NewKVStoreKey("iavlStoreKey")
|
2018-11-04 19:36:35 -08:00
|
|
|
|
2019-02-01 17:03:09 -08:00
|
|
|
store.MountStoreWithDB(iavlStoreKey, types.StoreTypeIAVL, nil)
|
2020-01-22 11:52:56 -08:00
|
|
|
require.NoError(t, store.LoadVersion(0))
|
2018-11-04 19:36:35 -08:00
|
|
|
|
2019-02-01 17:03:09 -08:00
|
|
|
iavlStore := store.GetCommitStore(iavlStoreKey).(*iavl.Store)
|
2018-11-04 19:36:35 -08:00
|
|
|
iavlStore.Set([]byte("MYKEY"), []byte("MYVALUE"))
|
|
|
|
cid := store.Commit()
|
|
|
|
|
|
|
|
// Get Proof
|
|
|
|
res := store.Query(abci.RequestQuery{
|
|
|
|
Path: "/iavlStoreKey/key", // required path to get key/value+proof
|
|
|
|
Data: []byte("MYKEY"),
|
|
|
|
Prove: true,
|
2018-08-29 21:50:41 -07:00
|
|
|
})
|
2020-08-14 10:58:53 -07:00
|
|
|
require.NotNil(t, res.ProofOps)
|
2018-08-29 21:50:41 -07:00
|
|
|
|
2018-11-04 19:36:35 -08:00
|
|
|
// Verify proof.
|
|
|
|
prt := DefaultProofRuntime()
|
2020-08-14 10:58:53 -07:00
|
|
|
err := prt.VerifyValue(res.ProofOps, cid.Hash, "/iavlStoreKey/MYKEY", []byte("MYVALUE"))
|
2018-09-12 00:13:14 -07:00
|
|
|
require.Nil(t, err)
|
2018-08-29 21:50:41 -07:00
|
|
|
|
2018-11-04 19:36:35 -08:00
|
|
|
// Verify proof.
|
2020-08-14 10:58:53 -07:00
|
|
|
err = prt.VerifyValue(res.ProofOps, cid.Hash, "/iavlStoreKey/MYKEY", []byte("MYVALUE"))
|
2018-11-04 19:36:35 -08:00
|
|
|
require.Nil(t, err)
|
2018-08-29 21:50:41 -07:00
|
|
|
|
2018-11-04 19:36:35 -08:00
|
|
|
// Verify (bad) proof.
|
2020-08-14 10:58:53 -07:00
|
|
|
err = prt.VerifyValue(res.ProofOps, cid.Hash, "/iavlStoreKey/MYKEY_NOT", []byte("MYVALUE"))
|
2018-11-04 19:36:35 -08:00
|
|
|
require.NotNil(t, err)
|
|
|
|
|
|
|
|
// Verify (bad) proof.
|
2020-08-14 10:58:53 -07:00
|
|
|
err = prt.VerifyValue(res.ProofOps, cid.Hash, "/iavlStoreKey/MYKEY/MYKEY", []byte("MYVALUE"))
|
2018-11-04 19:36:35 -08:00
|
|
|
require.NotNil(t, err)
|
|
|
|
|
|
|
|
// Verify (bad) proof.
|
2020-08-14 10:58:53 -07:00
|
|
|
err = prt.VerifyValue(res.ProofOps, cid.Hash, "iavlStoreKey/MYKEY", []byte("MYVALUE"))
|
2018-11-04 19:36:35 -08:00
|
|
|
require.NotNil(t, err)
|
|
|
|
|
|
|
|
// Verify (bad) proof.
|
2020-08-14 10:58:53 -07:00
|
|
|
err = prt.VerifyValue(res.ProofOps, cid.Hash, "/MYKEY", []byte("MYVALUE"))
|
2018-11-04 19:36:35 -08:00
|
|
|
require.NotNil(t, err)
|
|
|
|
|
|
|
|
// Verify (bad) proof.
|
2020-08-14 10:58:53 -07:00
|
|
|
err = prt.VerifyValue(res.ProofOps, cid.Hash, "/iavlStoreKey/MYKEY", []byte("MYVALUE_NOT"))
|
2018-11-04 19:36:35 -08:00
|
|
|
require.NotNil(t, err)
|
2018-08-29 21:50:41 -07:00
|
|
|
|
2018-11-04 19:36:35 -08:00
|
|
|
// Verify (bad) proof.
|
2020-08-14 10:58:53 -07:00
|
|
|
err = prt.VerifyValue(res.ProofOps, cid.Hash, "/iavlStoreKey/MYKEY", []byte(nil))
|
2018-11-04 19:36:35 -08:00
|
|
|
require.NotNil(t, err)
|
2018-08-30 00:52:17 -07:00
|
|
|
}
|
2018-11-07 00:03:02 -08:00
|
|
|
|
|
|
|
func TestVerifyMultiStoreQueryProofAbsence(t *testing.T) {
|
|
|
|
// Create main tree for testing.
|
|
|
|
db := dbm.NewMemDB()
|
2019-02-01 17:03:09 -08:00
|
|
|
store := NewStore(db)
|
|
|
|
iavlStoreKey := types.NewKVStoreKey("iavlStoreKey")
|
2018-11-07 00:03:02 -08:00
|
|
|
|
2019-02-01 17:03:09 -08:00
|
|
|
store.MountStoreWithDB(iavlStoreKey, types.StoreTypeIAVL, nil)
|
2020-03-20 12:24:18 -07:00
|
|
|
err := store.LoadVersion(0)
|
|
|
|
require.NoError(t, err)
|
2018-11-07 00:03:02 -08:00
|
|
|
|
2019-02-01 17:03:09 -08:00
|
|
|
iavlStore := store.GetCommitStore(iavlStoreKey).(*iavl.Store)
|
2018-11-07 00:03:02 -08:00
|
|
|
iavlStore.Set([]byte("MYKEY"), []byte("MYVALUE"))
|
|
|
|
cid := store.Commit() // Commit with empty iavl store.
|
|
|
|
|
|
|
|
// Get Proof
|
|
|
|
res := store.Query(abci.RequestQuery{
|
|
|
|
Path: "/iavlStoreKey/key", // required path to get key/value+proof
|
|
|
|
Data: []byte("MYABSENTKEY"),
|
|
|
|
Prove: true,
|
|
|
|
})
|
2020-08-14 10:58:53 -07:00
|
|
|
require.NotNil(t, res.ProofOps)
|
2018-11-07 00:03:02 -08:00
|
|
|
|
|
|
|
// Verify proof.
|
|
|
|
prt := DefaultProofRuntime()
|
2020-08-14 10:58:53 -07:00
|
|
|
err = prt.VerifyAbsence(res.ProofOps, cid.Hash, "/iavlStoreKey/MYABSENTKEY")
|
2018-11-07 00:03:02 -08:00
|
|
|
require.Nil(t, err)
|
|
|
|
|
|
|
|
// Verify (bad) proof.
|
|
|
|
prt = DefaultProofRuntime()
|
2020-08-14 10:58:53 -07:00
|
|
|
err = prt.VerifyAbsence(res.ProofOps, cid.Hash, "/MYABSENTKEY")
|
2018-11-07 00:03:02 -08:00
|
|
|
require.NotNil(t, err)
|
|
|
|
|
|
|
|
// Verify (bad) proof.
|
|
|
|
prt = DefaultProofRuntime()
|
2020-08-14 10:58:53 -07:00
|
|
|
err = prt.VerifyValue(res.ProofOps, cid.Hash, "/iavlStoreKey/MYABSENTKEY", []byte(""))
|
2018-11-07 00:03:02 -08:00
|
|
|
require.NotNil(t, err)
|
|
|
|
}
|