use crate::sync::rwlock::*;
use loom::future::block_on;
use loom::thread;
use std::sync::Arc;
#[test]
fn concurrent_write() {
let mut b = loom::model::Builder::new();
b.check(|| {
let rwlock = Arc::new(RwLock::<u32>::new(0));
let rwclone = rwlock.clone();
let t1 = thread::spawn(move || {
block_on(async {
let mut guard = rwclone.write().await;
*guard += 5;
});
});
let rwclone = rwlock.clone();
let t2 = thread::spawn(move || {
block_on(async {
let mut guard = rwclone.write().await;
*guard += 5;
});
});
t1.join().expect("thread 1 write should not panic");
t2.join().expect("thread 2 write should not panic");
let guard = block_on(rwlock.read());
assert_eq!(10, *guard);
});
}
#[test]
fn concurrent_read_write() {
let mut b = loom::model::Builder::new();
b.check(|| {
let rwlock = Arc::new(RwLock::<u32>::new(0));
let rwclone = rwlock.clone();
let t1 = thread::spawn(move || {
block_on(async {
let mut guard = rwclone.write().await;
*guard += 5;
});
});
let rwclone = rwlock.clone();
let t2 = thread::spawn(move || {
block_on(async {
let mut guard = rwclone.write().await;
*guard += 5;
});
});
let rwclone = rwlock.clone();
let t3 = thread::spawn(move || {
block_on(async {
let guard = rwclone.read().await;
assert!(*guard == 0 || *guard == 5 || *guard == 10);
});
});
t1.join().expect("thread 1 write should not panic");
t2.join().expect("thread 2 write should not panic");
t3.join().expect("thread 3 read should not panic");
let guard = block_on(rwlock.read());
assert_eq!(10, *guard);
});
}