feat(starknet_integration_tests): add snapshot tx generator fn

crates/mempool_test_utils/src/starknet_api_test_utils.rs

@@ -209,6 +209,21 @@ impl MultiAccountTransactionGenerator {
         Self::default()
     }
 
+    pub fn snapshot(&self) -> Self {
+        let nonce_manager = Rc::new(RefCell::new((*self.nonce_manager.borrow()).clone()));
+        let account_tx_generators = self
+            .account_tx_generators
+            .iter()
+            .map(|tx_gen| AccountTransactionGenerator {
+                account: tx_gen.account,
+                nonce_manager: nonce_manager.clone(),
+                contract_address_salt: tx_gen.contract_address_salt,
+            })
+            .collect();
+
+        Self { account_tx_generators, nonce_manager }
+    }
+
     /// Registers a new account with the given contract, assuming it is already deployed.
     /// Note: the state should reflect that the account is already deployed.
     pub fn register_deployed_account(&mut self, account_contract: FeatureContract) -> AccountId {

crates/starknet_api/src/test_utils.rs

@@ -72,7 +72,7 @@ pub struct TransactionTestData {
     pub block_number: BlockNumber,
 }
 
-#[derive(Debug, Default)]
+#[derive(Debug, Default, Clone)]
 pub struct NonceManager {
     next_nonce: HashMap<ContractAddress, Felt>,
 }

crates/starknet_integration_tests/src/sequencer_manager.rs

@@ -195,6 +195,14 @@ impl IntegrationTestManager {
         Self { node_indices, idle_nodes, running_nodes, tx_generator }
     }
 
+    pub fn tx_generator(&self) -> &MultiAccountTransactionGenerator {
+        &self.tx_generator
+    }
+
+    pub fn tx_generator_mut(&mut self) -> &mut MultiAccountTransactionGenerator {
+        &mut self.tx_generator
+    }
+
     pub async fn run_nodes(&mut self, nodes_to_run: HashSet<usize>) {
         info!("Running specified nodes.");