Merge pull request #890 from akoshelev/role-in-trace

Print helper role inside `TestWorld` spans
private-attribution · Dec 11, 2023 · dc7fcf7 · dc7fcf7
2 parents bc1b3f0 + 3e9f55c
commit dc7fcf7
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/ipa-core/src/test_fixture/world.rs b/ipa-core/src/test_fixture/world.rs
@@ -193,10 +193,12 @@ impl TestWorld {
     {
         let input_shares = input.share_with(&mut thread_rng());
         #[allow(clippy::disallowed_methods)] // It's just 3 items.
-        let output =
-            join_all(zip(contexts, input_shares).map(|(ctx, shares)| helper_fn(ctx, shares)))
-                .instrument(span)
-                .await;
+        let output = join_all(zip(contexts, input_shares).map(|(ctx, shares)| {
+            let role = ctx.role();
+            helper_fn(ctx, shares).instrument(tracing::trace_span!("", role = ?role))
+        }))
+        .instrument(span)
+        .await;
         <[_; 3]>::try_from(output).unwrap()
     }
 }