Compare the subtrees in Chain (#7562)

ZcashFoundation · Sep 18, 2023 · dea9a8c · dea9a8c
1 parent a7501c1
commit dea9a8c
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/zebra-state/src/service/non_finalized_state/chain.rs b/zebra-state/src/service/non_finalized_state/chain.rs
@@ -281,6 +281,10 @@ impl Chain {
             // history trees
             self.history_trees_by_height == other.history_trees_by_height &&
 
+            // note commitment subtrees
+            self.sapling_subtrees == other.sapling_subtrees &&
+            self.orchard_subtrees == other.orchard_subtrees &&
+
             // anchors
             self.sprout_anchors == other.sprout_anchors &&
             self.sprout_anchors_by_height == other.sprout_anchors_by_height &&

diff --git a/zebra-state/src/service/non_finalized_state/tests/prop.rs b/zebra-state/src/service/non_finalized_state/tests/prop.rs
@@ -624,6 +624,10 @@ fn different_blocks_different_chains() -> Result<()> {
                 chain1.sapling_trees_by_height = chain2.sapling_trees_by_height.clone();
                 chain1.orchard_trees_by_height = chain2.orchard_trees_by_height.clone();
 
+                // note commitment subtrees
+                chain1.sapling_subtrees = chain2.sapling_subtrees.clone();
+                chain1.orchard_subtrees = chain2.orchard_subtrees.clone();
+
                 // history trees
                 chain1.history_trees_by_height = chain2.history_trees_by_height.clone();