Add support for new commands for the client in e2e runner

input-output-hk · Feb 13, 2024 · 21705fc · 21705fc
1 parent 222c5a9
commit 21705fc
Showing 1 changed file with 36 additions and 1 deletion.
diff --git a/mithril-test-lab/mithril-end-to-end/src/mithril/client.rs b/mithril-test-lab/mithril-end-to-end/src/mithril/client.rs
@@ -1,6 +1,6 @@
 use crate::utils::MithrilCommand;
 use anyhow::{anyhow, Context};
-use mithril_common::StdResult;
+use mithril_common::{entities::TransactionHash, StdResult};
 use std::collections::HashMap;
 use std::path::Path;
 
@@ -22,10 +22,18 @@ pub enum MithrilStakeDistributionCommand {
     Download { hash: String },
 }
 
+#[derive(Debug)]
+pub enum CardanoTransactionCommand {
+    ListSets,
+    ShowSets { hash: String },
+    Certify { tx_hashes: Vec<TransactionHash> },
+}
+
 #[derive(Debug)]
 pub enum ClientCommand {
     Snapshot(SnapshotCommand),
     MithrilStakeDistribution(MithrilStakeDistributionCommand),
+    CardanoTransaction(CardanoTransactionCommand),
 }
 
 impl Client {
@@ -62,6 +70,33 @@ impl Client {
                     hash,
                 ],
             },
+            ClientCommand::CardanoTransaction(subcommand) => match subcommand {
+                CardanoTransactionCommand::ListSets => {
+                    vec![
+                        "--unstable".to_string(),
+                        "cardano-transaction".to_string(),
+                        "sets".to_string(),
+                        "list".to_string(),
+                    ]
+                }
+                CardanoTransactionCommand::ShowSets { hash } => {
+                    vec![
+                        "--unstable".to_string(),
+                        "cardano-transaction".to_string(),
+                        "sets".to_string(),
+                        "show".to_string(),
+                        hash,
+                    ]
+                }
+                CardanoTransactionCommand::Certify { tx_hashes } => {
+                    vec![
+                        "--unstable".to_string(),
+                        "cardano-transaction".to_string(),
+                        "certify".to_string(),
+                        tx_hashes.join(","),
+                    ]
+                }
+            },
         };
 
         let exit_status = self