test: make regtest run in parallel

Author: fdefelici

stacks-common/src/util/mod.rs

@@ -77,6 +77,15 @@ pub fn get_epoch_time_ms() -> u128 {
     since_the_epoch.as_millis()
 }
 
+#[cfg(any(test, feature = "testing"))]
+pub fn get_epoch_time_nanos() -> u128 {
+    let start = SystemTime::now();
+    let since_the_epoch = start
+        .duration_since(UNIX_EPOCH)
+        .expect("Time went backwards");
+    since_the_epoch.as_nanos()
+}
+
 pub fn sleep_ms(millis: u64) {
     let t = time::Duration::from_millis(millis);
     thread::sleep(t);

testnet/stacks-node/src/burnchains/bitcoin_regtest_controller.rs

@@ -2850,7 +2850,11 @@ mod tests {
     use crate::Keychain;
 
     mod utils {
+        use std::net::TcpListener;
+
         use super::*;
+        use crate::tests::bitcoin_regtest::BURNCHAIN_CONFIG_PEER_PORT_DISABLED;
+        use crate::util::get_epoch_time_nanos;
 
         pub fn create_config() -> Config {
             let mut config = Config::default();
@@ -2859,6 +2863,22 @@ mod tests {
             config.burnchain.password = Some(String::from("12345"));
             // overriding default "0.0.0.0" because doesn't play nicely on Windows.
             config.burnchain.peer_host = String::from("127.0.0.1");
+            // avoiding peer port biding to reduce the number of ports to bind to.
+            config.burnchain.peer_port = BURNCHAIN_CONFIG_PEER_PORT_DISABLED;
+
+            //Ask the OS for a free port. Not guaranteed to stay free, 
+            //after TcpListner is dropped, but good enough for testing
+            //and starting bitcoind right after config is created
+            let tmp_listener =
+                TcpListener::bind("127.0.0.1:0").expect("Failed to bind to get a free port");
+            let port = tmp_listener.local_addr().unwrap().port();
+
+            config.burnchain.rpc_port = port;
+
+            let now = get_epoch_time_nanos();
+            let dir = format!("/tmp/regtest-ctrl-{port}-{now}");
+            config.node.working_dir = dir;
+
             config
         }
 
@@ -2896,7 +2916,7 @@ mod tests {
                 key_vtxindex: 1,        // 0x0001
                 memo: vec![11],         // 0x5a >> 3
 
-                burn_fee: 110_000,      //relevant for fee calculation when sending the tx
+                burn_fee: 110_000, //relevant for fee calculation when sending the tx
                 input: (Txid([0x00; 32]), 0),
                 burn_parent_modulus: 2, // 0x5a & 0b111
 
@@ -2907,7 +2927,7 @@ mod tests {
                 ],
 
                 treatment: vec![],
-                sunset_burn: 5_500,     //relevant for fee calculation when sending the tx
+                sunset_burn: 5_500, //relevant for fee calculation when sending the tx
 
                 txid: Txid([0x00; 32]),
                 vtxindex: 0,
@@ -3141,14 +3161,14 @@ mod tests {
 
         let mut config = utils::create_config();
         config.burnchain.local_mining_public_key = Some(miner_pubkey.to_hex());
-        
+
         let mut btcd_controller = BitcoinCoreController::new(config.clone());
         btcd_controller
             .start_bitcoind()
             .expect("bitcoind should be started!");
 
         let btc_controller = BitcoinRegtestController::new(config.clone(), None);
-        
+
         btc_controller.bootstrap_chain(100);
         let utxos = btc_controller.get_all_utxos(&miner_pubkey);
         assert_eq!(0, utxos.len());
@@ -3160,11 +3180,13 @@ mod tests {
         assert_eq!(5_000_000_000, utxos[0].amount);
 
         btc_controller.build_next_block(1);
-        let utxos = btc_controller.get_all_utxos(&miner_pubkey);
+        let mut utxos = btc_controller.get_all_utxos(&miner_pubkey);
+        utxos.sort_by(|a, b| b.confirmations.cmp(&a.confirmations));
+
         assert_eq!(2, utxos.len());
-        assert_eq!(101, utxos[0].confirmations);
+        assert_eq!(102, utxos[0].confirmations);
         assert_eq!(5_000_000_000, utxos[0].amount);
-        assert_eq!(102, utxos[1].confirmations);
+        assert_eq!(101, utxos[1].confirmations);
         assert_eq!(5_000_000_000, utxos[1].amount);
     }
 
@@ -3175,15 +3197,15 @@ mod tests {
 
         let mut config = utils::create_config();
         config.burnchain.local_mining_public_key = Some(miner_pubkey.to_hex());
-        
+
         let mut btcd_controller = BitcoinCoreController::new(config.clone());
         btcd_controller
             .start_bitcoind()
             .expect("bitcoind should be started!");
 
         let btc_controller = BitcoinRegtestController::new(config.clone(), None);
         btc_controller.bootstrap_chain(101); // one utxo exists
-        
+
         let utxos = btc_controller.get_all_utxos(&other_pubkey);
         assert_eq!(0, utxos.len());
     }
@@ -3203,7 +3225,7 @@ mod tests {
         let btc_controller = BitcoinRegtestController::new(config.clone(), None);
         btc_controller.bootstrap_chain(101);
 
-        let utxos = btc_controller.get_utxos(StacksEpochId::Epoch31, &miner_pubkey, 1, None, 0);
+        let utxos = btc_controller.get_utxos(StacksEpochId::Epoch31, &miner_pubkey, 1, None, 101);
 
         let uxto_set = utxos.expect("Shouldn't be None!");
         assert_eq!(1, uxto_set.num_utxos());
@@ -3254,13 +3276,7 @@ mod tests {
         btc_controller.bootstrap_chain(101); // one utxo exists
 
         let other_pubkey = utils::create_miner2_pubkey();
-        let utxos = btc_controller.get_utxos(
-            StacksEpochId::Epoch31,
-            &other_pubkey,
-            1,
-            None,
-            0,
-        );
+        let utxos = btc_controller.get_utxos(StacksEpochId::Epoch31, &other_pubkey, 1, None, 0);
         assert!(
             utxos.is_none(),
             "None because utxos for other pubkey don't exist"
@@ -3306,7 +3322,7 @@ mod tests {
 
         let mut config = utils::create_config();
         config.burnchain.local_mining_public_key = Some(miner_pubkey.to_hex());
-        
+
         let mut btcd_controller = BitcoinCoreController::new(config.clone());
         btcd_controller
             .start_bitcoind()
@@ -3331,14 +3347,15 @@ mod tests {
     }
 
     #[test]
-    fn test_build_leader_block_commit_tx_fails_resubmitting_same_commit_op_while_prev_not_confirmed() {
+    fn test_build_leader_block_commit_tx_fails_resubmitting_same_commit_op_while_prev_not_confirmed(
+    ) {
         let keychain = utils::create_keychain();
         let miner_pubkey = keychain.get_pub_key();
         let mut signer = keychain.generate_op_signer();
 
         let mut config = utils::create_config();
         config.burnchain.local_mining_public_key = Some(miner_pubkey.to_hex());
-        
+
         let mut btcd_controller = BitcoinCoreController::new(config.clone());
         btcd_controller
             .start_bitcoind()
@@ -3372,14 +3389,15 @@ mod tests {
     }
 
     #[test]
-    fn test_build_leader_block_commit_tx_fails_resubmitting_same_commit_op_while_prev_is_confirmed() {
+    fn test_build_leader_block_commit_tx_fails_resubmitting_same_commit_op_while_prev_is_confirmed()
+    {
         let keychain = utils::create_keychain();
         let miner_pubkey = keychain.get_pub_key();
         let mut signer = keychain.generate_op_signer();
 
         let mut config = utils::create_config();
         config.burnchain.local_mining_public_key = Some(miner_pubkey.to_hex());
-        
+
         let mut btcd_controller = BitcoinCoreController::new(config.clone());
         btcd_controller
             .start_bitcoind()
@@ -3398,7 +3416,9 @@ mod tests {
             .expect("At first, building leader block commit should work");
 
         let ser = SerializedTx::new(first_tx_ok);
-        btc_controller.send_transaction(ser).expect("Tx should be sent to the burnchain!");
+        btc_controller
+            .send_transaction(ser)
+            .expect("Tx should be sent to the burnchain!");
         btc_controller.build_next_block(1); // Now tx is confirmed
 
         // re-submitting same commit while previous it is confirmed by the burnchain
@@ -3424,7 +3444,7 @@ mod tests {
 
         let mut config = utils::create_config();
         config.burnchain.local_mining_public_key = Some(miner_pubkey.to_hex());
-        
+
         let mut btcd_controller = BitcoinCoreController::new(config.clone());
         btcd_controller
             .start_bitcoind()
@@ -3443,7 +3463,9 @@ mod tests {
             .expect("At first, building leader block commit should work");
 
         let ser = SerializedTx::new(first_tx_ok);
-        btc_controller.send_transaction(ser).expect("Tx should be sent to the burnchain!");
+        btc_controller
+            .send_transaction(ser)
+            .expect("Tx should be sent to the burnchain!");
         btc_controller.build_next_block(1); // Now tx is confirmed
 
         //re-gen signer othewise fails because it will be disposed during previous commit tx.
@@ -3459,9 +3481,6 @@ mod tests {
         assert_eq!(0, rbf_tx.lock_time);
         assert_eq!(1, rbf_tx.input.len());
         assert_eq!(4, rbf_tx.output.len());
-
-        //let ong= btc_controller.ongoing_block_commit.unwrap();
-        //assert_eq!(2, ong.txids.len());
     }
 
     #[test]
@@ -3472,7 +3491,7 @@ mod tests {
 
         let mut config = utils::create_config();
         config.burnchain.local_mining_public_key = Some(miner_pubkey.to_hex());
-        
+
         let mut btcd_controller = BitcoinCoreController::new(config.clone());
         btcd_controller
             .start_bitcoind()
@@ -3491,7 +3510,9 @@ mod tests {
             .expect("At first, building leader block commit should work");
 
         let ser = SerializedTx::new(first_tx_ok);
-        btc_controller.send_transaction(ser).expect("Tx should be sent to the burnchain!");
+        btc_controller
+            .send_transaction(ser)
+            .expect("Tx should be sent to the burnchain!");
         btc_controller.build_next_block(1); // Now tx is confirmed
 
         //re-gen signer othewise fails because it will be disposed during previous commit tx.
@@ -3508,6 +3529,4 @@ mod tests {
         assert_eq!(1, rbf_tx.input.len());
         assert_eq!(4, rbf_tx.output.len());
     }
-
-    
 }

testnet/stacks-node/src/tests/bitcoin_regtest.rs

@@ -17,6 +17,9 @@ use crate::helium::RunLoop;
 use crate::tests::to_addr;
 use crate::Config;
 
+// Value usable as `BurnchainConfig::peer_port` to avoid bitcoind peer port binding
+pub const BURNCHAIN_CONFIG_PEER_PORT_DISABLED: u16 = 0;
+
 #[derive(Debug, thiserror::Error)]
 pub enum BitcoinCoreError {
     #[error("bitcoind spawn failed: {0}")]
@@ -69,9 +72,17 @@ impl BitcoinCoreController {
             .arg("-server=1")
             .arg("-listenonion=0")
             .arg("-rpcbind=127.0.0.1")
-            .arg(format!("-port={}", self.config.burnchain.peer_port))
+            //.arg(format!("-port={}", self.config.burnchain.peer_port))
             .arg(format!("-datadir={}", self.config.get_burnchain_path_str()));
 
+        let peer_port = self.config.burnchain.peer_port;
+        if peer_port == BURNCHAIN_CONFIG_PEER_PORT_DISABLED {
+            info!("Peer Port is disabled. So `-listen=0` flag will be used");
+            command.arg("-listen=0");
+        } else {
+            command.arg(format!("-port={}", peer_port));
+        }
+
         self.add_rpc_cli_args(&mut command);
 
         eprintln!("bitcoind spawn: {command:?}");