apollo_consensus_orchestrator: add helper func to test sending proposal to validate

Author: guy-starkware

crates/apollo_consensus_orchestrator/src/sequencer_consensus_context.rs

@@ -440,6 +440,10 @@ impl SequencerConsensusContext {
             error!("Failed to prepare blob for next height at height {height}: {e:?}");
         }
     }
+
+    pub fn get_config(&self) -> &ContextConfig {
+        &self.config
+    }
 }
 
 #[async_trait]

crates/apollo_consensus_orchestrator/src/sequencer_consensus_context_test.rs

@@ -46,6 +46,7 @@ use crate::orchestrator_versioned_constants::VersionedConstants;
 use crate::test_utils::{
     block_info,
     create_test_and_network_deps,
+    send_proposal_to_validator_context,
     SetupDepsArgs,
     ETH_TO_FRI_RATE,
     INTERNAL_TX_BATCH,
@@ -81,26 +82,10 @@ async fn validate_proposal_success() {
     // Initialize the context for a specific height, starting with round 0.
     context.set_height_and_round(BlockNumber(0), 0).await.unwrap();
 
-    let (mut content_sender, content_receiver) =
-        mpsc::channel(context.config.static_config.proposal_buffer_size);
-    content_sender.send(ProposalPart::BlockInfo(block_info(BlockNumber(0)))).await.unwrap();
-    content_sender
-        .send(ProposalPart::Transactions(TransactionBatch { transactions: TX_BATCH.to_vec() }))
-        .await
-        .unwrap();
-    content_sender
-        .send(ProposalPart::ExecutedTransactionCount(INTERNAL_TX_BATCH.len().try_into().unwrap()))
-        .await
-        .unwrap();
-    content_sender
-        .send(ProposalPart::Fin(ProposalFin {
-            proposal_commitment: ProposalCommitment(STATE_DIFF_COMMITMENT.0.0),
-        }))
-        .await
-        .unwrap();
+    let content_receiver =
+        send_proposal_to_validator_context(&mut context, block_info(BlockNumber(0))).await;
     let fin_receiver =
         context.validate_proposal(ProposalInit::default(), TIMEOUT, content_receiver).await;
-    content_sender.close_channel();
     assert_eq!(fin_receiver.await.unwrap().0, STATE_DIFF_COMMITMENT.0.0);
 }
 
@@ -258,23 +243,8 @@ async fn interrupt_active_proposal() {
     let fin_receiver_0 =
         context.validate_proposal(ProposalInit::default(), TIMEOUT, content_receiver).await;
 
-    let (mut content_sender_1, content_receiver) =
-        mpsc::channel(context.config.static_config.proposal_buffer_size);
-    content_sender_1.send(ProposalPart::BlockInfo(block_info(BlockNumber(0)))).await.unwrap();
-    content_sender_1
-        .send(ProposalPart::Transactions(TransactionBatch { transactions: TX_BATCH.to_vec() }))
-        .await
-        .unwrap();
-    content_sender_1
-        .send(ProposalPart::ExecutedTransactionCount(INTERNAL_TX_BATCH.len().try_into().unwrap()))
-        .await
-        .unwrap();
-    content_sender_1
-        .send(ProposalPart::Fin(ProposalFin {
-            proposal_commitment: ProposalCommitment(STATE_DIFF_COMMITMENT.0.0),
-        }))
-        .await
-        .unwrap();
+    let content_receiver =
+        send_proposal_to_validator_context(&mut context, block_info(BlockNumber(0))).await;
     let fin_receiver_1 = context
         .validate_proposal(
             ProposalInit { round: 1, ..Default::default() },
@@ -605,8 +575,6 @@ async fn gas_price_limits(#[case] maximum: bool) {
     let mut context = deps.build_context();
 
     context.set_height_and_round(BlockNumber(0), 0).await.unwrap();
-    let (mut content_sender, content_receiver) =
-        mpsc::channel(context.config.static_config.proposal_buffer_size);
 
     let mut block_info = block_info(BlockNumber(0));
 
@@ -621,21 +589,7 @@ async fn gas_price_limits(#[case] maximum: bool) {
     }
 
     // Send the block info, some transactions and then fin.
-    content_sender.send(ProposalPart::BlockInfo(block_info).clone()).await.unwrap();
-    content_sender
-        .send(ProposalPart::Transactions(TransactionBatch { transactions: TX_BATCH.to_vec() }))
-        .await
-        .unwrap();
-    content_sender
-        .send(ProposalPart::ExecutedTransactionCount(INTERNAL_TX_BATCH.len().try_into().unwrap()))
-        .await
-        .unwrap();
-    content_sender
-        .send(ProposalPart::Fin(ProposalFin {
-            proposal_commitment: ProposalCommitment(STATE_DIFF_COMMITMENT.0.0),
-        }))
-        .await
-        .unwrap();
+    let content_receiver = send_proposal_to_validator_context(&mut context, block_info).await;
 
     // Even though we used the minimum/maximum gas price, not the values we gave the provider,
     // the proposal should be still be valid due to the clamping of limit prices.
@@ -845,26 +799,10 @@ async fn oracle_fails_on_second_block(#[case] l1_oracle_failure: bool) {
     // Initialize the context for a specific height, starting with round 0.
     context.set_height_and_round(BlockNumber(0), 0).await.unwrap();
 
-    let (mut content_sender, content_receiver) =
-        mpsc::channel(context.config.static_config.proposal_buffer_size);
-    content_sender.send(ProposalPart::BlockInfo(block_info(BlockNumber(0)))).await.unwrap();
-    content_sender
-        .send(ProposalPart::Transactions(TransactionBatch { transactions: TX_BATCH.to_vec() }))
-        .await
-        .unwrap();
-    content_sender
-        .send(ProposalPart::ExecutedTransactionCount(INTERNAL_TX_BATCH.len().try_into().unwrap()))
-        .await
-        .unwrap();
-    content_sender
-        .send(ProposalPart::Fin(ProposalFin {
-            proposal_commitment: ProposalCommitment(STATE_DIFF_COMMITMENT.0.0),
-        }))
-        .await
-        .unwrap();
+    let content_receiver =
+        send_proposal_to_validator_context(&mut context, block_info(BlockNumber(0))).await;
     let fin_receiver =
         context.validate_proposal(ProposalInit::default(), TIMEOUT, content_receiver).await;
-    content_sender.close_channel();
     let proposal_commitment = fin_receiver.await.unwrap();
     assert_eq!(proposal_commitment.0, STATE_DIFF_COMMITMENT.0.0);
 

crates/apollo_consensus_orchestrator/src/test_utils.rs

@@ -28,11 +28,20 @@ use apollo_network::network_manager::test_utils::{
     TestSubscriberChannels,
 };
 use apollo_network::network_manager::{BroadcastTopicChannels, BroadcastTopicClient};
-use apollo_protobuf::consensus::{ConsensusBlockInfo, HeightAndRound, ProposalPart, Vote};
+use apollo_protobuf::consensus::{
+    ConsensusBlockInfo,
+    HeightAndRound,
+    ProposalCommitment as ProtoProposalCommitment,
+    ProposalFin,
+    ProposalPart,
+    TransactionBatch,
+    Vote,
+};
 use apollo_state_sync_types::communication::MockStateSyncClient;
 use apollo_time::time::{Clock, DefaultClock};
 use futures::channel::mpsc;
 use futures::executor::block_on;
+use futures::SinkExt;
 use mockall::Sequence;
 use starknet_api::block::{
     BlockNumber,
@@ -350,6 +359,34 @@ pub(crate) fn block_info(height: BlockNumber) -> ConsensusBlockInfo {
         eth_to_fri_rate: ETH_TO_FRI_RATE,
     }
 }
+
+// Send the given block info, some txs, the number of txs, and the fin, returning the
+// content_receiver.
+pub(crate) async fn send_proposal_to_validator_context(
+    context: &mut SequencerConsensusContext,
+    block_info: ConsensusBlockInfo,
+) -> mpsc::Receiver<ProposalPart> {
+    let (mut content_sender, content_receiver) =
+        mpsc::channel(context.get_config().static_config.proposal_buffer_size);
+    content_sender.send(ProposalPart::BlockInfo(block_info)).await.unwrap();
+    content_sender
+        .send(ProposalPart::Transactions(TransactionBatch { transactions: TX_BATCH.to_vec() }))
+        .await
+        .unwrap();
+    content_sender
+        .send(ProposalPart::ExecutedTransactionCount(INTERNAL_TX_BATCH.len().try_into().unwrap()))
+        .await
+        .unwrap();
+    content_sender
+        .send(ProposalPart::Fin(ProposalFin {
+            proposal_commitment: ProtoProposalCommitment(STATE_DIFF_COMMITMENT.0.0),
+        }))
+        .await
+        .unwrap();
+    content_sender.close_channel();
+    content_receiver
+}
+
 // Structs which aren't utilized but should not be dropped.
 pub(crate) struct NetworkDependencies {
     _vote_network: BroadcastNetworkMock<Vote>,