71 lines
1.5 KiB
Rust
71 lines
1.5 KiB
Rust
//! Runtime compatibility tests for smol
|
|
|
|
use crate::common::task_fn;
|
|
use dagx::DagRunner;
|
|
|
|
struct Value(i32);
|
|
#[dagx::task]
|
|
impl Value {
|
|
async fn run(&mut self) -> i32 {
|
|
self.0
|
|
}
|
|
}
|
|
|
|
struct Add;
|
|
#[dagx::task]
|
|
impl Add {
|
|
async fn run(&mut self, a: &i32, b: &i32) -> i32 {
|
|
a + b
|
|
}
|
|
}
|
|
|
|
#[test]
|
|
fn test_basic_dag_smol() {
|
|
smol::block_on(async {
|
|
let dag = DagRunner::new();
|
|
|
|
let x = dag.add_task(Value(2));
|
|
let y = dag.add_task(Value(3));
|
|
let sum = dag.add_task(Add).depends_on((x, y));
|
|
|
|
dag.run(smol::spawn).await.unwrap();
|
|
|
|
assert_eq!(dag.get(sum).unwrap(), 5);
|
|
});
|
|
}
|
|
|
|
#[test]
|
|
fn test_parallel_execution_smol() {
|
|
smol::block_on(async {
|
|
let dag = DagRunner::new();
|
|
|
|
let tasks: Vec<_> = (0..10)
|
|
.map(|i| dag.add_task(task_fn::<(), _, _>(move |_: ()| i * 2)))
|
|
.collect();
|
|
|
|
dag.run(smol::spawn).await.unwrap();
|
|
|
|
for (i, task) in tasks.into_iter().enumerate() {
|
|
assert_eq!(dag.get(task).unwrap(), i * 2);
|
|
}
|
|
});
|
|
}
|
|
|
|
#[test]
|
|
fn test_complex_dependencies_smol() {
|
|
smol::block_on(async {
|
|
let dag = DagRunner::new();
|
|
|
|
let a = dag.add_task(Value(10));
|
|
let b = dag.add_task(Value(20));
|
|
let sum = dag.add_task(Add).depends_on((a, b));
|
|
let double = dag
|
|
.add_task(task_fn::<i32, _, _>(|&x: &i32| x * 2))
|
|
.depends_on(sum);
|
|
|
|
dag.run(smol::spawn).await.unwrap();
|
|
|
|
assert_eq!(dag.get(double).unwrap().clone(), 60);
|
|
});
|
|
}
|