chore(network): test 10K messages 2MB size 1000000 sleep 1Mps - fix1

starkware-libs · Dec 10, 2024 · aa5bdd3 · aa5bdd3
1 parent c785c1a
commit aa5bdd3
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/crates/papyrus_network/src/bin/network_stress_test/main.rs b/crates/papyrus_network/src/bin/network_stress_test/main.rs
@@ -34,7 +34,7 @@ async fn main() {
         _ = network_manager.run() => {}
         _ = async {
             let mut i = 0;
-            tokio::time::sleep(std::time::Duration::from_secs(80)).await;
+            tokio::time::sleep(std::time::Duration::from_secs(20)).await;
             loop {
                 let message = StressTestMessage::new(i, vec![0; message_size - METADATA_SIZE], peer_id.clone());
                 network_channels.broadcast_topic_client.broadcast_message(message).await.unwrap();
@@ -53,7 +53,7 @@ async fn main() {
             let mut i = 0;
             loop {
                 let maybe_response = timeout(
-                    std::time::Duration::from_secs(120),
+                    std::time::Duration::from_secs(150),
                     network_channels.broadcasted_messages_receiver.next(),
                 ).await;
                 match maybe_response {