client_wasm tests now use autogenerated test data from fake aggregator 'default_data' json files

Add `get_cardano_transaction_set` function

Author: dlachaume

Diff for: Cargo.lock

@@ -15,7 +15,7 @@ crate-type = ["cdylib"]
 [dependencies]
 async-trait = "0.1.77"
 futures = "0.3.30"
-mithril-client = { path = "../mithril-client" }
+mithril-client = { path = "../mithril-client", features = ["unstable"] }
 serde = { version = "1.0.196", features = ["derive"] }
 serde-wasm-bindgen = "0.6.3"
 wasm-bindgen = "0.2.90"
@@ -25,6 +25,9 @@ web-sys = { version = "0.3.67", features = ["BroadcastChannel"] }
 [dev-dependencies]
 wasm-bindgen-test = "0.3.40"
 
+[build-dependencies]
+mithril-build-script = { path = "../mithril-build-script" }
+
 [features]
 # Include nothing by default
 default = []

Diff for: mithril-client-wasm/Cargo.toml

@@ -0,0 +1,23 @@
+// build.rs
+
+use mithril_build_script::fake_aggregator::FakeAggregatorData;
+use std::env;
+use std::fs;
+use std::path::Path;
+
+fn main() {
+    let out_dir = env::var_os("OUT_DIR").unwrap();
+    let dest_path = Path::new(&out_dir).join("imported_data.rs");
+    let fake_aggregator_crate_path =
+        mithril_build_script::get_package_path("mithril-aggregator-fake");
+
+    let data_folder_path = fake_aggregator_crate_path.join("default_data");
+    let data = FakeAggregatorData::load_from_folder(&data_folder_path);
+    let generated_code = data.generate_code_for_ids();
+    fs::write(dest_path, generated_code).unwrap();
+
+    println!(
+        "cargo:rerun-if-changed={}/",
+        fake_aggregator_crate_path.display()
+    );
+}

Diff for: mithril-client-wasm/build.rs

@@ -51,9 +51,18 @@ impl From<MithrilEvent> for MithrilEventWasm {
 }
 
 /// Structure that wraps a [Client] and enables its functions to be used in WASM
-#[wasm_bindgen]
+#[wasm_bindgen(getter_with_clone)]
 pub struct MithrilClient {
     client: Client,
+
+    /// Unstable fonctions
+    pub unstable: MithrilUnstableClient,
+}
+
+#[wasm_bindgen]
+#[derive(Clone)]
+pub struct MithrilUnstableClient {
+    client: Client,
 }
 
 #[wasm_bindgen]
@@ -67,7 +76,8 @@ impl MithrilClient {
             .build()
             .map_err(|err| format!("{err:?}"))
             .unwrap();
-        MithrilClient { client }
+        let unstable = MithrilUnstableClient::new(client.clone());
+        MithrilClient { client, unstable }
     }
 
     /// Call the client to get a snapshot from a digest
@@ -200,24 +210,58 @@ impl MithrilClient {
     }
 }
 
