From 4591a9001cbe62359ac2dbea5b380dd37b0cd4df Mon Sep 17 00:00:00 2001
From: Magamedrasul Ibragimov <curryrasul@gmail.com>
Date: Tue, 1 Nov 2022 20:31:18 +0300
Subject: [PATCH] fix: change initialization for cached_vec

---
 src/tree.rs            | 4 ++--
 tests/memory_keccak.rs | 8 +++++++-
 2 files changed, 9 insertions(+), 3 deletions(-)

diff --git a/src/tree.rs b/src/tree.rs
index a7ed713..751f054 100644
--- a/src/tree.rs
+++ b/src/tree.rs
@@ -50,7 +50,7 @@ where
         db.put(NEXT_INDEX_KEY, next_index_val);
 
         // Cache nodes
-        let mut cache = Vec::with_capacity(depth + 1);
+        let mut cache = vec![H::default_leaf(); depth + 1];
 
         // Initialize one branch of the `Merkle Tree` from bottom to top
         cache[depth] = H::default_leaf();
@@ -82,7 +82,7 @@ where
         let next_index = usize::from_be_bytes(next_index);
 
         // Load cache vec
-        let mut cache = Vec::with_capacity(depth + 1);
+        let mut cache = vec![H::default_leaf(); depth + 1];
         cache[depth] = H::default_leaf();
         for i in (0..depth).rev() {
             cache[i] = H::hash(&[cache[i + 1], cache[i + 1]]);
diff --git a/tests/memory_keccak.rs b/tests/memory_keccak.rs
index 6dbaedb..ba9ebf0 100644
--- a/tests/memory_keccak.rs
+++ b/tests/memory_keccak.rs
@@ -63,4 +63,10 @@ impl Hasher for MyKeccak {
 }
 
 #[test]
-fn insert_delete() {}
+fn insert_delete() {
+    let mt = MerkleTree::<MemoryDB, MyKeccak>::new(3, "abacaba");
+
+    assert_eq!(mt.capacity(), 8);
+    assert_eq!(mt.depth(), 3);
+    println!("{:?}", mt.root());
+}