chore(starknet_integration_tests): pass tx_generator as mut ref

commit-id:aeeae6b7
starkware-libs · Jan 15, 2025 · a00727a · a00727a
1 parent 53b06ea
commit a00727a
Show file tree

Hide file tree

Showing 4 changed files with 7 additions and 7 deletions.
diff --git a/crates/starknet_integration_tests/src/bin/sequencer_node_end_to_end_integration_test.rs b/crates/starknet_integration_tests/src/bin/sequencer_node_end_to_end_integration_test.rs
@@ -26,8 +26,8 @@ async fn main() {
     );
 
     // Creates a multi-account transaction generator for integration test
-    let tx_generator = create_integration_test_tx_generator();
+    let mut tx_generator = create_integration_test_tx_generator();
 
     // Run end to end integration test.
-    end_to_end_integration(tx_generator).await;
+    end_to_end_integration(&mut tx_generator).await;
 }
diff --git a/crates/starknet_integration_tests/src/end_to_end_integration.rs b/crates/starknet_integration_tests/src/end_to_end_integration.rs
@@ -5,7 +5,7 @@ use tracing::info;
 
 use crate::sequencer_manager::{get_sequencer_setup_configs, SequencerSetupManager};
 
-pub async fn end_to_end_integration(tx_generator: MultiAccountTransactionGenerator) {
+pub async fn end_to_end_integration(tx_generator: &mut MultiAccountTransactionGenerator) {
     const EXPECTED_BLOCK_NUMBER: BlockNumber = BlockNumber(15);
     const N_TXS: usize = 50;
     const SENDER_ACCOUNT: AccountId = 0;
@@ -15,7 +15,7 @@ pub async fn end_to_end_integration(tx_generator: MultiAccountTransactionGenerat
     get_node_executable_path();
 
     // Get the sequencer configurations.
-    let sequencers_setup = get_sequencer_setup_configs(&tx_generator).await;
+    let sequencers_setup = get_sequencer_setup_configs(tx_generator).await;
 
     // Run the sequencers.
     // TODO(Nadin, Tsabary): Refactor to separate the construction of SequencerManager from its

diff --git a/crates/starknet_integration_tests/src/sequencer_manager.rs b/crates/starknet_integration_tests/src/sequencer_manager.rs
@@ -121,7 +121,7 @@ impl SequencerSetupManager {
 
     pub async fn run_integration_test_simulator(
         &self,
-        tx_generator: MultiAccountTransactionGenerator,
+        tx_generator: &mut MultiAccountTransactionGenerator,
         n_txs: usize,
         sender_account: AccountId,
     ) {

diff --git a/crates/starknet_integration_tests/src/utils.rs b/crates/starknet_integration_tests/src/utils.rs
@@ -222,7 +222,7 @@ fn fund_new_account(
 }
 
 fn create_account_txs(
-    mut tx_generator: MultiAccountTransactionGenerator,
+    tx_generator: &mut MultiAccountTransactionGenerator,
     account_id: AccountId,
     n_txs: usize,
 ) -> Vec<RpcTransaction> {
@@ -278,7 +278,7 @@ pub fn test_tx_hashes_for_integration_test(tx_hashes: &[TransactionHash]) -> Vec
 
 /// Returns a list of the transaction hashes, in the order they are expected to be in the mempool.
 pub async fn send_account_txs<'a, Fut>(
-    tx_generator: MultiAccountTransactionGenerator,
+    tx_generator: &mut MultiAccountTransactionGenerator,
     account_id: AccountId,
     n_txs: usize,
     send_rpc_tx_fn: &'a mut dyn FnMut(RpcTransaction) -> Fut,