From e1d8f6efb892d48615c5a57113c2100a5f52ea36 Mon Sep 17 00:00:00 2001
From: sfauvel <sfauvel@palo-it.com>
Date: Thu, 16 May 2024 10:56:51 +0200
Subject: [PATCH] Keep only first part of stress test to be faster

---
 .../src/bin/load-aggregator/main.rs           | 92 +++++++++----------
 1 file changed, 46 insertions(+), 46 deletions(-)

diff --git a/mithril-test-lab/mithril-end-to-end/src/bin/load-aggregator/main.rs b/mithril-test-lab/mithril-end-to-end/src/bin/load-aggregator/main.rs
index 0f472191c11..a4c518720ef 100644
--- a/mithril-test-lab/mithril-end-to-end/src/bin/load-aggregator/main.rs
+++ b/mithril-test-lab/mithril-end-to-end/src/bin/load-aggregator/main.rs
@@ -67,52 +67,52 @@ async fn main() -> StdResult<()> {
     let aggregator_endpoint = aggregator.endpoint();
     reporter.stop();
 
-    let mut scenario_parameters = ScenarioParameters {
-        aggregator,
-        aggregator_parameters,
-        signers_fixture,
-        immutable_db,
-        reporter,
-        precomputed_mithril_stake_distribution_signatures: mithril_stake_distribution_signatures,
-    };
-
-    let scenario_counters = ScenarioCounters {
-        current_epoch,
-        number_of_certificates: 1,
-        number_of_mithril_stake_distributions: 0,
-        number_of_snapshots: 0,
-    };
-
-    info!(">> Run phase 1 without client load");
-    let scenario_counters = main_scenario(scenario_counters, &mut scenario_parameters).await?;
-
-    info!(">> Run phase 2 with client load");
-    let (shutdown_tx, mut shutdown_rx) = oneshot::channel();
-    let clients_handle = tokio::spawn(async move {
-        if opts.num_clients > 0 {
-            loop {
-                tokio::select! {
-                    _msg = &mut shutdown_rx => {
-                        break;
-                    }
-                    _ = clients_scenario(aggregator_endpoint.clone(), opts.num_clients) => {
-
-                    }
-                }
-            }
-        }
-    });
-    main_scenario(scenario_counters, &mut scenario_parameters)
-        .await
-        .expect("the main scenario should not fail");
-    let _ = shutdown_tx.send(());
-    clients_handle.await.unwrap();
-
-    info!(">> Display execution timings:");
-    scenario_parameters.reporter.print_report();
-
-    info!(">> All steps executed successfully, stopping all tasks...");
-    scenario_parameters.aggregator.stop().await.unwrap();
+    // let mut scenario_parameters = ScenarioParameters {
+    //     aggregator,
+    //     aggregator_parameters,
+    //     signers_fixture,
+    //     immutable_db,
+    //     reporter,
+    //     precomputed_mithril_stake_distribution_signatures: mithril_stake_distribution_signatures,
+    // };
+
+    // let scenario_counters = ScenarioCounters {
+    //     current_epoch,
+    //     number_of_certificates: 1,
+    //     number_of_mithril_stake_distributions: 0,
+    //     number_of_snapshots: 0,
+    // };
+
+    // info!(">> Run phase 1 without client load");
+    // let scenario_counters = main_scenario(scenario_counters, &mut scenario_parameters).await?;
+
+    // info!(">> Run phase 2 with client load");
+    // let (shutdown_tx, mut shutdown_rx) = oneshot::channel();
+    // let clients_handle = tokio::spawn(async move {
+    //     if opts.num_clients > 0 {
+    //         loop {
+    //             tokio::select! {
+    //                 _msg = &mut shutdown_rx => {
+    //                     break;
+    //                 }
+    //                 _ = clients_scenario(aggregator_endpoint.clone(), opts.num_clients) => {
+
+    //                 }
+    //             }
+    //         }
+    //     }
+    // });
+    // main_scenario(scenario_counters, &mut scenario_parameters)
+    //     .await
+    //     .expect("the main scenario should not fail");
+    // let _ = shutdown_tx.send(());
+    // clients_handle.await.unwrap();
+
+    // info!(">> Display execution timings:");
+    // scenario_parameters.reporter.print_report();
+
+    // info!(">> All steps executed successfully, stopping all tasks...");
+    // scenario_parameters.aggregator.stop().await.unwrap();
 
     Ok(())
 }