fix(network): increase stress test sleep

starkware-libs · Nov 14, 2024 · fca05b6 · fca05b6
1 parent 5097548
commit fca05b6
Showing 1 changed file with 2 additions and 3 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
@@ -49,7 +49,7 @@ async fn main() {
             let mut i = 0;
             loop {
                 let maybe_response = timeout(
-                    std::time::Duration::from_secs(60),
+                    std::time::Duration::from_secs(120),
                     network_channels.broadcasted_messages_receiver.next(),
                 ).await;
                 match maybe_response {
@@ -72,7 +72,6 @@ async fn main() {
                         });
                         i += 1;
                         if i == num_messages * 4 {
-                            tokio::time::sleep(std::time::Duration::from_secs(60)).await;
                             break;
                         }
                     }
@@ -84,7 +83,7 @@ async fn main() {
             for record in output_vector {
                 wtr.serialize(record).unwrap();
             }
-            tokio::time::sleep(std::time::Duration::from_secs(60)).await;
+            tokio::time::sleep(std::time::Duration::from_secs(120)).await;
         }
     }
 }