5d54d8cd79
This patch fixes a bug that prevents sent data from being flushed to the socket. When data is sent, the task managing the connection must be notified. A guard exists that prevents unnecessary notification of the connection when the stream does not have any send capacity. However, this guard was buggy. Instead of notifying the connection if *any* data can be sent, it notified the connection only when *all* data could be sent. This patch fixes the check as well as adds some tests that ensure the connection task is notified.
67 lines
1.5 KiB
Rust
67 lines
1.5 KiB
Rust
//! Utilities to support tests.
|
|
|
|
#[cfg(not(feature = "unstable"))]
|
|
compile_error!(
|
|
"Tests depend on the 'unstable' feature on h2. \
|
|
Retry with `cargo test --features unstable`"
|
|
);
|
|
|
|
pub extern crate bytes;
|
|
pub extern crate env_logger;
|
|
pub extern crate futures;
|
|
pub extern crate h2;
|
|
pub extern crate http;
|
|
pub extern crate mock_io;
|
|
pub extern crate string;
|
|
pub extern crate tokio_io;
|
|
|
|
// Kind of annoying, but we can't use macros from crates that aren't specified
|
|
// at the root.
|
|
macro_rules! try_ready {
|
|
($e:expr) => ({
|
|
use $crate::support::futures::Async;
|
|
match $e {
|
|
Ok(Async::Ready(t)) => t,
|
|
Ok(Async::NotReady) => return Ok(Async::NotReady),
|
|
Err(e) => return Err(From::from(e)),
|
|
}
|
|
})
|
|
}
|
|
|
|
macro_rules! try_nb {
|
|
($e:expr) => ({
|
|
use $crate::support::futures::Async;
|
|
use ::std::io::ErrorKind::WouldBlock;
|
|
|
|
match $e {
|
|
Ok(t) => t,
|
|
Err(ref e) if e.kind() == WouldBlock => {
|
|
return Ok(Async::NotReady)
|
|
}
|
|
Err(e) => return Err(e.into()),
|
|
}
|
|
})
|
|
}
|
|
|
|
#[macro_use]
|
|
mod assert;
|
|
|
|
#[macro_use]
|
|
pub mod raw;
|
|
|
|
pub mod frames;
|
|
pub mod prelude;
|
|
pub mod mock;
|
|
pub mod notify;
|
|
pub mod util;
|
|
|
|
mod future_ext;
|
|
|
|
pub use self::future_ext::{FutureExt, Unwrap};
|
|
|
|
// This is our test Codec type
|
|
pub type Codec<T> = h2::Codec<T, ::std::io::Cursor<::bytes::Bytes>>;
|
|
|
|
// This is the frame type that is sent
|
|
pub type SendFrame = h2::frame::Frame<::std::io::Cursor<::bytes::Bytes>>;
|