feat: add 'header_table_size()' to server builder (#888)

improve test

Author: ArniDagur

src/server.rs

@@ -800,6 +800,18 @@ impl Builder {
         self
     }
 
+    /// Sets the header table size.
+    ///
+    /// This setting informs the peer of the maximum size of the header compression
+    /// table used to encode header blocks, in octets. The encoder may select any value
+    /// equal to or less than the header table size specified by the sender.
+    ///
+    /// The default value is 4,096.
+    pub fn header_table_size(&mut self, size: u32) -> &mut Self {
+        self.settings.set_header_table_size(Some(size));
+        self
+    }
+
     /// Sets the maximum number of concurrent streams.
     ///
     /// The maximum concurrent streams setting only controls the maximum number

tests/h2-tests/tests/server.rs

@@ -69,6 +69,51 @@ async fn server_builder_set_max_concurrent_streams() {
     join(client, h2).await;
 }
 
+#[tokio::test]
+async fn server_builder_header_table_size() {
+    h2_support::trace_init!();
+
+    for size in [0, 10000] {
+        let (io, mut client) = mock::new();
+
+        let mut expected = frame::Settings::default();
+        expected.set_header_table_size(Some(size));
+
+        let client = async move {
+            let recv_settings = client.assert_server_handshake().await;
+            assert_frame_eq(recv_settings, expected);
+            client
+                .send_frame(
+                    frames::headers(1)
+                        .request("GET", "https://example.com/")
+                        .eos(),
+                )
+                .await;
+            client
+                .recv_frame(frames::headers(1).response(200).eos())
+                .await;
+        };
+
+        let mut builder = server::Builder::new();
+        builder.header_table_size(size);
+
+        let h2 = async move {
+            let mut srv = builder.handshake::<_, Bytes>(io).await.expect("handshake");
+            let (req, mut stream) = srv.next().await.unwrap().unwrap();
+            assert_eq!(req.method(), &http::Method::GET);
+            stream
+                .send_response(
+                    http::Response::builder().status(200).body(()).unwrap(),
+                    true,
+                )
+                .unwrap();
+            assert!(srv.next().await.is_none());
+        };
+
+        join(client, h2).await;
+    }
+}
+
 #[tokio::test]
 async fn serve_request() {
     h2_support::trace_init!();