@@ -7,6 +7,7 @@ use std::{
77use anyhow:: anyhow;
88use futures_util:: future:: FutureExt ;
99use iroh:: SecretKey ;
10+ use tracing_test:: traced_test;
1011
1112use super :: * ;
1213use crate :: {
@@ -48,8 +49,6 @@ impl Downloader {
4849 let lp = LocalPool :: default ( ) ;
4950 lp. spawn_detached ( move || async move {
5051 // we want to see the logs of the service
51- let _guard = iroh_test:: logging:: setup ( ) ;
52-
5352 let service = Service :: new ( getter, dialer, concurrency_limits, retry_config, msg_rx) ;
5453 service. run ( ) . await
5554 } ) ;
@@ -66,8 +65,8 @@ impl Downloader {
6665
6766/// Tests that receiving a download request and performing it doesn't explode.
6867#[ tokio:: test]
68+ #[ traced_test]
6969async fn smoke_test ( ) {
70- let _guard = iroh_test:: logging:: setup ( ) ;
7170 let dialer = dialer:: TestingDialer :: default ( ) ;
7271 let getter = getter:: TestingGetter :: default ( ) ;
7372 let concurrency_limits = ConcurrencyLimits :: default ( ) ;
@@ -90,8 +89,8 @@ async fn smoke_test() {
9089
9190/// Tests that multiple intents produce a single request.
9291#[ tokio:: test]
92+ #[ traced_test]
9393async fn deduplication ( ) {
94- let _guard = iroh_test:: logging:: setup ( ) ;
9594 let dialer = dialer:: TestingDialer :: default ( ) ;
9695 let getter = getter:: TestingGetter :: default ( ) ;
9796 // make request take some time to ensure the intents are received before completion
@@ -123,8 +122,8 @@ async fn deduplication() {
123122/// Tests that the request is cancelled only when all intents are cancelled.
124123#[ ignore = "flaky" ]
125124#[ tokio:: test]
125+ #[ traced_test]
126126async fn cancellation ( ) {
127- let _guard = iroh_test:: logging:: setup ( ) ;
128127 let dialer = dialer:: TestingDialer :: default ( ) ;
129128 let getter = getter:: TestingGetter :: default ( ) ;
130129 // make request take some time to ensure cancellations are received on time
@@ -159,8 +158,8 @@ async fn cancellation() {
159158/// maximum number of concurrent requests is not exceed.
160159/// NOTE: This is internally tested by [`Service::check_invariants`].
161160#[ tokio:: test]
161+ #[ traced_test]
162162async fn max_concurrent_requests_total ( ) {
163- let _guard = iroh_test:: logging:: setup ( ) ;
164163 let dialer = dialer:: TestingDialer :: default ( ) ;
165164 let getter = getter:: TestingGetter :: default ( ) ;
166165 // make request take some time to ensure concurreny limits are hit
@@ -202,8 +201,8 @@ async fn max_concurrent_requests_total() {
202201/// the maximum number of requests per peer is still respected.
203202/// NOTE: This is internally tested by [`Service::check_invariants`].
204203#[ tokio:: test]
204+ #[ traced_test]
205205async fn max_concurrent_requests_per_peer ( ) {
206- let _guard = iroh_test:: logging:: setup ( ) ;
207206 let dialer = dialer:: TestingDialer :: default ( ) ;
208207 let getter = getter:: TestingGetter :: default ( ) ;
209208 // make request take some time to ensure concurreny limits are hit
@@ -239,8 +238,8 @@ async fn max_concurrent_requests_per_peer() {
239238/// state. The download then finishes, and we make sure that all events are emitted properly, and
240239/// the progress state of the handles converges.
241240#[ tokio:: test]
241+ #[ traced_test]
242242async fn concurrent_progress ( ) {
243- let _guard = iroh_test:: logging:: setup ( ) ;
244243 let dialer = dialer:: TestingDialer :: default ( ) ;
245244 let getter = getter:: TestingGetter :: default ( ) ;
246245
@@ -356,8 +355,8 @@ async fn concurrent_progress() {
356355}
357356
358357#[ tokio:: test]
358+ #[ traced_test]
359359async fn long_queue ( ) {
360- let _guard = iroh_test:: logging:: setup ( ) ;
361360 let dialer = dialer:: TestingDialer :: default ( ) ;
362361 let getter = getter:: TestingGetter :: default ( ) ;
363362 let concurrency_limits = ConcurrencyLimits {
@@ -394,8 +393,8 @@ async fn long_queue() {
394393/// If a download errors with [`FailureAction::DropPeer`], make sure that the peer is not dropped
395394/// while other transfers are still running.
396395#[ tokio:: test]
396+ #[ traced_test]
397397async fn fail_while_running ( ) {
398- let _guard = iroh_test:: logging:: setup ( ) ;
399398 let dialer = dialer:: TestingDialer :: default ( ) ;
400399 let getter = getter:: TestingGetter :: default ( ) ;
401400 let ( downloader, _lp) =
@@ -432,8 +431,8 @@ async fn fail_while_running() {
432431}
433432
434433#[ tokio:: test]
434+ #[ traced_test]
435435async fn retry_nodes_simple ( ) {
436- let _guard = iroh_test:: logging:: setup ( ) ;
437436 let dialer = dialer:: TestingDialer :: default ( ) ;
438437 let getter = getter:: TestingGetter :: default ( ) ;
439438 let ( downloader, _lp) =
@@ -453,8 +452,8 @@ async fn retry_nodes_simple() {
453452}
454453
455454#[ tokio:: test]
455+ #[ traced_test]
456456async fn retry_nodes_fail ( ) {
457- let _guard = iroh_test:: logging:: setup ( ) ;
458457 let dialer = dialer:: TestingDialer :: default ( ) ;
459458 let getter = getter:: TestingGetter :: default ( ) ;
460459 let config = RetryConfig {
@@ -491,8 +490,8 @@ async fn retry_nodes_fail() {
491490}
492491
493492#[ tokio:: test]
493+ #[ traced_test]
494494async fn retry_nodes_jump_queue ( ) {
495- let _guard = iroh_test:: logging:: setup ( ) ;
496495 let dialer = dialer:: TestingDialer :: default ( ) ;
497496 let getter = getter:: TestingGetter :: default ( ) ;
498497 let concurrency_limits = ConcurrencyLimits {
0 commit comments