diff --git a/banks-client/src/lib.rs b/banks-client/src/lib.rs index 2c954175bc..d6162a0157 100644 --- a/banks-client/src/lib.rs +++ b/banks-client/src/lib.rs @@ -385,7 +385,9 @@ mod tests { let message = Message::new(&[instruction], Some(&mint_pubkey)); Runtime::new()?.block_on(async { - let client_transport = start_local_server(bank_forks, block_commitment_cache).await; + let client_transport = + start_local_server(bank_forks, block_commitment_cache, Duration::from_millis(1)) + .await; let mut banks_client = start_client(client_transport).await?; let recent_blockhash = banks_client.get_recent_blockhash().await?; @@ -416,7 +418,9 @@ mod tests { let message = Message::new(&[instruction], Some(mint_pubkey)); Runtime::new()?.block_on(async { - let client_transport = start_local_server(bank_forks, block_commitment_cache).await; + let client_transport = + start_local_server(bank_forks, block_commitment_cache, Duration::from_millis(1)) + .await; let mut banks_client = start_client(client_transport).await?; let (_, recent_blockhash, last_valid_block_height) = banks_client.get_fees().await?; let transaction = Transaction::new(&[&genesis.mint_keypair], message, recent_blockhash); diff --git a/banks-server/src/banks_server.rs b/banks-server/src/banks_server.rs index ee896ad202..62a8570894 100644 --- a/banks-server/src/banks_server.rs +++ b/banks-server/src/banks_server.rs @@ -43,6 +43,7 @@ struct BanksServer { bank_forks: Arc>, block_commitment_cache: Arc>, transaction_sender: Sender, + poll_signature_status_sleep_duration: Duration, } impl BanksServer { @@ -54,11 +55,13 @@ impl BanksServer { bank_forks: Arc>, block_commitment_cache: Arc>, transaction_sender: Sender, + poll_signature_status_sleep_duration: Duration, ) -> Self { Self { bank_forks, block_commitment_cache, transaction_sender, + poll_signature_status_sleep_duration, } } @@ -81,6 +84,7 @@ impl BanksServer { fn new_loopback( bank_forks: Arc>, block_commitment_cache: Arc>, + poll_signature_status_sleep_duration: Duration, ) -> Self { let (transaction_sender, transaction_receiver) = channel(); let bank = bank_forks.read().unwrap().working_bank(); @@ -95,7 +99,12 @@ impl BanksServer { .name("solana-bank-forks-client".to_string()) .spawn(move || Self::run(server_bank_forks, transaction_receiver)) .unwrap(); - Self::new(bank_forks, block_commitment_cache, transaction_sender) + Self::new( + bank_forks, + block_commitment_cache, + transaction_sender, + poll_signature_status_sleep_duration, + ) } fn slot(&self, commitment: CommitmentLevel) -> Slot { @@ -120,7 +129,7 @@ impl BanksServer { .bank(commitment) .get_signature_status_with_blockhash(signature, blockhash); while status.is_none() { - sleep(Duration::from_millis(200)).await; + sleep(self.poll_signature_status_sleep_duration).await; let bank = self.bank(commitment); if bank.block_height() > last_valid_block_height { break; @@ -264,8 +273,13 @@ impl Banks for BanksServer { pub async fn start_local_server( bank_forks: Arc>, block_commitment_cache: Arc>, + poll_signature_status_sleep_duration: Duration, ) -> UnboundedChannel, ClientMessage> { - let banks_server = BanksServer::new_loopback(bank_forks, block_commitment_cache); + let banks_server = BanksServer::new_loopback( + bank_forks, + block_commitment_cache, + poll_signature_status_sleep_duration, + ); let (client_transport, server_transport) = transport::channel::unbounded(); let server = server::new(server::Config::default()) .incoming(stream::once(future::ready(server_transport))) @@ -300,8 +314,12 @@ pub async fn start_tcp_server( SendTransactionService::new(tpu_addr, &bank_forks, receiver); - let server = - BanksServer::new(bank_forks.clone(), block_commitment_cache.clone(), sender); + let server = BanksServer::new( + bank_forks.clone(), + block_commitment_cache.clone(), + sender, + Duration::from_millis(200), + ); chan.respond_with(server.serve()).execute() }) // Max 10 channels. diff --git a/program-test/src/lib.rs b/program-test/src/lib.rs index ee730c2c89..adc06d99a1 100644 --- a/program-test/src/lib.rs +++ b/program-test/src/lib.rs @@ -27,6 +27,7 @@ use { instruction::InstructionError, message::Message, native_token::sol_to_lamports, + poh_config::PohConfig, process_instruction::{ stable_log, BpfComputeBudget, InvokeContext, ProcessInstructionWithContext, }, @@ -754,7 +755,7 @@ impl ProgramTest { let mint_keypair = Keypair::new(); let voting_keypair = Keypair::new(); - let genesis_config = create_genesis_config_with_leader_ex( + let mut genesis_config = create_genesis_config_with_leader_ex( sol_to_lamports(1_000_000.0), &mint_keypair.pubkey(), &bootstrap_validator_pubkey, @@ -767,6 +768,8 @@ impl ProgramTest { ClusterType::Development, vec![], ); + let target_tick_duration = Duration::from_micros(100); + genesis_config.poh_config = PohConfig::new_sleep(target_tick_duration); debug!("Payer address: {}", mint_keypair.pubkey()); debug!("Genesis config: {}", genesis_config); @@ -836,8 +839,13 @@ impl ProgramTest { pub async fn start(self) -> (BanksClient, Keypair, Hash) { let (bank_forks, block_commitment_cache, last_blockhash, gci) = self.setup_bank(); - let transport = - start_local_server(bank_forks.clone(), block_commitment_cache.clone()).await; + let target_tick_duration = gci.genesis_config.poh_config.target_tick_duration; + let transport = start_local_server( + bank_forks.clone(), + block_commitment_cache.clone(), + target_tick_duration, + ) + .await; let banks_client = start_client(transport) .await .unwrap_or_else(|err| panic!("Failed to start banks client: {}", err)); @@ -845,7 +853,6 @@ impl ProgramTest { // Run a simulated PohService to provide the client with new blockhashes. New blockhashes // are required when sending multiple otherwise identical transactions in series from a // test - let target_tick_duration = gci.genesis_config.poh_config.target_tick_duration; tokio::spawn(async move { loop { bank_forks @@ -866,8 +873,13 @@ impl ProgramTest { /// with SOL for sending transactions pub async fn start_with_context(self) -> ProgramTestContext { let (bank_forks, block_commitment_cache, last_blockhash, gci) = self.setup_bank(); - let transport = - start_local_server(bank_forks.clone(), block_commitment_cache.clone()).await; + let target_tick_duration = gci.genesis_config.poh_config.target_tick_duration; + let transport = start_local_server( + bank_forks.clone(), + block_commitment_cache.clone(), + target_tick_duration, + ) + .await; let banks_client = start_client(transport) .await .unwrap_or_else(|err| panic!("Failed to start banks client: {}", err));