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:
@@ -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();
|
||||
|
||||
Reference in New Issue
Block a user