2019-12-04 12:48:38 +00:00
|
|
|
package trie
|
|
|
|
|
|
|
|
import (
|
|
|
|
"bytes"
|
|
|
|
"testing"
|
|
|
|
|
|
|
|
"github.com/ledgerwatch/turbo-geth/common"
|
|
|
|
"github.com/ledgerwatch/turbo-geth/core/types/accounts"
|
|
|
|
)
|
|
|
|
|
|
|
|
func TestBlockWitnessBinary(t *testing.T) {
|
|
|
|
tr := New(common.Hash{})
|
2020-02-06 10:53:09 +00:00
|
|
|
tr.Update([]byte("ABCD0001"), []byte("val1"))
|
|
|
|
tr.Update([]byte("ABCE0002"), []byte("val2"))
|
2019-12-04 12:48:38 +00:00
|
|
|
|
|
|
|
trBin := HexToBin(tr)
|
|
|
|
|
|
|
|
rs := NewBinaryResolveSet(2)
|
|
|
|
rs.AddKey([]byte("ABCD0001"))
|
2020-01-15 13:56:50 +00:00
|
|
|
|
2020-05-12 07:22:45 +00:00
|
|
|
bwb := NewWitnessBuilder(trBin.Trie().root, false)
|
2020-01-15 13:56:50 +00:00
|
|
|
|
|
|
|
hr := newHasher(false)
|
|
|
|
defer returnHasherToPool(hr)
|
|
|
|
|
|
|
|
var w *Witness
|
|
|
|
var err error
|
2020-01-24 10:58:01 +00:00
|
|
|
if w, err = bwb.Build(&MerklePathLimiter{rs, hr.hash}); err != nil {
|
2019-12-04 12:48:38 +00:00
|
|
|
t.Errorf("Could not make block witness: %v", err)
|
|
|
|
}
|
|
|
|
|
2020-03-23 22:10:36 +00:00
|
|
|
trBin1, err := BuildTrieFromWitness(w, true /*is-binary*/, false /*trace*/)
|
2019-12-04 12:48:38 +00:00
|
|
|
if err != nil {
|
|
|
|
t.Errorf("Could not restore trie from the block witness: %v", err)
|
|
|
|
}
|
|
|
|
if trBin.Trie().Hash() != trBin1.Hash() {
|
|
|
|
t.Errorf("Reconstructed block witness has different root hash than source trie")
|
|
|
|
}
|
|
|
|
|
|
|
|
expected := []byte("val1")
|
|
|
|
got, _ := trBin1.Get([]byte("ABCD0001"))
|
|
|
|
if !bytes.Equal(got, expected) {
|
|
|
|
t.Errorf("unexpected value: %x (expected %x)", got, expected)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
func TestBlockWitnessBinaryAccount(t *testing.T) {
|
|
|
|
tr := New(common.Hash{})
|
|
|
|
|
|
|
|
account := accounts.NewAccount()
|
|
|
|
account.Balance.SetInt64(1 * 1000 * 1000)
|
|
|
|
|
|
|
|
tr.UpdateAccount([]byte("ABCD0001"), &account)
|
|
|
|
|
|
|
|
trBin := HexToBin(tr)
|
|
|
|
|
|
|
|
rs := NewBinaryResolveSet(2)
|
|
|
|
rs.AddKey([]byte("ABCD0001"))
|
2020-01-15 13:56:50 +00:00
|
|
|
|
2020-05-12 07:22:45 +00:00
|
|
|
bwb := NewWitnessBuilder(trBin.Trie().root, false)
|
2020-01-15 13:56:50 +00:00
|
|
|
|
|
|
|
hr := newHasher(false)
|
|
|
|
defer returnHasherToPool(hr)
|
|
|
|
|
|
|
|
var w *Witness
|
|
|
|
var err error
|
2020-01-24 10:58:01 +00:00
|
|
|
if w, err = bwb.Build(&MerklePathLimiter{rs, hr.hash}); err != nil {
|
2019-12-04 12:48:38 +00:00
|
|
|
t.Errorf("Could not make block witness: %v", err)
|
|
|
|
}
|
|
|
|
|
2020-03-23 22:10:36 +00:00
|
|
|
trBin1, err := BuildTrieFromWitness(w, true /*is-binary*/, false /*trace*/)
|
2019-12-04 12:48:38 +00:00
|
|
|
if err != nil {
|
|
|
|
t.Errorf("Could not restore trie from the block witness: %v", err)
|
|
|
|
}
|
|
|
|
if trBin.Trie().Hash() != trBin1.Hash() {
|
|
|
|
t.Errorf("Reconstructed block witness has different root hash than source trie")
|
|
|
|
}
|
|
|
|
|
|
|
|
got, _ := trBin1.GetAccount([]byte("ABCD0001"))
|
|
|
|
if !account.Equals(got) {
|
|
|
|
t.Errorf("received account is not equal to the initial one")
|
|
|
|
}
|
|
|
|
}
|