fix(e2e): delegate stakes only from the first aggregator

Author: jpraynaud

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

@@ -138,7 +138,7 @@ impl Spec {
         )
         .await?;
 
-        if aggregator.is_master() {
+        if aggregator.index() == 0 {
             // Delegate some stakes to pools
             let delegation_round = 1;
             assertions::delegate_stakes_to_pools(infrastructure.devnet(), delegation_round).await?;

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

@@ -18,6 +18,7 @@ use tokio::sync::RwLock;
 #[derive(Debug)]
 pub struct AggregatorConfig<'a> {
     pub is_master: bool,
+    pub index: usize,
     pub name: &'a str,
     pub server_port: u64,
     pub pool_node: &'a PoolNode,
@@ -39,6 +40,7 @@ pub struct AggregatorConfig<'a> {
 #[derive(Debug)]
 pub struct Aggregator {
     is_master: bool,
+    index: usize,
     name_suffix: String,
     server_port: u64,
     db_directory: PathBuf,
@@ -134,6 +136,7 @@ impl Aggregator {
 
         Ok(Self {
             is_master: aggregator_config.is_master,
+            index: aggregator_config.index,
             name_suffix: aggregator_config.name.to_string(),
             server_port: aggregator_config.server_port,
             db_directory: aggregator_config.pool_node.db_path.clone(),
@@ -145,6 +148,7 @@ impl Aggregator {
     pub fn copy_configuration(other: &Aggregator) -> Self {
         Self {
             is_master: other.is_master,
+            index: other.index,
             name_suffix: other.name_suffix.clone(),
             server_port: other.server_port,
             db_directory: other.db_directory.clone(),
@@ -157,6 +161,10 @@ impl Aggregator {
         self.is_master
     }
 
+    pub fn index(&self) -> usize {
+        self.index
+    }
+
     pub fn name(&self) -> String {
         format!("mithril-aggregator-{}", self.name_suffix)
     }

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

@@ -238,6 +238,7 @@ impl MithrilInfrastructure {
                 .join(format!("aggregator-{aggregator_name}"));
             let aggregator = Aggregator::new(&AggregatorConfig {
                 is_master: config.is_master_aggregator(index),
+                index,
                 name: &aggregator_name,
                 server_port: config.server_port + index as u64,
                 pool_node,
@@ -264,16 +265,19 @@ impl MithrilInfrastructure {
                 })
                 .await;
 
-            if master_aggregator_endpoint.is_none() {
+            if master_aggregator_endpoint.is_none() && config.is_master_aggregator(0) {
                 master_aggregator_endpoint = Some(aggregator.endpoint());
-                Self::register_startup_era(&aggregator, config).await?;
             }
 
-            aggregator.serve().await?;
-
             aggregators.push(aggregator);
         }
 
+        Self::register_startup_era(&aggregators[0], config).await?;
+
+        for aggregator in &aggregators {
+            aggregator.serve().await?;
+        }
+
         Ok(aggregators)
     }
 

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

@@ -94,7 +94,7 @@ impl RunOnly {
         assertions::bootstrap_genesis_certificate(aggregator).await?;
         assertions::wait_for_epoch_settings(&aggregator.endpoint()).await?;
 
-        if aggregator.is_master() {
+        if aggregator.index() == 0 {
             // Transfer some funds on the devnet to have some Cardano transactions to sign
             assertions::transfer_funds(infrastructure.devnet()).await?;
         }

mithril-test-lab/mithril-end-to-end/src/stress_test/aggregator_helpers.rs

@@ -20,6 +20,7 @@ pub async fn bootstrap_aggregator(
 
     let mut aggregator = Aggregator::new(&AggregatorConfig {
         is_master: false,
+        index: 0,
         name: "genesis",
         server_port: args.server_port as u64,
         pool_node: &args.pool_node,