mirror of
https://github.com/tokio-rs/tokio.git
synced 2025-09-28 12:10:37 +00:00
06c473e628
A first pass at updating Tokio to use `std::future`. Implementations of `Future` from the futures crate are updated to implement `Future` from std. Implementations of `Stream` are moved to a feature flag. This commits disables a number of crates that have not yet been updated.
51 lines
1.2 KiB
Rust
51 lines
1.2 KiB
Rust
#![deny(warnings, rust_2018_idioms)]
|
|
|
|
use tokio_sync::lock::Lock;
|
|
use tokio_test::task::MockTask;
|
|
use tokio_test::{assert_pending, assert_ready};
|
|
|
|
#[test]
|
|
fn straight_execution() {
|
|
let mut task = MockTask::new();
|
|
let mut l = Lock::new(100);
|
|
|
|
// We can immediately acquire the lock and take the value
|
|
task.enter(|cx| {
|
|
let mut g = assert_ready!(l.poll_lock(cx));
|
|
assert_eq!(&*g, &100);
|
|
*g = 99;
|
|
drop(g);
|
|
|
|
let mut g = assert_ready!(l.poll_lock(cx));
|
|
assert_eq!(&*g, &99);
|
|
*g = 98;
|
|
drop(g);
|
|
|
|
let mut g = assert_ready!(l.poll_lock(cx));
|
|
assert_eq!(&*g, &98);
|
|
|
|
// We can continue to access the guard even if the lock is dropped
|
|
drop(l);
|
|
*g = 97;
|
|
assert_eq!(&*g, &97);
|
|
});
|
|
}
|
|
|
|
#[test]
|
|
fn readiness() {
|
|
let mut t1 = MockTask::new();
|
|
let mut t2 = MockTask::new();
|
|
|
|
let mut l = Lock::new(100);
|
|
|
|
let g = assert_ready!(t1.enter(|cx| l.poll_lock(cx)));
|
|
|
|
// We can't now acquire the lease since it's already held in g
|
|
assert_pending!(t2.enter(|cx| l.poll_lock(cx)));
|
|
|
|
// But once g unlocks, we can acquire it
|
|
drop(g);
|
|
assert!(t2.is_woken());
|
|
assert_ready!(t2.enter(|cx| l.poll_lock(cx)));
|
|
}
|