diff options
Diffstat (limited to 'trie')
| -rw-r--r-- | trie/proof_test.go | 10 | ||||
| -rw-r--r-- | trie/trie_test.go | 2 |
2 files changed, 6 insertions, 6 deletions
diff --git a/trie/proof_test.go b/trie/proof_test.go index bcb241bd7..c488f342c 100644 --- a/trie/proof_test.go +++ b/trie/proof_test.go @@ -34,17 +34,17 @@ func init() { // makeProvers creates Merkle trie provers based on different implementations to // test all variations. -func makeProvers(trie *Trie) []func(key []byte) *memorydb.MemoryDatabase { - var provers []func(key []byte) *memorydb.MemoryDatabase +func makeProvers(trie *Trie) []func(key []byte) *memorydb.Database { + var provers []func(key []byte) *memorydb.Database // Create a direct trie based Merkle prover - provers = append(provers, func(key []byte) *memorydb.MemoryDatabase { + provers = append(provers, func(key []byte) *memorydb.Database { proof := memorydb.New() trie.Prove(key, 0, proof) return proof }) // Create a leaf iterator based Merkle prover - provers = append(provers, func(key []byte) *memorydb.MemoryDatabase { + provers = append(provers, func(key []byte) *memorydb.Database { proof := memorydb.New() if it := NewIterator(trie.NodeIterator(key)); it.Next() && bytes.Equal(key, it.Key) { for _, p := range it.Prove() { @@ -180,7 +180,7 @@ func BenchmarkVerifyProof(b *testing.B) { trie, vals := randomTrie(100) root := trie.Hash() var keys []string - var proofs []*memorydb.MemoryDatabase + var proofs []*memorydb.Database for k := range vals { keys = append(keys, k) proof := memorydb.New() diff --git a/trie/trie_test.go b/trie/trie_test.go index cf133706f..1c874370c 100644 --- a/trie/trie_test.go +++ b/trie/trie_test.go @@ -542,7 +542,7 @@ func benchGet(b *testing.B, commit bool) { b.StopTimer() if commit { - ldb := trie.db.diskdb.(*leveldb.LevelDBDatabase) + ldb := trie.db.diskdb.(*leveldb.Database) ldb.Close() os.RemoveAll(ldb.Path()) } |