test: add a test to check that the compression is enabled in signer and aggregator

Author: Alenar

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

@@ -7,7 +7,6 @@ mod client;
 mod error;
 mod external_trait_impls;
 pub mod query;
-#[cfg(test)]
 mod test;
 
 pub use builder::AggregatorClientBuilder;

internal/mithril-aggregator-client/src/test/mod.rs

@@ -1,5 +1,7 @@
+#[cfg(test)]
 use httpmock::MockServer;
 
+#[cfg(test)]
 use crate::AggregatorHttpClient;
 
 #[cfg(test)]
@@ -29,3 +31,36 @@ macro_rules! assert_error_matches {
 }
 #[cfg(test)]
 pub(crate) use assert_error_matches;
+
+/// Define a test that checks that http compression is enabled for the client.
+///
+/// Requires `httpmock` in dev-dependencies
+#[macro_export]
+macro_rules! test_http_compression_is_enabled {
+    () => {
+        #[tokio::test]
+        async fn test_http_compression_is_enabled_and_send_accept_encoding_header_with_correct_values() {
+            let server = httpmock::MockServer::start();
+            server.mock(|when, then| {
+                when.is_true(|req| {
+                    let headers = req.headers();
+                    let accept_encoding_header = headers
+                        .get("accept-encoding")
+                        .expect("Accept-Encoding header not found");
+
+                    ["gzip", "br", "deflate", "zstd"].iter().all(|&encoding| {
+                        accept_encoding_header.to_str().is_ok_and(|h| h.contains(encoding))
+                    })
+                });
+
+                then.status(200).body("[]");
+            });
+
+            let client = $crate::AggregatorHttpClient::builder(server.base_url()).build().unwrap();
+            client
+                .send($crate::query::GetCertificatesListQuery::latest())
+                .await
+                .expect("GET request should succeed with Accept-Encoding header");
+        }
+    };
+}

mithril-aggregator/src/lib.rs

@@ -69,3 +69,8 @@ use tikv_jemallocator::Jemalloc;
 #[cfg(all(not(target_env = "msvc"), feature = "jemallocator"))]
 #[global_allocator]
 static GLOBAL: Jemalloc = Jemalloc;
+
+#[cfg(test)]
+mod tests {
+    mithril_aggregator_client::test_http_compression_is_enabled!();
+}

mithril-signer/src/lib.rs

@@ -43,3 +43,8 @@ static GLOBAL: Jemalloc = Jemalloc;
 pub(crate) mod test_tools {
     mithril_common::define_test_logger!();
 }
+
+#[cfg(test)]
+mod tests {
+    mithril_aggregator_client::test_http_compression_is_enabled!();
+}