eth-iterator-utils/iterator_test.go

101 lines
2.6 KiB
Go
Raw Normal View History

2022-01-13 01:57:23 +00:00
package iterator_test
import (
2022-02-22 08:13:29 +00:00
"bytes"
"fmt"
2022-01-13 01:57:23 +00:00
"testing"
2022-02-22 08:13:29 +00:00
"github.com/ethereum/go-ethereum/core/rawdb"
"github.com/ethereum/go-ethereum/core/state"
2022-01-13 01:57:23 +00:00
iter "github.com/vulcanize/go-eth-state-node-iterator"
2022-02-22 08:13:29 +00:00
fixt "github.com/vulcanize/go-eth-state-node-iterator/fixture"
2022-01-13 01:57:23 +00:00
)
func TestMakePaths(t *testing.T) {
var prefix []byte
for i := 0; i < 4; i++ {
nbins := uint(1) << i
paths := iter.MakePaths(prefix, nbins)
if len(paths) != int(nbins) {
2022-02-24 14:56:27 +00:00
t.Errorf("wrong number of paths; expected %d, have %d", nbins, len(paths))
2022-01-13 01:57:23 +00:00
}
}
}
2022-02-22 08:13:29 +00:00
func TestIterator(t *testing.T) {
edb, err := rawdb.NewLevelDBDatabaseWithFreezer(
fixt.ChainDataPath, 1024, 256, fixt.AncientDataPath,
"eth-pg-ipfs-state-snapshot", false,
)
if err != nil {
t.Fatal(err)
}
defer edb.Close()
height := uint64(1)
hash := rawdb.ReadCanonicalHash(edb, height)
header := rawdb.ReadHeader(edb, hash, height)
if header == nil {
t.Fatalf("unable to read canonical header at height %d", height)
}
sdb := state.NewDatabase(edb)
tree, err := sdb.OpenTrie(header.Root)
if err != nil {
t.Fatal(err)
}
t.Run("in bounds", func(t *testing.T) {
type testCase struct {
lower, upper []byte
}
cases := []testCase{
2022-05-31 15:08:23 +00:00
{nil, []byte{0, 0}},
{[]byte{1, 0}, []byte{2, 0}},
{[]byte{3, 5}, []byte{4, 2, 0}},
{[]byte{5, 6, 9, 0}, []byte{7, 0}},
{[]byte{8, 0}, []byte{8, 0}},
{[]byte{8, 0}, []byte{7, 0}},
2022-02-22 08:13:29 +00:00
}
runCase := func(t *testing.T, tc testCase) {
2022-05-31 15:08:23 +00:00
it := iter.NewPrefixBoundIterator(tree.NodeIterator(iter.HexToKeyBytes(tc.lower)), tc.upper)
2022-02-22 08:13:29 +00:00
for it.Next(true) {
if bytes.Compare(it.Path(), tc.lower) < 0 {
t.Fatalf("iterator outside lower bound: %v", it.Path())
}
2022-05-31 15:08:23 +00:00
if bytes.Compare(tc.upper, it.Path()) < 0 {
t.Fatalf("iterator outside upper bound: %v <= %v", tc.upper, it.Path())
2022-02-22 08:13:29 +00:00
}
}
}
for _, tc := range cases {
t.Run("case", func(t *testing.T) { runCase(t, tc) })
}
})
t.Run("trie is covered", func(t *testing.T) {
allPaths := fixt.Block1_Paths
cases := []uint{1, 2, 4, 8, 16, 32}
runCase := func(t *testing.T, nbins uint) {
iters := iter.SubtrieIterators(tree, nbins)
ix := 0
for b := uint(0); b < nbins; b++ {
2022-02-24 14:56:27 +00:00
for it := iters[b]; it.Next(true); ix++ {
2022-02-22 08:13:29 +00:00
if !bytes.Equal(allPaths[ix], it.Path()) {
t.Fatalf("wrong path value\nexpected:\t%v\nactual:\t\t%v",
allPaths[ix], it.Path())
}
}
2022-05-31 15:08:23 +00:00
// if the last node path was even-length, it will be duplicated
if len(allPaths[ix-1])&0b1 == 0 {
ix--
}
2022-02-22 08:13:29 +00:00
}
}
for _, tc := range cases {
t.Run(fmt.Sprintf("%d bins", tc), func(t *testing.T) { runCase(t, tc) })
}
})
}