Fix end to end test stake delegation

Author: jpraynaud

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

@@ -16,10 +16,10 @@ pub async fn bootstrap_genesis_certificate(aggregator: &mut Aggregator) -> StdRe
     Ok(())
 }
 
-pub async fn delegate_stakes_to_pools(devnet: &Devnet) -> StdResult<()> {
+pub async fn delegate_stakes_to_pools(devnet: &Devnet, delegation_round: u16) -> StdResult<()> {
     info!("Delegate stakes to the cardano pools");
 
-    devnet.delegate_stakes().await?;
+    devnet.delegate_stakes(delegation_round).await?;
 
     Ok(())
 }

mithril-test-lab/mithril-end-to-end/src/devnet/runner.rs

@@ -204,13 +204,14 @@ impl Devnet {
         }
     }
 
-    pub async fn delegate_stakes(&self) -> StdResult<()> {
+    pub async fn delegate_stakes(&self, delegation_round: u16) -> StdResult<()> {
         let run_script = "delegate.sh";
         let run_script_path = self.artifacts_dir.join(run_script);
         let mut run_command = Command::new(&run_script_path);
         run_command
             .current_dir(&self.artifacts_dir)
             .kill_on_drop(true);
+        run_command.env("DELEGATION_ROUND", delegation_round.to_string());
 
         info!("Delegating stakes to the pools"; "script" => &run_script_path.display());
 

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

@@ -42,7 +42,9 @@ impl<'a> Spec<'a> {
             "epoch after which the stake distribution will change".to_string(),
         )
         .await?;
-        assertions::delegate_stakes_to_pools(self.infrastructure.devnet()).await?;
+        let delegation_round = 1;
+        assertions::delegate_stakes_to_pools(self.infrastructure.devnet(), delegation_round)
+            .await?;
 
         // Wait 2 epochs before changing protocol parameters
         target_epoch += 2;