diff --git a/client/src/connection_cache.rs b/client/src/connection_cache.rs index e6387ce008..c66dda1fba 100644 --- a/client/src/connection_cache.rs +++ b/client/src/connection_cache.rs @@ -472,7 +472,10 @@ mod tests { }, signature::Keypair, }, - solana_streamer::{quic::StreamStats, streamer::StakedNodes}, + solana_streamer::{ + nonblocking::quic::DEFAULT_WAIT_FOR_CHUNK_TIMEOUT_MS, quic::StreamStats, + streamer::StakedNodes, + }, std::{ net::{IpAddr, Ipv4Addr, SocketAddr, UdpSocket}, sync::{ @@ -665,7 +668,7 @@ mod tests { 10, 10, response_recv_stats, - 1000, + DEFAULT_WAIT_FOR_CHUNK_TIMEOUT_MS, ) .unwrap(); diff --git a/quic-client/tests/quic_client.rs b/quic-client/tests/quic_client.rs index 8a5feace37..946cd0bab6 100644 --- a/quic-client/tests/quic_client.rs +++ b/quic-client/tests/quic_client.rs @@ -9,8 +9,8 @@ mod tests { }, solana_sdk::{packet::PACKET_DATA_SIZE, signature::Keypair}, solana_streamer::{ - quic::StreamStats, streamer::StakedNodes, - tls_certificates::new_self_signed_tls_certificate_chain, + nonblocking::quic::DEFAULT_WAIT_FOR_CHUNK_TIMEOUT_MS, quic::StreamStats, + streamer::StakedNodes, tls_certificates::new_self_signed_tls_certificate_chain, }, solana_tpu_client::connection_cache_stats::ConnectionCacheStats, std::{ @@ -85,7 +85,7 @@ mod tests { 10, 10, stats, - 1000, + DEFAULT_WAIT_FOR_CHUNK_TIMEOUT_MS, ) .unwrap(); @@ -189,7 +189,7 @@ mod tests { 10, 10, request_recv_stats, - 1000, + DEFAULT_WAIT_FOR_CHUNK_TIMEOUT_MS, ) .unwrap(); @@ -218,7 +218,7 @@ mod tests { 10, 10, response_recv_stats, - 1000, + DEFAULT_WAIT_FOR_CHUNK_TIMEOUT_MS, ) .unwrap(); diff --git a/streamer/src/quic.rs b/streamer/src/quic.rs index 3de5800df0..8e23f1e029 100644 --- a/streamer/src/quic.rs +++ b/streamer/src/quic.rs @@ -345,7 +345,9 @@ pub fn spawn_server( #[cfg(test)] mod test { use { - super::*, crate::nonblocking::quic::test::*, crossbeam_channel::unbounded, + super::*, + crate::nonblocking::quic::{test::*, DEFAULT_WAIT_FOR_CHUNK_TIMEOUT_MS}, + crossbeam_channel::unbounded, std::net::SocketAddr, }; @@ -374,7 +376,7 @@ mod test { MAX_STAKED_CONNECTIONS, MAX_UNSTAKED_CONNECTIONS, stats, - 100, + DEFAULT_WAIT_FOR_CHUNK_TIMEOUT_MS, ) .unwrap(); (t, exit, receiver, server_address) @@ -430,7 +432,7 @@ mod test { MAX_STAKED_CONNECTIONS, MAX_UNSTAKED_CONNECTIONS, stats, - 100, + DEFAULT_WAIT_FOR_CHUNK_TIMEOUT_MS, ) .unwrap(); @@ -473,7 +475,7 @@ mod test { MAX_STAKED_CONNECTIONS, 0, // Do not allow any connection from unstaked clients/nodes stats, - 100, + DEFAULT_WAIT_FOR_CHUNK_TIMEOUT_MS, ) .unwrap();