mirror of
https://github.com/tokio-rs/tokio.git
synced 2026-08-18 00:00:09 +02:00
Jon Gjengset
e3415d8d61
This renames `Lock` to `Mutex`, and brings the API more in line with `std::sync::Mutex`. In partcular, locking now only takes `&self`, with the expectation that you place the `Mutex` in an `Arc` (or something similar) to share it between threads. Fixes #1544. Part of #1210.
80 lines
1.8 KiB
Rust
80 lines
1.8 KiB
Rust
#![warn(rust_2018_idioms)]
|
|
|
|
use std::sync::Arc;
|
|
use tokio_sync::Mutex;
|
|
use tokio_test::task::spawn;
|
|
use tokio_test::{assert_pending, assert_ready};
|
|
|
|
#[test]
|
|
fn straight_execution() {
|
|
let l = Mutex::new(100);
|
|
|
|
{
|
|
let mut t = spawn(l.lock());
|
|
let mut g = assert_ready!(t.poll());
|
|
assert_eq!(&*g, &100);
|
|
*g = 99;
|
|
}
|
|
{
|
|
let mut t = spawn(l.lock());
|
|
let mut g = assert_ready!(t.poll());
|
|
assert_eq!(&*g, &99);
|
|
*g = 98;
|
|
}
|
|
{
|
|
let mut t = spawn(l.lock());
|
|
let g = assert_ready!(t.poll());
|
|
assert_eq!(&*g, &98);
|
|
}
|
|
}
|
|
|
|
#[test]
|
|
fn readiness() {
|
|
let l1 = Arc::new(Mutex::new(100));
|
|
let l2 = Arc::clone(&l1);
|
|
let mut t1 = spawn(l1.lock());
|
|
let mut t2 = spawn(l2.lock());
|
|
|
|
let g = assert_ready!(t1.poll());
|
|
|
|
// We can't now acquire the lease since it's already held in g
|
|
assert_pending!(t2.poll());
|
|
|
|
// But once g unlocks, we can acquire it
|
|
drop(g);
|
|
assert!(t2.is_woken());
|
|
assert_ready!(t2.poll());
|
|
}
|
|
|
|
/*
|
|
#[test]
|
|
#[ignore]
|
|
fn lock() {
|
|
let mut lock = Mutex::new(false);
|
|
|
|
let mut lock2 = lock.clone();
|
|
std::thread::spawn(move || {
|
|
let l = lock2.lock();
|
|
pin_mut!(l);
|
|
|
|
let mut task = MockTask::new();
|
|
let mut g = assert_ready!(task.poll(&mut l));
|
|
std::thread::sleep(std::time::Duration::from_millis(500));
|
|
*g = true;
|
|
drop(g);
|
|
});
|
|
|
|
std::thread::sleep(std::time::Duration::from_millis(50));
|
|
let mut task = MockTask::new();
|
|
let l = lock.lock();
|
|
pin_mut!(l);
|
|
|
|
assert_pending!(task.poll(&mut l));
|
|
|
|
std::thread::sleep(std::time::Duration::from_millis(500));
|
|
assert!(task.is_woken());
|
|
let result = assert_ready!(task.poll(&mut l));
|
|
assert!(*result);
|
|
}
|
|
*/
|