summaryrefslogtreecommitdiffstats
path: root/tokio-sync/tests/fuzz_mpsc.rs
diff options
context:
space:
mode:
Diffstat (limited to 'tokio-sync/tests/fuzz_mpsc.rs')
-rw-r--r--tokio-sync/tests/fuzz_mpsc.rs7
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());
});
}