fix: end to end test generates datum for the correct era

It was previously using the current era (and not the next when appropriate)
to generate the transaction datum.

Author: jpraynaud

mithril-test-lab/mithril-end-to-end/src/assertions/exec.rs

@@ -18,15 +18,19 @@ pub async fn bootstrap_genesis_certificate(aggregator: &mut Aggregator) -> StdRe
     Ok(())
 }
 
-pub async fn register_era_marker(aggregator: &mut Aggregator, devnet: &Devnet) -> StdResult<()> {
+pub async fn register_era_marker(
+    aggregator: &mut Aggregator,
+    devnet: &Devnet,
+    mithril_era: &str,
+) -> StdResult<()> {
     info!("Register era marker");
 
     info!("> generating era marker tx datum...");
     let tx_datum_file_path = devnet
         .artifacts_dir
         .join(PathBuf::from("era-tx-datum.txt".to_string()));
     aggregator
-        .era_generate_tx_datum(&tx_datum_file_path)
+        .era_generate_tx_datum(&tx_datum_file_path, mithril_era)
         .await?;
 
     info!("> writing era marker on the Cardano chain...");

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

@@ -5,6 +5,7 @@ use crate::{
     GENESIS_VERIFICATION_KEY,
 };
 use anyhow::{anyhow, Context};
+use mithril_common::era::SupportedEra;
 use mithril_common::{entities, StdResult};
 use std::collections::HashMap;
 use std::path::{Path, PathBuf};
@@ -145,10 +146,19 @@ impl Aggregator {
     }
 
     // TODO: This works only for the current epoch, and needs to be fixed
-    pub async fn era_generate_tx_datum(&mut self, target_path: &Path) -> StdResult<()> {
-        let exit_status = self
-            .command
-            .start(&[
+    pub async fn era_generate_tx_datum(
+        &mut self,
+        target_path: &Path,
+        mithril_era: &str,
+    ) -> StdResult<()> {
+        let mithril_eras = SupportedEra::eras();
+        let args = if mithril_eras
+            .iter()
+            .position(|&era| era.to_string() == mithril_era)
+            == Some(0)
+        {
+            // Current era
+            vec![
                 "era".to_string(),
                 "generate-tx-datum".to_string(),
                 "--current-era-epoch".to_string(),
@@ -157,7 +167,26 @@ impl Aggregator {
                 ERA_MARKERS_SECRET_KEY.to_string(),
                 "--target-path".to_string(),
                 target_path.to_str().unwrap().to_string(),
-            ])?
+            ]
+        } else {
+            // Next era
+            vec![
+                "era".to_string(),
+                "generate-tx-datum".to_string(),
+                "--current-era-epoch".to_string(),
+                "0".to_string(),
+                "--next-era-epoch".to_string(),
+                "1".to_string(),
+                "--era-markers-secret-key".to_string(),
+                ERA_MARKERS_SECRET_KEY.to_string(),
+                "--target-path".to_string(),
+                target_path.to_str().unwrap().to_string(),
+            ]
+        };
+
+        let exit_status = self
+            .command
+            .start(&args)?
             .wait()
             .await
             .with_context(|| "`mithril-aggregator era generate-tx-datum` crashed")?;

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

@@ -63,7 +63,8 @@ impl MithrilInfrastructure {
             m: 100,
             phi_f: 0.95,
         });
-        assertions::register_era_marker(&mut aggregator, &config.devnet).await?;
+        assertions::register_era_marker(&mut aggregator, &config.devnet, &config.mithril_era)
+            .await?;
         aggregator.serve()?;
 
         let mut relay_aggregators: Vec<RelayAggregator> = vec![];