Mock cardano-cli now read epoch from file

Author: Alenar

mithril-test-lab/mithril-end-to-end/script/mock-cardano-cli

@@ -20,13 +20,15 @@ main () {
     done
 
     case $command_to_run in
-        stake-snapshot) cat $STAKE_DISTRIBUTION_FILE ;;
-        chain-tip) cat <<EOF
+        stake-snapshot) cat $MOCK_STAKE_DISTRIBUTION_FILE ;;
+        chain-tip) epoch=$(cat $MOCK_EPOCH_FILE) 
+        
+        cat <<EOF
 {
     "era": "Babbage",
     "syncProgress": "100.00",
     "hash": "f6d1b8c328697c7a4a8e7f718c79510acbcd411ff4ca19401ded13534d45a38d",
-    "epoch": 1,
+    "epoch": $epoch,
     "slot": 0,
     "block": 0
 }

mithril-test-lab/mithril-end-to-end/src/bin/load-aggregator/main.rs

@@ -1,4 +1,10 @@
-use std::{fs::File, path::PathBuf, sync::Arc, time::Duration};
+use std::{
+    fs::File,
+    io::Write,
+    path::{Path, PathBuf},
+    sync::Arc,
+    time::Duration,
+};
 
 use anyhow::Context;
 use clap::Parser;
@@ -87,6 +93,24 @@ pub async fn wait_for_http_response(url: &str, timeout: Duration, message: &str)
     }
 }
 
+pub fn write_stake_distribution(
+    mock_stake_distribution_file_path: &Path,
+    signers_fixture: &MithrilFixture,
+) {
+    let mock_stake_distribution_file = File::create(mock_stake_distribution_file_path).unwrap();
+    serde_json::to_writer(
+        &mock_stake_distribution_file,
+        &signers_fixture.cardano_cli_stake_distribution(),
+    )
+    .expect("Writing the stake distribution into a file for the mock cardano cli failed");
+}
+
+pub fn write_epoch(mock_epoch_file_path: &Path, epoch: Epoch) {
+    let mock_epoch_file = File::create(mock_epoch_file_path).unwrap();
+    write!(&mock_epoch_file, "{}", *epoch)
+        .expect("Writing the epoch into a file for the mock cardano cli failed");
+}
+
 #[derive(Debug, Parser)]
 #[command(author, version, about, long_about = None)]
 pub struct MainOpts {
@@ -205,6 +229,7 @@ async fn main() -> StdResult<()> {
     let _logger = init_logger(&opts);
     let args = AggregatorParameters::new(&opts)?;
     let mock_stake_distribution_file_path = args.work_dir.join("stake_distribution.json");
+    let mock_epoch_file_path = args.work_dir.join("epoch.txt");
     info!(">> Starting stress test with options: {opts:?}");
 
     info!(">> Creation of the Signer Key Registrations payloads");
@@ -222,15 +247,12 @@ async fn main() -> StdResult<()> {
     )
     .unwrap();
 
-    let mock_stake_distribution_file = File::create(&mock_stake_distribution_file_path).unwrap();
-    serde_json::to_writer(
-        &mock_stake_distribution_file,
-        &signers_fixture.cardano_cli_stake_distribution(),
-    )
-    .expect("Writing the stake distribution into a file for the mock cardano cli failed");
+    write_epoch(&mock_epoch_file_path, Epoch(26));
+    write_stake_distribution(&mock_stake_distribution_file_path, &signers_fixture);
 
     aggregator.change_run_interval(Duration::from_secs(6));
-    aggregator.set_mock_cardano_cli_stake_distribution_file(&mock_stake_distribution_file_path);
+    aggregator
+        .set_mock_cardano_cli_file_path(&mock_stake_distribution_file_path, &mock_epoch_file_path);
     aggregator.set_protocol_parameters(&ProtocolParameters::default());
     aggregator.serve().unwrap();
 

mithril-test-lab/mithril-end-to-end/src/mithril/aggregator.rs

@@ -124,9 +124,17 @@ impl Aggregator {
         );
     }
 
-    pub fn set_mock_cardano_cli_stake_distribution_file(&mut self, file_path: &Path) {
+    pub fn set_mock_cardano_cli_file_path(
+        &mut self,
+        stake_distribution_file: &Path,
+        epoch_file_path: &Path,
+    ) {
+        self.command.set_env_var(
+            "MOCK_STAKE_DISTRIBUTION_FILE",
+            stake_distribution_file.to_str().unwrap(),
+        );
         self.command
-            .set_env_var("STAKE_DISTRIBUTION_FILE", file_path.to_str().unwrap())
+            .set_env_var("MOCK_EPOCH_FILE", epoch_file_path.to_str().unwrap());
     }
 
     /// Change the run interval of the aggregator state machine (default: 400ms)