Add to SnapshotDownloadMessage adapter

Author: jpraynaud

mithril-client/src/message_adapters/mod.rs

@@ -1,7 +1,9 @@
 mod from_certificate_message_adapter;
 mod from_mithril_stake_distribution_message;
 mod from_snapshot_message;
+mod to_snapshot_download_message;
 
 pub use from_certificate_message_adapter::FromCertificateMessageAdapter;
 pub use from_mithril_stake_distribution_message::FromMithrilStakeDistributionMessageAdapter;
 pub use from_snapshot_message::FromSnapshotMessageAdapter;
+pub use to_snapshot_download_message::ToSnapshotDownloadMessageAdapter;

mithril-client/src/message_adapters/to_snapshot_download_message.rs

@@ -0,0 +1,35 @@
+use mithril_common::entities::Snapshot;
+use mithril_common::messages::{SnapshotDownloadMessage, ToMessageAdapter};
+
+/// Adapter to convert [Snapshot] to [SnapshotDownloadMessage] instances
+pub struct ToSnapshotDownloadMessageAdapter;
+
+impl ToMessageAdapter<&Snapshot, SnapshotDownloadMessage> for ToSnapshotDownloadMessageAdapter {
+    /// Method to trigger the conversion
+    fn adapt(snapshot: &Snapshot) -> SnapshotDownloadMessage {
+        SnapshotDownloadMessage {
+            digest: snapshot.digest.clone(),
+            beacon: snapshot.beacon.clone(),
+            size: snapshot.size,
+            locations: snapshot.locations.clone(),
+            compression_algorithm: snapshot.compression_algorithm,
+            cardano_node_version: snapshot.cardano_node_version.clone(),
+        }
+    }
+}
+
+#[cfg(test)]
+mod tests {
+    use mithril_common::test_utils::fake_data;
+
+    use super::*;
+
+    #[test]
+    fn adapt_ok() {
+        let mut snapshot = fake_data::snapshots(1)[0].to_owned();
+        snapshot.digest = "digest123".to_string();
+        let snapshot_download_message = ToSnapshotDownloadMessageAdapter::adapt(&snapshot);
+
+        assert_eq!("digest123".to_string(), snapshot_download_message.digest);
+    }
+}