fix: change where TaskHandle is borrowed to prevent repeat output fetches
This commit is contained in:
1 parent
c8efb00175
commit
9fbe4c2fd8
47 files changed
+323
-513
No files matched your search
@@ -11,13 +11,13 @@ async fn test_two_dependencies() -> DagResult<()> {
|
||||
let b = dag.add_task(task_fn::<(), _, _>(|_: ()| 20));
|
||||
let sum = dag
|
||||
.add_task(task_fn::<(i32, i32), _, _>(|(x, y): (&i32, &i32)| x + y))
|
||||
.depends_on((a, b));
|
||||
.depends_on((&a, &b));
|
||||
|
||||
let mut output = dag
|
||||
.run(|fut| async move { tokio::spawn(fut).await.unwrap() })
|
||||
.await?;
|
||||
|
||||
assert_eq!(output.get(sum)?, 30);
|
||||
assert_eq!(output.get(sum), 30);
|
||||
Ok(())
|
||||
}
|
||||
|
||||
@@ -32,13 +32,13 @@ async fn test_three_dependencies() -> DagResult<()> {
|
||||
.add_task(task_fn::<(i32, i32, i32), _, _>(
|
||||
|(x, y, z): (&i32, &i32, &i32)| x + y + z,
|
||||
))
|
||||
.depends_on((a, b, c));
|
||||
.depends_on((&a, &b, &c));
|
||||
|
||||
let mut output = dag
|
||||
.run(|fut| async move { tokio::spawn(fut).await.unwrap() })
|
||||
.await?;
|
||||
|
||||
assert_eq!(output.get(sum)?, 6);
|
||||
assert_eq!(output.get(sum), 6);
|
||||
Ok(())
|
||||
}
|
||||
|
||||
@@ -54,13 +54,13 @@ async fn test_four_dependencies() -> DagResult<()> {
|
||||
.add_task(task_fn::<(i32, i32, i32, i32), _, _>(
|
||||
|(a, b, c, d): (&i32, &i32, &i32, &i32)| a + b + c + d,
|
||||
))
|
||||
.depends_on((deps[0], deps[1], deps[2], deps[3]));
|
||||
.depends_on((&deps[0], &deps[1], &deps[2], &deps[3]));
|
||||
|
||||
let mut output = dag
|
||||
.run(|fut| async move { tokio::spawn(fut).await.unwrap() })
|
||||
.await?;
|
||||
|
||||
assert_eq!(output.get(sum)?, 10); // 1+2+3+4
|
||||
assert_eq!(output.get(sum), 10); // 1+2+3+4
|
||||
Ok(())
|
||||
}
|
||||
|
||||
@@ -76,13 +76,13 @@ async fn test_five_dependencies() -> DagResult<()> {
|
||||
.add_task(task_fn::<(i32, i32, i32, i32, i32), _, _>(
|
||||
|(a, b, c, d, e): (&i32, &i32, &i32, &i32, &i32)| a + b + c + d + e,
|
||||
))
|
||||
.depends_on((deps[0], deps[1], deps[2], deps[3], deps[4]));
|
||||
.depends_on((&deps[0], &deps[1], &deps[2], &deps[3], &deps[4]));
|
||||
|
||||
let mut output = dag
|
||||
.run(|fut| async move { tokio::spawn(fut).await.unwrap() })
|
||||
.await?;
|
||||
|
||||
assert_eq!(output.get(sum)?, 15); // 1+2+3+4+5
|
||||
assert_eq!(output.get(sum), 15); // 1+2+3+4+5
|
||||
Ok(())
|
||||
}
|
||||
|
||||
@@ -98,13 +98,13 @@ async fn test_six_dependencies() -> DagResult<()> {
|
||||
.add_task(task_fn::<(i32, i32, i32, i32, i32, i32), _, _>(
|
||||
|(a, b, c, d, e, f): (&i32, &i32, &i32, &i32, &i32, &i32)| a + b + c + d + e + f,
|
||||
))
|
||||
.depends_on((deps[0], deps[1], deps[2], deps[3], deps[4], deps[5]));
|
||||
.depends_on((&deps[0], &deps[1], &deps[2], &deps[3], &deps[4], &deps[5]));
|
||||
|
||||
let mut output = dag
|
||||
.run(|fut| async move { tokio::spawn(fut).await.unwrap() })
|
||||
.await?;
|
||||
|
||||
assert_eq!(output.get(sum)?, 21); // 1+2+3+4+5+6
|
||||
assert_eq!(output.get(sum), 21); // 1+2+3+4+5+6
|
||||
Ok(())
|
||||
}
|
||||
|
||||
@@ -123,14 +123,14 @@ async fn test_seven_dependencies() -> DagResult<()> {
|
||||
},
|
||||
))
|
||||
.depends_on((
|
||||
deps[0], deps[1], deps[2], deps[3], deps[4], deps[5], deps[6],
|
||||
&deps[0], &deps[1], &deps[2], &deps[3], &deps[4], &deps[5], &deps[6],
|
||||
));
|
||||
|
||||
let mut output = dag
|
||||
.run(|fut| async move { tokio::spawn(fut).await.unwrap() })
|
||||
.await?;
|
||||
|
||||
assert_eq!(output.get(sum)?, 28); // 1+2+3+4+5+6+7
|
||||
assert_eq!(output.get(sum), 28); // 1+2+3+4+5+6+7
|
||||
Ok(())
|
||||
}
|
||||
|
||||
@@ -149,13 +149,13 @@ async fn test_eight_dependencies() -> DagResult<()> {
|
||||
},
|
||||
))
|
||||
.depends_on((
|
||||
deps[0], deps[1], deps[2], deps[3], deps[4], deps[5], deps[6], deps[7],
|
||||
&deps[0], &deps[1], &deps[2], &deps[3], &deps[4], &deps[5], &deps[6], &deps[7],
|
||||
));
|
||||
|
||||
let mut output = dag
|
||||
.run(|fut| async move { tokio::spawn(fut).await.unwrap() })
|
||||
.await?;
|
||||
|
||||
assert_eq!(output.get(sum)?, 36); // Sum of 1..=8
|
||||
assert_eq!(output.get(sum), 36); // Sum of 1..=8
|
||||
Ok(())
|
||||
}
|
||||
Reference in new issue
Block a user