From 13080c9d42860e0873d97a5a5218f2e1a4d97837 Mon Sep 17 00:00:00 2001
From: Yair <92672946+yair-starkware@users.noreply.github.com>
Date: Tue, 10 Dec 2024 14:44:12 +0200
Subject: [PATCH] fix(starknet_integration_tests): temporary fixes to make the
 e2e flow test pass (#2591)

---
 crates/starknet_batcher/src/batcher.rs                 |  3 ++-
 crates/starknet_integration_tests/src/utils.rs         | 10 +++++++++-
 .../tests/end_to_end_flow_test.rs                      |  2 +-
 3 files changed, 12 insertions(+), 3 deletions(-)

diff --git a/crates/starknet_batcher/src/batcher.rs b/crates/starknet_batcher/src/batcher.rs
index 09278e778f..c6188a96de 100644
--- a/crates/starknet_batcher/src/batcher.rs
+++ b/crates/starknet_batcher/src/batcher.rs
@@ -469,7 +469,8 @@ pub fn deadline_as_instant(deadline: chrono::DateTime<Utc>) -> BatcherResult<tok
     let time_to_deadline = deadline - chrono::Utc::now();
     let as_duration =
         time_to_deadline.to_std().map_err(|_| BatcherError::TimeToDeadlineError { deadline })?;
-    Ok((std::time::Instant::now() + as_duration).into())
+    // TODO(Matan): this is a temporary solution to the timeout issue.
+    Ok((std::time::Instant::now() + (as_duration / 2)).into())
 }
 
 fn verify_block_input(
diff --git a/crates/starknet_integration_tests/src/utils.rs b/crates/starknet_integration_tests/src/utils.rs
index eab4376332..4adda594ef 100644
--- a/crates/starknet_integration_tests/src/utils.rs
+++ b/crates/starknet_integration_tests/src/utils.rs
@@ -91,15 +91,23 @@ pub fn create_consensus_manager_configs_and_channels(
         );
     // TODO: Need to also add a channel for votes, in addition to the proposals channel.
 
+    // TODO(Matan, Dan): set reasonable default timeouts.
+    let mut timeouts = papyrus_consensus::config::TimeoutsConfig::default();
+    timeouts.precommit_timeout *= 3;
+    timeouts.prevote_timeout *= 3;
+    timeouts.proposal_timeout *= 3;
+
     let consensus_manager_configs = network_configs
         .into_iter()
         // TODO(Matan): Get config from default config file.
         .map(|network_config| ConsensusManagerConfig {
             consensus_config: ConsensusConfig {
                 start_height: BlockNumber(1),
-                consensus_delay: Duration::from_secs(1),
+		// TODO(Matan, Dan): Set the right amount
+                consensus_delay: Duration::from_secs(5),
                 network_config,
                 num_validators: u64::try_from(n_managers).unwrap(),
+                timeouts: timeouts.clone(),
                 ..Default::default()
             },
         })
diff --git a/crates/starknet_integration_tests/tests/end_to_end_flow_test.rs b/crates/starknet_integration_tests/tests/end_to_end_flow_test.rs
index f6737ac0e0..551d6ba109 100644
--- a/crates/starknet_integration_tests/tests/end_to_end_flow_test.rs
+++ b/crates/starknet_integration_tests/tests/end_to_end_flow_test.rs
@@ -39,7 +39,7 @@ async fn end_to_end_flow(mut tx_generator: MultiAccountTransactionGenerator) {
     configure_tracing().await;
 
     const LISTEN_TO_BROADCAST_MESSAGES_TIMEOUT: std::time::Duration =
-        std::time::Duration::from_secs(5);
+        std::time::Duration::from_secs(50);
     // Setup.
     let mut mock_running_system = FlowTestSetup::new_from_tx_generator(&tx_generator).await;