test(aggregator-client): add tests for CertificateRetriver impl

Author: Alenar

internal/mithril-aggregator-client/src/external_trait_impls.rs

@@ -28,3 +28,66 @@ impl CertificateRetriever for AggregatorHttpClient {
         message.try_into().map_err(CertificateRetrieverError)
     }
 }
+
+#[cfg(test)]
+mod tests {
+    use mithril_common::{
+        entities::ServerError, messages::CertificateMessage, test::double::Dummy,
+    };
+
+    use super::*;
+
+    #[tokio::test]
+    async fn test_retrieve_certificate_that_exist() {
+        let certificate_message = CertificateMessage::dummy();
+        let expected_certificate = certificate_message.clone().try_into().unwrap();
+
+        let (server, client) = crate::test::setup_server_and_client();
+        server.mock(|when, then| {
+            when.method(httpmock::Method::GET)
+                .path(format!("/certificate/{}", certificate_message.hash));
+            then.status(200)
+                .body(serde_json::to_string(&certificate_message).unwrap());
+        });
+
+        let certificate = client
+            .get_certificate_details(&certificate_message.hash)
+            .await
+            .unwrap();
+
+        assert_eq!(certificate, expected_certificate);
+    }
+
+    #[tokio::test]
+    async fn test_retrieve_certificate_that_does_not_exist() {
+        let (server, client) = crate::test::setup_server_and_client();
+        server.mock(|when, then| {
+            when.method(httpmock::Method::GET);
+            then.status(404);
+        });
+
+        let error = client.get_certificate_details("whatever").await.unwrap_err();
+
+        assert!(
+            format!("{error:?}").contains("Certificate does not exist"),
+            "Error message should contain 'Certificate does not exist', error:\n{error:?}",
+        );
+    }
+
+    #[tokio::test]
+    async fn test_retrieve_certificate_when_request_fails() {
+        let (server, client) = crate::test::setup_server_and_client();
+        server.mock(|when, then| {
+            when.method(httpmock::Method::GET);
+            then.status(500)
+                .body(serde_json::to_string(&ServerError::new("an error")).unwrap());
+        });
+
+        let error = client.get_certificate_details("whatever").await.unwrap_err();
+
+        assert!(
+            format!("{error:?}").contains("Failed to retrieve certificate with hash"),
+            "Error message should contain 'Failed to retrieve certificate with hash', error:\n{error:?}",
+        );
+    }
+}