Add snapshot list message adapter to client

Author: jpraynaud

mithril-client/src/aggregator.rs

@@ -16,11 +16,13 @@ use thiserror::Error;
 use mithril_common::{
     certificate_chain::{CertificateRetriever, CertificateRetrieverError},
     entities::{Certificate, Snapshot},
-    messages::{CertificateMessage, SnapshotMessage},
+    messages::{CertificateMessage, SnapshotListMessage, SnapshotMessage},
     MITHRIL_API_VERSION,
 };
 
-use crate::{FromCertificateMessageAdapter, FromSnapshotMessageAdapter};
+use crate::{
+    FromCertificateMessageAdapter, FromSnapshotListMessageAdapter, FromSnapshotMessageAdapter,
+};
 
 /// [AggregatorHandler] related errors.
 #[derive(Error, Debug)]
@@ -173,8 +175,8 @@ impl AggregatorHandler for AggregatorHTTPClient {
 
         match response {
             Ok(response) => match response.status() {
-                StatusCode::OK => match response.json::<Vec<Snapshot>>().await {
-                    Ok(snapshots) => Ok(snapshots),
+                StatusCode::OK => match response.json::<SnapshotListMessage>().await {
+                    Ok(snapshots) => Ok(FromSnapshotListMessageAdapter::adapt(snapshots)),
                     Err(err) => Err(AggregatorHandlerError::JsonParseFailed(err.to_string())),
                 },
                 StatusCode::PRECONDITION_FAILED => Err(self.handle_api_error(&response)),

mithril-client/src/lib.rs

@@ -19,7 +19,9 @@ mod runtime;
 
 pub use aggregator::{AggregatorHTTPClient, AggregatorHandler, AggregatorHandlerError};
 pub use entities::Config;
-pub use message_adapters::{FromCertificateMessageAdapter, FromSnapshotMessageAdapter};
+pub use message_adapters::{
+    FromCertificateMessageAdapter, FromSnapshotListMessageAdapter, FromSnapshotMessageAdapter,
+};
 pub use runtime::{Runtime, RuntimeError};
 
 pub use runtime::convert_to_field_items;

mithril-client/src/message_adapters/from_snapshot_list_message.rs

@@ -0,0 +1,37 @@
+use mithril_common::entities::Snapshot;
+use mithril_common::messages::SnapshotListMessage;
+
+/// Adapter to convert [SnapshotListMessage] to [SnapshotList] instances
+pub struct FromSnapshotListMessageAdapter;
+
+impl FromSnapshotListMessageAdapter {
+    /// Method to trigger the conversion
+    pub fn adapt(snapshot_list_message: SnapshotListMessage) -> Vec<Snapshot> {
+        snapshot_list_message
+            .into_iter()
+            .map(|snapshot_list_item_message| Snapshot {
+                digest: snapshot_list_item_message.digest,
+                beacon: snapshot_list_item_message.beacon,
+                certificate_hash: snapshot_list_item_message.certificate_hash,
+                size: snapshot_list_item_message.size,
+                created_at: snapshot_list_item_message.created_at,
+                locations: snapshot_list_item_message.locations,
+            })
+            .collect()
+    }
+}
+
+#[cfg(test)]
+mod tests {
+    use mithril_common::messages::SnapshotListItemMessage;
+
+    use super::*;
+
+    #[test]
+    fn adapt_ok() {
+        let snapshot_list_message: SnapshotListMessage = vec![SnapshotListItemMessage::dummy()];
+        let snapshot_list = FromSnapshotListMessageAdapter::adapt(snapshot_list_message.clone());
+
+        assert_eq!(snapshot_list_message[0].digest, snapshot_list[0].digest);
+    }
+}

mithril-client/src/message_adapters/mod.rs

@@ -1,5 +1,7 @@
 mod from_certificate_message_adapter;
+mod from_snapshot_list_message;
 mod from_snapshot_message;
 
 pub use from_certificate_message_adapter::FromCertificateMessageAdapter;
+pub use from_snapshot_list_message::FromSnapshotListMessageAdapter;
 pub use from_snapshot_message::FromSnapshotMessageAdapter;