+#[wasm_bindgen]
+impl MithrilUnstableClient {
+    /// Constructor for unstable wasm client
+    fn new(inner_client: Client) -> MithrilUnstableClient {
+        Self {
+            client: inner_client,
+        }
+    }
+
+    /// Call the client for the list of available Cardano transactions sets
+    #[wasm_bindgen]
+    pub async fn list_cardano_transaction_sets(&self) -> WasmResult {
+        let result = self
+            .client
+            .cardano_transaction_proof()
+            .list()
+            .await
+            .map_err(|err| format!("{err:?}"))?;
+
+        Ok(serde_wasm_bindgen::to_value(&result)?)
+    }
+
+    /// Call the client to get a Cardano transactions set from a hash
+    #[wasm_bindgen]
+    pub async fn get_cardano_transaction_set(&self, hash: &str) -> WasmResult {
+        let result = self
+            .client
+            .cardano_transaction_proof()
+            .get(hash)
+            .await
+            .map_err(|err| format!("{err:?}"))?
+            .ok_or(JsValue::from_str(&format!(
+                "No cardano transactions commitment found for hash: '{hash}'"
+            )))?;
+
+        Ok(serde_wasm_bindgen::to_value(&result)?)
+    }
+}
+
 #[cfg(test)]
 mod tests {
     use super::*;
+    use crate::test_data;
     use mithril_client::{
-        common::ProtocolMessage, MithrilCertificateListItem, MithrilStakeDistribution,
-        MithrilStakeDistributionListItem, Snapshot, SnapshotListItem,
+        common::ProtocolMessage, CardanoTransactionCommitment, MithrilCertificateListItem,
+        MithrilStakeDistribution, MithrilStakeDistributionListItem, Snapshot, SnapshotListItem,
     };
     use wasm_bindgen_test::*;
 
     const GENESIS_VERIFICATION_KEY: &str = "5b33322c3235332c3138362c3230312c3137372c31312c3131372c3133352c3138372c3136372c3138312c3138382c32322c35392c3230362c3130352c3233312c3135302c3231352c33302c37382c3231322c37362c31362c3235322c3138302c37322c3133342c3133372c3234372c3136312c36385d";
     const FAKE_AGGREGATOR_IP: &str = "127.0.0.1";
     const FAKE_AGGREGATOR_PORT: &str = "8000";
-    const FAKE_AGGREGATOR_SNAPSHOT_DIGEST: &str =
-        "000ee4c84c7b64a62dc30ec78a765a1f3bb81cd9dd4bd1eccf9f2da785e70877";
-    const FAKE_AGGREGATOR_MSD_HASH: &str =
-        "03ebb00e6626037f2e58eb7cc50d308fd57c253baa1fe2b04eb5945ced16b5bd";
-    const FAKE_CERTIFICATE_HASH: &str =
-        "05bf6740e781e649dd2fe7e3319818747d8038ca759c67711c90cf24cdade8a9";
 
     fn get_mithril_client() -> MithrilClient {
         MithrilClient::new(
@@ -240,19 +284,23 @@ mod tests {
             serde_wasm_bindgen::from_value::<Vec<SnapshotListItem>>(snapshots_list_js_value)
                 .expect("conversion should not fail");
 
-        assert_eq!(snapshots_list.len(), 3);
+        assert_eq!(
+            snapshots_list.len(),
+            // Aggregator return up to 20 items for a list route
+            test_data::snapshot_digests().len().min(20)
+        );
     }
 
     #[wasm_bindgen_test]
     async fn get_snapshot_should_return_value_convertible_in_rust_type() {
         let snapshot_js_value = get_mithril_client()
-            .get_snapshot(FAKE_AGGREGATOR_SNAPSHOT_DIGEST)
+            .get_snapshot(test_data::snapshot_digests()[0])
             .await
             .expect("get_snapshot should not fail");
         let snapshot = serde_wasm_bindgen::from_value::<Snapshot>(snapshot_js_value)
             .expect("conversion should not fail");
 
-        assert_eq!(snapshot.digest, FAKE_AGGREGATOR_SNAPSHOT_DIGEST);
+        assert_eq!(snapshot.digest, test_data::snapshot_digests()[0]);
     }
 
     #[wasm_bindgen_test]
@@ -274,19 +322,23 @@ mod tests {
         )
         .expect("conversion should not fail");
 
-        assert_eq!(msd_list.len(), 3);
+        assert_eq!(
+            msd_list.len(),
+            // Aggregator return up to 20 items for a list route
+            test_data::msd_hashes().len().min(20)
+        );
     }
 
     #[wasm_bindgen_test]
     async fn get_mithril_stake_distribution_should_return_value_convertible_in_rust_type() {
         let msd_js_value = get_mithril_client()
-            .get_mithril_stake_distribution(FAKE_AGGREGATOR_MSD_HASH)
+            .get_mithril_stake_distribution(test_data::msd_hashes()[0])
             .await
             .expect("get_mithril_stake_distribution should not fail");
         let msd = serde_wasm_bindgen::from_value::<MithrilStakeDistribution>(msd_js_value)
             .expect("conversion should not fail");
 
-        assert_eq!(msd.hash, FAKE_AGGREGATOR_MSD_HASH);
+        assert_eq!(msd.hash, test_data::msd_hashes()[0]);
     }
 
     #[wasm_bindgen_test]
@@ -308,20 +360,24 @@ mod tests {
         )
         .expect("conversion should not fail");
 
-        assert_eq!(certificates_list.len(), 7);
+        assert_eq!(
+            certificates_list.len(),
+            // Aggregator return up to 20 items for a list route
+            test_data::certificate_hashes().len().min(20)
+        );
     }
 
     #[wasm_bindgen_test]
     async fn get_mithril_certificate_should_return_value_convertible_in_rust_type() {
         let certificate_js_value = get_mithril_client()
-            .get_mithril_certificate(FAKE_CERTIFICATE_HASH)
+            .get_mithril_certificate(test_data::certificate_hashes()[0])
             .await
             .expect("get_mithril_certificate should not fail");
         let certificate =
             serde_wasm_bindgen::from_value::<MithrilCertificate>(certificate_js_value)
                 .expect("conversion should not fail");
 
-        assert_eq!(certificate.hash, FAKE_CERTIFICATE_HASH);
+        assert_eq!(certificate.hash, test_data::certificate_hashes()[0]);
     }
 
     #[wasm_bindgen_test]
@@ -337,7 +393,7 @@ mod tests {
     ) {
         let client = get_mithril_client();
         let msd_js_value = client
-            .get_mithril_stake_distribution(FAKE_AGGREGATOR_MSD_HASH)
+            .get_mithril_stake_distribution(test_data::msd_hashes()[0])
             .await
             .unwrap();
 
@@ -353,7 +409,7 @@ mod tests {
     async fn verify_certificate_chain_should_return_value_convertible_in_rust_type() {
         let client = get_mithril_client();
         let msd_js_value = client
-            .get_mithril_stake_distribution(FAKE_AGGREGATOR_MSD_HASH)
+            .get_mithril_stake_distribution(test_data::msd_hashes()[0])
             .await
             .unwrap();
         let msd = serde_wasm_bindgen::from_value::<MithrilStakeDistribution>(msd_js_value).unwrap();
@@ -370,7 +426,7 @@ mod tests {
     async fn verify_message_match_certificate_should_return_true() {
         let client = get_mithril_client();
         let msd_js_value = client
-            .get_mithril_stake_distribution(FAKE_AGGREGATOR_MSD_HASH)
+            .get_mithril_stake_distribution(test_data::msd_hashes()[0])
             .await
             .unwrap();
         let msd = serde_wasm_bindgen::from_value::<MithrilStakeDistribution>(msd_js_value.clone())
@@ -389,4 +445,46 @@ mod tests {
             .await
             .expect("verify_message_match_certificate should not fail");
     }
+
+    #[wasm_bindgen_test]
+    async fn list_cardano_transaction_sets_should_return_value_convertible_in_rust_type() {
+        let cardano_tx_sets_js_value = get_mithril_client()
+            .unstable
+            .list_cardano_transaction_sets()
+            .await
+            .expect("list_cardano_transaction_sets should not fail");
+        let cardano_tx_sets = serde_wasm_bindgen::from_value::<Vec<CardanoTransactionCommitment>>(
+            cardano_tx_sets_js_value,
+        )
+        .expect("conversion should not fail");
+
+        assert_eq!(
+            cardano_tx_sets.len(),
+            // Aggregator return up to 20 items for a list route
+            test_data::ctx_commitment_hashes().len().min(20)
+        );
+    }
+
+    #[wasm_bindgen_test]
+    async fn get_cardano_transaction_set_should_return_value_convertible_in_rust_type() {
+        let cardano_tx_set_js_value = get_mithril_client()
+            .unstable
+            .get_cardano_transaction_set(test_data::ctx_commitment_hashes()[0])
+            .await
+            .expect("get_cardano_transaction_set should not fail");
+        let cardano_tx_set =
+            serde_wasm_bindgen::from_value::<CardanoTransactionCommitment>(cardano_tx_set_js_value)
+                .expect("conversion should not fail");
+
+        assert_eq!(cardano_tx_set.hash, test_data::ctx_commitment_hashes()[0]);
+    }
+
+    #[wasm_bindgen_test]
+    async fn get_cardano_transaction_set_should_fail_with_unknown_digest() {
+        get_mithril_client()
+            .unstable
+            .get_cardano_transaction_set("whatever")
+            .await
+            .expect_err("get_cardano_transaction_set should fail");
+    }
 }

Diff for: mithril-client-wasm/src/client_wasm.rs

@@ -3,3 +3,6 @@
 mod client_wasm;
 
 pub use client_wasm::MithrilClient;
+
+#[cfg(test)]
+mod test_data;

Diff for: mithril-client-wasm/src/lib.rs

@@ -0,0 +1,3 @@
+#![allow(dead_code)]
+
+include!(concat!(env!("OUT_DIR"), "/imported_data.rs"));

Diff for: mithril-client-wasm/src/test_data.rs

@@ -20,6 +20,7 @@ use crate::MithrilResult;
 /// Structure that aggregates the available clients for each of the Mithril types of certified data.
 ///
 /// Use the [ClientBuilder] to instantiate it easily.
+#[derive(Clone)]
 pub struct Client {
     #[cfg(feature = "unstable")]
     cardano_transaction_proof_client: Arc<CardanoTransactionProofClient>,