0c59957d88
When all Streams are dropped / finished, the Connection was held open until the peer hangs up. Instead, the Connection should hang up once it knows that nothing more will be sent. To fix this, we notify the Connection when a stream is no longer referenced. On the Connection poll(), we check that there are no active, held, reset streams or any references to the Streams and transition to sending a GOAWAY if that is case. The specific behavior depends on if running as a client or server.
110 lines
3.0 KiB
Rust
110 lines
3.0 KiB
Rust
#[macro_use]
|
|
pub mod support;
|
|
use support::prelude::*;
|
|
|
|
#[test]
|
|
fn recv_single_ping() {
|
|
let _ = ::env_logger::init();
|
|
let (m, mock) = mock::new();
|
|
|
|
// Create the handshake
|
|
let h2 = client::handshake(m)
|
|
.unwrap()
|
|
.and_then(|(client, conn)| {
|
|
conn.unwrap()
|
|
.map(|c| (client, c))
|
|
});
|
|
|
|
let mock = mock.assert_client_handshake()
|
|
.unwrap()
|
|
.and_then(|(_, mut mock)| {
|
|
let frame = frame::Ping::new(Default::default());
|
|
mock.send(frame.into()).unwrap();
|
|
|
|
mock.into_future().unwrap()
|
|
})
|
|
.and_then(|(frame, _)| {
|
|
let pong = assert_ping!(frame.unwrap());
|
|
|
|
// Payload is correct
|
|
assert_eq!(*pong.payload(), <[u8; 8]>::default());
|
|
|
|
// Is ACK
|
|
assert!(pong.is_ack());
|
|
|
|
Ok(())
|
|
});
|
|
|
|
let _ = h2.join(mock).wait().unwrap();
|
|
}
|
|
|
|
#[test]
|
|
fn recv_multiple_pings() {
|
|
let _ = ::env_logger::init();
|
|
let (io, client) = mock::new();
|
|
|
|
let client = client.assert_server_handshake()
|
|
.expect("client handshake")
|
|
.recv_settings()
|
|
.send_frame(frames::ping([1; 8]))
|
|
.send_frame(frames::ping([2; 8]))
|
|
.recv_frame(frames::ping([1; 8]).pong())
|
|
.recv_frame(frames::ping([2; 8]).pong())
|
|
.close();
|
|
|
|
let srv = server::handshake(io)
|
|
.expect("handshake")
|
|
.and_then(|srv| {
|
|
// future of first request, which never comes
|
|
srv.into_future().unwrap()
|
|
});
|
|
|
|
srv.join(client).wait().expect("wait");
|
|
}
|
|
|
|
#[test]
|
|
fn pong_has_highest_priority() {
|
|
let _ = ::env_logger::init();
|
|
let (io, client) = mock::new();
|
|
|
|
let data = Bytes::from(vec![0; 16_384]);
|
|
|
|
let client = client.assert_server_handshake()
|
|
.expect("client handshake")
|
|
.recv_settings()
|
|
.send_frame(
|
|
frames::headers(1)
|
|
.request("POST", "https://http2.akamai.com/")
|
|
)
|
|
.send_frame(frames::data(1, data.clone()).eos())
|
|
.send_frame(frames::ping([1; 8]))
|
|
.recv_frame(frames::ping([1; 8]).pong())
|
|
.recv_frame(frames::headers(1).response(200).eos())
|
|
.close();
|
|
|
|
let srv = server::handshake(io)
|
|
.expect("handshake")
|
|
.and_then(|srv| {
|
|
// future of first request
|
|
srv.into_future().unwrap()
|
|
}).and_then(move |(reqstream, srv)| {
|
|
let (req, mut stream) = reqstream.expect("request");
|
|
assert_eq!(req.method(), "POST");
|
|
let body = req.into_parts().1;
|
|
|
|
body.concat2()
|
|
.expect("body")
|
|
.and_then(move |body| {
|
|
assert_eq!(body.len(), data.len());
|
|
let res = Response::builder()
|
|
.status(200)
|
|
.body(())
|
|
.unwrap();
|
|
stream.send_response(res, true).expect("response");
|
|
srv.into_future().unwrap()
|
|
})
|
|
});
|
|
|
|
srv.join(client).wait().expect("wait");
|
|
}
|