Add methods to {client, server}::Builder to set max concurrent streams (#150)

This PR adds `max_concurrent_streams()` methods to the client and server `Builder`s to set the `max_concurrent_streams` setting. I've added unit tests to ensure the correct SETTINGS frame is sent.

Closes #106
This commit is contained in:
Eliza Weisman
2017-10-10 17:36:45 -05:00
committed by GitHub
parent c6a233281a
commit 2fcf8c3740
8 changed files with 138 additions and 6 deletions

View File

@@ -22,6 +22,56 @@ fn read_preface_in_multiple_frames() {
assert!(Stream::wait(h2).next().is_none());
}
#[test]
fn server_builder_set_max_concurrent_streams() {
let _ = ::env_logger::init();
let (io, client) = mock::new();
let mut settings = frame::Settings::default();
settings.set_max_concurrent_streams(Some(1));
let client = client
.assert_server_handshake()
.unwrap()
.recv_custom_settings(settings)
.send_frame(
frames::headers(1)
.request("GET", "https://example.com/"),
)
.send_frame(
frames::headers(3)
.request("GET", "https://example.com/"),
)
.send_frame(frames::data(1, &b"hello"[..]).eos(),)
.recv_frame(frames::reset(3).refused())
.recv_frame(frames::headers(1).response(200).eos())
.close();
let mut builder = Server::builder();
builder.max_concurrent_streams(1);
let h2 = builder
.handshake::<_, Bytes>(io)
.expect("handshake")
.and_then(|srv| {
srv.into_future().unwrap().and_then(|(reqstream, srv)| {
let (req, mut stream) = reqstream.unwrap();
assert_eq!(req.method(), &http::Method::GET);
let rsp =
http::Response::builder()
.status(200).body(())
.unwrap();
stream.send_response(rsp, true).unwrap();
srv.into_future().unwrap()
})
});
h2.join(client).wait().expect("wait");
}
#[test]
fn serve_request() {
let _ = ::env_logger::init();