aboutsummaryrefslogtreecommitdiff
path: root/core
diff options
context:
space:
mode:
Diffstat (limited to 'core')
-rw-r--r--core/src/async_utils/condwait.rs43
1 files changed, 40 insertions, 3 deletions
diff --git a/core/src/async_utils/condwait.rs b/core/src/async_utils/condwait.rs
index f16a99e..e31fac3 100644
--- a/core/src/async_utils/condwait.rs
+++ b/core/src/async_utils/condwait.rs
@@ -77,20 +77,57 @@ impl Default for CondWait {
#[cfg(test)]
mod tests {
use super::*;
- use std::sync::Arc;
+ use std::sync::{
+ atomic::{AtomicUsize, Ordering},
+ Arc,
+ };
#[test]
fn test_cond_wait() {
smol::block_on(async {
let cond_wait = Arc::new(CondWait::new());
+ let count = Arc::new(AtomicUsize::new(0));
+
let cond_wait_cloned = cond_wait.clone();
+ let count_cloned = count.clone();
let task = smol::spawn(async move {
cond_wait_cloned.wait().await;
- true
+ count_cloned.fetch_add(1, Ordering::Relaxed);
+ // do something
});
+ // Send a signal to the waiting task
cond_wait.signal().await;
- assert!(task.await);
+
+ task.await;
+
+ // Reset the boolean flag
+ cond_wait.reset().await;
+
+ assert_eq!(count.load(Ordering::Relaxed), 1);
+
+ let cond_wait_cloned = cond_wait.clone();
+ let count_cloned = count.clone();
+ let task1 = smol::spawn(async move {
+ cond_wait_cloned.wait().await;
+ count_cloned.fetch_add(1, Ordering::Relaxed);
+ // do something
+ });
+
+ let cond_wait_cloned = cond_wait.clone();
+ let count_cloned = count.clone();
+ let task2 = smol::spawn(async move {
+ cond_wait_cloned.wait().await;
+ count_cloned.fetch_add(1, Ordering::Relaxed);
+ // do something
+ });
+
+ // Broadcast a signal to all waiting tasks
+ cond_wait.broadcast().await;
+
+ task1.await;
+ task2.await;
+ assert_eq!(count.load(Ordering::Relaxed), 3);
});
}
}