feat(relay): enable HTTP compression using reqwest features that add an Accept-Encoding header by default.

Author: dlachaume

Cargo.lock

@@ -33,7 +33,13 @@ libp2p = { version = "0.54.1", features = [
 ] }
 mithril-common = { path = "../mithril-common", features = ["full"] }
 mithril-doc = { path = "../internal/mithril-doc" }
-reqwest = { version = "0.12.12", features = ["json"] }
+reqwest = { version = "0.12.12", features = [
+    "json",
+    "gzip",
+    "zstd",
+    "deflate",
+    "brotli"
+    ] }
 serde = { version = "1.0.217", features = ["derive"] }
 serde_json = "1.0.138"
 serde_yaml = "0.9.34"
@@ -48,6 +54,7 @@ tokio = { version = "1.43.0", features = ["full"] }
 warp = "0.3.7"
 
 [dev-dependencies]
+httpmock = "0.7.0"
 slog-scope = "4.4.0"
 slog-term = "2.9.1"
 

mithril-relay/Cargo.toml

@@ -142,3 +142,42 @@ impl AggregatorRelay {
         self.peer.addr_peer.to_owned()
     }
 }
+
+#[cfg(test)]
+mod tests {
+    use httpmock::MockServer;
+
+    use crate::test_tools::TestLogger;
+
+    use super::*;
+
+    #[tokio::test]
+    async fn sends_accept_encoding_header_with_correct_values() {
+        let server = MockServer::start();
+        server.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(201).body("ok");
+        });
+        let addr: Multiaddr = "/ip4/0.0.0.0/tcp/0".parse().unwrap();
+        let relay = AggregatorRelay::start(&addr, &server.url(""), &TestLogger::stdout())
+            .await
+            .unwrap();
+
+        relay
+            .notify_signature_to_aggregator(&RegisterSignatureMessage::dummy())
+            .await
+            .expect("Should succeed with Accept-Encoding header");
+    }
+}

mithril-relay/src/relay/aggregator.rs

@@ -329,3 +329,39 @@ mod handlers {
         }
     }
 }
+
+#[cfg(test)]
+mod tests {
+    use httpmock::MockServer;
+
+    use crate::test_tools::TestLogger;
+
+    use super::*;
+
+    #[tokio::test]
+    async fn sends_accept_encoding_header_with_correct_values() {
+        let server = MockServer::start();
+        let mock = server.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");
+        });
+
+        handlers::aggregator_features_handler(TestLogger::stdout(), server.url(""))
+            .await
+            .unwrap();
+
+        mock.assert();
+    }
+}