064c9ba638
Co-authored-by: marbar3778 <marbar3778@yahoo.com> Co-authored-by: Marko <marko@baricevic.me> Co-authored-by: Alex | Interchain Labs <alex@skip.money>
72 lines
1.9 KiB
Go
72 lines
1.9 KiB
Go
package proof
|
|
|
|
import (
|
|
"fmt"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestGetStoreProof(t *testing.T) {
|
|
tests := []struct {
|
|
storeInfos []*StoreInfo
|
|
}{
|
|
{[]*StoreInfo{
|
|
{("key1"), &CommitID{1, []byte("value1")}, "iavl"},
|
|
}},
|
|
{[]*StoreInfo{
|
|
{("key2"), &CommitID{1, []byte("value2")}, "iavl"},
|
|
{("key1"), &CommitID{1, []byte("value1")}, "iavl"},
|
|
}},
|
|
{[]*StoreInfo{
|
|
{("key3"), &CommitID{1, []byte("value3")}, "iavl"},
|
|
{("key2"), &CommitID{1, []byte("value2")}, "iavl"},
|
|
{("key1"), &CommitID{1, []byte("value1")}, "iavl"},
|
|
}},
|
|
{[]*StoreInfo{
|
|
{("key2"), &CommitID{1, []byte("value2")}, "iavl"},
|
|
{("key1"), &CommitID{1, []byte("value1")}, "iavl"},
|
|
{("key3"), &CommitID{1, []byte("value3")}, "iavl"},
|
|
}},
|
|
{[]*StoreInfo{
|
|
{("key4"), &CommitID{1, []byte("value4")}, "iavl"},
|
|
{("key1"), &CommitID{1, []byte("value1")}, "iavl"},
|
|
{("key3"), &CommitID{1, []byte("value3")}, "iavl"},
|
|
{("key2"), &CommitID{1, []byte("value2")}, "iavl"},
|
|
}},
|
|
}
|
|
|
|
for i, tc := range tests {
|
|
t.Run(fmt.Sprintf("%d", i), func(t *testing.T) {
|
|
// create a commit info
|
|
ci := &CommitInfo{
|
|
Version: 1,
|
|
Timestamp: time.Now(),
|
|
StoreInfos: tc.storeInfos,
|
|
}
|
|
commitHash := ci.Hash()
|
|
// make sure the store infos are sorted
|
|
require.Equal(t, ci.StoreInfos[0].Name, "key1")
|
|
for _, si := range tc.storeInfos {
|
|
// get the proof
|
|
_, proof, err := ci.GetStoreProof([]byte(si.Name))
|
|
require.NoError(t, err, "test case %d", i)
|
|
// verify the proof
|
|
expRoots, err := proof.Run([][]byte{si.CommitId.Hash})
|
|
require.NoError(t, err, "test case %d", i)
|
|
require.Equal(t, commitHash, expRoots[0], "test case %d", i)
|
|
|
|
bz, err := ci.Marshal()
|
|
require.NoError(t, err)
|
|
ci2 := &CommitInfo{}
|
|
err = ci2.Unmarshal(bz)
|
|
require.NoError(t, err)
|
|
require.True(t, ci.Timestamp.Equal(ci2.Timestamp))
|
|
ci2.Timestamp = ci.Timestamp
|
|
require.Equal(t, ci, ci2)
|
|
}
|
|
})
|
|
}
|
|
}
|