diff options
Diffstat (limited to 'tokio-sync/tests/fuzz_mpsc.rs')
-rw-r--r-- | tokio-sync/tests/fuzz_mpsc.rs | 7 |
1 files changed, 4 insertions, 3 deletions
diff --git a/tokio-sync/tests/fuzz_mpsc.rs b/tokio-sync/tests/fuzz_mpsc.rs index 2a7b7552..4a0b6441 100644 --- a/tokio-sync/tests/fuzz_mpsc.rs +++ b/tokio-sync/tests/fuzz_mpsc.rs @@ -17,7 +17,8 @@ mod mpsc; #[allow(warnings)] mod semaphore; -use futures::{future::poll_fn, Stream}; +// use futures::{future::poll_fn, Stream}; +use async_util::future::poll_fn; use loom::futures::block_on; use loom::thread; @@ -31,10 +32,10 @@ fn closing_tx() { drop(tx); }); - let v = block_on(poll_fn(|| rx.poll())).unwrap(); + let v = block_on(poll_fn(|cx| rx.poll_next(cx))); assert!(v.is_some()); - let v = block_on(poll_fn(|| rx.poll())).unwrap(); + let v = block_on(poll_fn(|cx| rx.poll_next(cx))); assert!(v.is_none()); }); } |