Skip to content

Commit

Permalink
trie: reduce unit test time
Browse files Browse the repository at this point in the history
  • Loading branch information
MariusVanDerWijden committed Mar 17, 2023
1 parent d40ba33 commit 6168ec3
Show file tree
Hide file tree
Showing 2 changed files with 21 additions and 13 deletions.
25 changes: 19 additions & 6 deletions trie/proof_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -20,6 +20,7 @@ import (
"bytes"
crand "crypto/rand"
"encoding/binary"
"fmt"
mrand "math/rand"
"sort"
"testing"
Expand All @@ -30,6 +31,24 @@ import (
"github.com/ethereum/go-ethereum/ethdb/memorydb"
)

// Prng is a pseudo random number generator seeded by strong randomness.
// The randomness is printed on startup in order to make failures reproducible.
var prng = initRnd()

func initRnd() *mrand.Rand {
var seed [8]byte
crand.Read(seed[:])
rnd := mrand.New(mrand.NewSource(int64(binary.LittleEndian.Uint64(seed[:]))))
fmt.Printf("Seed: %x\n", seed)
return rnd
}

func randBytes(n int) []byte {
r := make([]byte, n)
prng.Read(r)
return r
}

// makeProvers creates Merkle trie provers based on different implementations to
// test all variations.
func makeProvers(trie *Trie) []func(key []byte) *memorydb.Database {
Expand Down Expand Up @@ -1041,12 +1060,6 @@ func randomTrie(n int) (*Trie, map[string]*kv) {
return trie, vals
}

func randBytes(n int) []byte {
r := make([]byte, n)
crand.Read(r)
return r
}

func nonRandomTrie(n int) (*Trie, map[string]*kv) {
trie := NewEmpty(NewDatabase(rawdb.NewMemoryDatabase()))
vals := make(map[string]*kv)
Expand Down
9 changes: 2 additions & 7 deletions trie/trie_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -18,7 +18,6 @@ package trie

import (
"bytes"
crand "crypto/rand"
"encoding/binary"
"errors"
"fmt"
Expand Down Expand Up @@ -1147,17 +1146,13 @@ func deleteString(trie *Trie, k string) {
func TestDecodeNode(t *testing.T) {
t.Parallel()

var seed [8]byte
crand.Read(seed[:])
rnd := rand.New(rand.NewSource(int64(binary.LittleEndian.Uint64(seed[:]))))
fmt.Printf("Seed: %v\n", seed)
var (
hash = make([]byte, 20)
elems = make([]byte, 20)
)
for i := 0; i < 5000000; i++ {
rnd.Read(hash)
rnd.Read(elems)
prng.Read(hash)
prng.Read(elems)
decodeNode(hash, elems)
}
}

0 comments on commit 6168ec3

Please sign in to comment.