Add a condition (#3749)

EspressoSystems · Oct 10, 2024 · b4ecb8f · b4ecb8f
1 parent 5a9064b
commit b4ecb8f
Showing 1 changed file with 14 additions and 7 deletions.
diff --git a/crates/types/src/consensus.rs b/crates/types/src/consensus.rs
@@ -563,15 +563,22 @@ impl<TYPES: NodeType> Consensus<TYPES> {
     pub fn update_validated_state_map(
         &mut self,
         view_number: TYPES::Time,
-        view: View<TYPES>,
+        new_view: View<TYPES>,
     ) -> Result<()> {
         if let Some(existing_view) = self.validated_state_map().get(&view_number) {
-            if let ViewInner::Leaf { .. } = existing_view.view_inner {
-                match view.view_inner {
-                    ViewInner::Leaf { ref delta, .. } => {
+            if let ViewInner::Leaf {
+                delta: ref existing_delta,
+                ..
+            } = existing_view.view_inner
+            {
+                match new_view.view_inner {
+                    ViewInner::Leaf {
+                        delta: ref new_delta,
+                        ..
+                    } => {
                         ensure!(
-                            delta.is_some(),
-                            "Skipping the state update to not override a `Leaf` view with `None` state delta."
+                            new_delta.is_some() || existing_delta.is_none(),
+                            "Skipping the state update to not override a `Leaf` view with `Some` state delta."
                         );
                     }
                     _ => {
@@ -580,7 +587,7 @@ impl<TYPES: NodeType> Consensus<TYPES> {
                 }
             }
         }
-        self.validated_state_map.insert(view_number, view);
+        self.validated_state_map.insert(view_number, new_view);
         Ok(())
     }