From e17e5477d97059d26b3c79d3467c41c6883421f9 Mon Sep 17 00:00:00 2001
From: "Conrado P. L. Gouvea" <conradoplg@gmail.com>
Date: Tue, 20 Jul 2021 19:47:22 -0300
Subject: [PATCH] Pass empty trees to tests; use empty trees by default in
 Chain

---
 zebra-state/src/service/non_finalized_state/chain.rs  | 11 +++++++++++
 .../src/service/non_finalized_state/tests/prop.rs     |  2 +-
 2 files changed, 12 insertions(+), 1 deletion(-)

diff --git a/zebra-state/src/service/non_finalized_state/chain.rs b/zebra-state/src/service/non_finalized_state/chain.rs
index b6ab4d4f4ad..b7d5f437c2b 100644
--- a/zebra-state/src/service/non_finalized_state/chain.rs
+++ b/zebra-state/src/service/non_finalized_state/chain.rs
@@ -758,3 +758,14 @@ impl PartialEq for Chain {
 }
 
 impl Eq for Chain {}
+
+impl Default for Chain {
+    fn default() -> Self {
+        Self::new(
+            // TODO: decide if we want to default to None or empty tree
+            Some(Default::default()),
+            Some(Default::default()),
+            Some(Default::default()),
+        )
+    }
+}
diff --git a/zebra-state/src/service/non_finalized_state/tests/prop.rs b/zebra-state/src/service/non_finalized_state/tests/prop.rs
index e1aad540606..f6860f1df96 100644
--- a/zebra-state/src/service/non_finalized_state/tests/prop.rs
+++ b/zebra-state/src/service/non_finalized_state/tests/prop.rs
@@ -39,7 +39,7 @@ fn forked_equals_pushed() -> Result<()> {
             }
 
             // TODO: pass correct trees
-            let forked = full_chain.fork(fork_tip_hash, None, None, None).expect("fork works").expect("hash is present");
+            let forked = full_chain.fork(fork_tip_hash, Some(Default::default()), Some(Default::default()), Some(Default::default())).expect("fork works").expect("hash is present");
 
             // the first check is redundant, but it's useful for debugging
             prop_assert_eq!(forked.blocks.len(), partial_chain.blocks.len());