Restore insertion of proposal

informalsystems · Dec 7, 2023 · d61222b · d61222b
1 parent 9993707
commit d61222b
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 1 deletion.
diff --git a/Code/driver/src/driver.rs b/Code/driver/src/driver.rs
@@ -176,7 +176,13 @@ where
     ) -> Result<Option<RoundOutput<Ctx>>, Error<Ctx>> {
         let round = proposal.round();
 
-        match mux::multiplex_proposal(&self.round_state, &self.votes, proposal, validity) {
+        match mux::multiplex_proposal(
+            &self.round_state,
+            &self.votes,
+            &mut self.proposals,
+            proposal,
+            validity,
+        ) {
             Some(round_input) => self.apply_input(round, round_input),
             None => Ok(None),
         }

diff --git a/Code/driver/src/mux.rs b/Code/driver/src/mux.rs
@@ -13,6 +13,7 @@ use crate::Validity;
 pub fn multiplex_proposal<Ctx>(
     round_state: &RoundState<Ctx>,
     votekeeper: &VoteKeeper<Ctx>,
+    proposals: &mut Proposals<Ctx>,
     proposal: Ctx::Proposal,
     validity: Validity,
 ) -> Option<RoundInput<Ctx>>
@@ -29,6 +30,9 @@ where
         return None;
     }
 
+    // Store the proposal
+    proposals.insert(proposal.clone());
+
     let polka_for_pol = votekeeper.is_threshold_met(
         &proposal.pol_round(),
         VoteType::Prevote,