feat(client): enable HTTP compression using reqwest features

Author: dlachaume

Cargo.lock

@@ -84,7 +84,7 @@ warp = "0.3.7"
 
 [features]
 # Include native-tls in reqwest by default
-default = ["native-tls", "rug-backend"]
+default = ["native-tls", "rug-backend", "enable-http-compression"]
 
 # Full feature set
 full = ["fs"]
@@ -105,6 +105,9 @@ rustls-tls-manual-roots = ["reqwest/rustls-tls-manual-roots"]
 rustls-tls-webpki-roots = ["reqwest/rustls-tls-webpki-roots"]
 rustls-tls-native-roots = ["reqwest/rustls-tls-native-roots"]
 
+# Support compressed traffic with `request`
+enable-http-compression = ["reqwest/gzip", "reqwest/zstd", "reqwest/deflate", "reqwest/brotli"]
+
 # Enables `rug-backend` features for `mithril-common` dependency
 rug-backend = ["mithril-common/rug-backend"]
 # Enables `num-integer-backend` features for `mithril-common` dependency

mithril-client/Cargo.toml

@@ -850,4 +850,58 @@ mod tests {
             .await
             .expect("GET request should succeed");
     }
+
+    #[tokio::test]
+    async fn test_client_sends_accept_encoding_header_with_correct_values() {
+        let (aggregator, client) = setup_server_and_client();
+        aggregator.mock(|when, then| {
+            when.matches(|req| {
+                let headers = req.headers.clone().expect("HTTP headers not found");
+                let accept_encoding_header = headers
+                    .iter()
+                    .find(|(name, _values)| name.to_lowercase() == "accept-encoding")
+                    .expect("Accept-Encoding header not found");
+
+                let header_value = accept_encoding_header.clone().1;
+                ["gzip", "br", "deflate", "zstd"]
+                    .iter()
+                    .all(|&value| header_value.contains(value))
+            });
+
+            then.status(200).body("ok");
+        });
+
+        client
+            .get_content(AggregatorRequest::ListCertificates)
+            .await
+            .expect("GET request should succeed with Accept-Encoding header");
+    }
+
+    #[tokio::test]
+    async fn test_client_with_custom_headers_sends_accept_encoding_header_with_correct_values() {
+        let mut http_headers = HashMap::new();
+        http_headers.insert("Custom-Header".to_string(), "CustomValue".to_string());
+        let (aggregator, client) = setup_server_and_client_with_custom_headers(http_headers);
+        aggregator.mock(|when, then| {
+            when.matches(|req| {
+                let headers = req.headers.clone().expect("HTTP headers not found");
+                let accept_encoding_header = headers
+                    .iter()
+                    .find(|(name, _values)| name.to_lowercase() == "accept-encoding")
+                    .expect("Accept-Encoding header not found");
+
+                let header_value = accept_encoding_header.clone().1;
+                ["gzip", "br", "deflate", "zstd"]
+                    .iter()
+                    .all(|&value| header_value.contains(value))
+            });
+
+            then.status(200).body("ok");
+        });
+
+        client
+            .get_content(AggregatorRequest::ListCertificates)
+            .await
+            .expect("GET request should succeed with Accept-Encoding header");
+    }
 }