diff --git a/tokio-stream/tests/stream_timeout.rs b/tokio-stream/tests/stream_timeout.rs index 5697ace69..2338f8335 100644 --- a/tokio-stream/tests/stream_timeout.rs +++ b/tokio-stream/tests/stream_timeout.rs @@ -1,10 +1,10 @@ -#![cfg(feature = "full")] +#![cfg(all(feature = "time", feature = "sync", feature = "io-util"))] use tokio::time::{self, sleep, Duration}; use tokio_stream::{self, StreamExt}; use tokio_test::*; -use futures::StreamExt as _; +use futures::stream; async fn maybe_sleep(idx: i32) -> i32 { if idx % 2 == 0 { diff --git a/tokio-stream/tests/time_throttle.rs b/tokio-stream/tests/time_throttle.rs index 42a643bfa..e6c9917be 100644 --- a/tokio-stream/tests/time_throttle.rs +++ b/tokio-stream/tests/time_throttle.rs @@ -1,5 +1,5 @@ #![warn(rust_2018_idioms)] -#![cfg(feature = "full")] +#![cfg(all(feature = "time", feature = "sync", feature = "io-util"))] use tokio::time; use tokio_stream::StreamExt;