@@ -7,6 +7,7 @@ use std::{
7
7
use anyhow:: anyhow;
8
8
use futures_util:: future:: FutureExt ;
9
9
use iroh:: SecretKey ;
10
+ use tracing_test:: traced_test;
10
11
11
12
use super :: * ;
12
13
use crate :: {
@@ -48,8 +49,6 @@ impl Downloader {
48
49
let lp = LocalPool :: default ( ) ;
49
50
lp. spawn_detached ( move || async move {
50
51
// we want to see the logs of the service
51
- let _guard = iroh_test:: logging:: setup ( ) ;
52
-
53
52
let service = Service :: new ( getter, dialer, concurrency_limits, retry_config, msg_rx) ;
54
53
service. run ( ) . await
55
54
} ) ;
@@ -66,8 +65,8 @@ impl Downloader {
66
65
67
66
/// Tests that receiving a download request and performing it doesn't explode.
68
67
#[ tokio:: test]
68
+ #[ traced_test]
69
69
async fn smoke_test ( ) {
70
- let _guard = iroh_test:: logging:: setup ( ) ;
71
70
let dialer = dialer:: TestingDialer :: default ( ) ;
72
71
let getter = getter:: TestingGetter :: default ( ) ;
73
72
let concurrency_limits = ConcurrencyLimits :: default ( ) ;
@@ -90,8 +89,8 @@ async fn smoke_test() {
90
89
91
90
/// Tests that multiple intents produce a single request.
92
91
#[ tokio:: test]
92
+ #[ traced_test]
93
93
async fn deduplication ( ) {
94
- let _guard = iroh_test:: logging:: setup ( ) ;
95
94
let dialer = dialer:: TestingDialer :: default ( ) ;
96
95
let getter = getter:: TestingGetter :: default ( ) ;
97
96
// make request take some time to ensure the intents are received before completion
@@ -123,8 +122,8 @@ async fn deduplication() {
123
122
/// Tests that the request is cancelled only when all intents are cancelled.
124
123
#[ ignore = "flaky" ]
125
124
#[ tokio:: test]
125
+ #[ traced_test]
126
126
async fn cancellation ( ) {
127
- let _guard = iroh_test:: logging:: setup ( ) ;
128
127
let dialer = dialer:: TestingDialer :: default ( ) ;
129
128
let getter = getter:: TestingGetter :: default ( ) ;
130
129
// make request take some time to ensure cancellations are received on time
@@ -159,8 +158,8 @@ async fn cancellation() {
159
158
/// maximum number of concurrent requests is not exceed.
160
159
/// NOTE: This is internally tested by [`Service::check_invariants`].
161
160
#[ tokio:: test]
161
+ #[ traced_test]
162
162
async fn max_concurrent_requests_total ( ) {
163
- let _guard = iroh_test:: logging:: setup ( ) ;
164
163
let dialer = dialer:: TestingDialer :: default ( ) ;
165
164
let getter = getter:: TestingGetter :: default ( ) ;
166
165
// make request take some time to ensure concurreny limits are hit
@@ -202,8 +201,8 @@ async fn max_concurrent_requests_total() {
202
201
/// the maximum number of requests per peer is still respected.
203
202
/// NOTE: This is internally tested by [`Service::check_invariants`].
204
203
#[ tokio:: test]
204
+ #[ traced_test]
205
205
async fn max_concurrent_requests_per_peer ( ) {
206
- let _guard = iroh_test:: logging:: setup ( ) ;
207
206
let dialer = dialer:: TestingDialer :: default ( ) ;
208
207
let getter = getter:: TestingGetter :: default ( ) ;
209
208
// make request take some time to ensure concurreny limits are hit
@@ -239,8 +238,8 @@ async fn max_concurrent_requests_per_peer() {
239
238
/// state. The download then finishes, and we make sure that all events are emitted properly, and
240
239
/// the progress state of the handles converges.
241
240
#[ tokio:: test]
241
+ #[ traced_test]
242
242
async fn concurrent_progress ( ) {
243
- let _guard = iroh_test:: logging:: setup ( ) ;
244
243
let dialer = dialer:: TestingDialer :: default ( ) ;
245
244
let getter = getter:: TestingGetter :: default ( ) ;
246
245
@@ -356,8 +355,8 @@ async fn concurrent_progress() {
356
355
}
357
356
358
357
#[ tokio:: test]
358
+ #[ traced_test]
359
359
async fn long_queue ( ) {
360
- let _guard = iroh_test:: logging:: setup ( ) ;
361
360
let dialer = dialer:: TestingDialer :: default ( ) ;
362
361
let getter = getter:: TestingGetter :: default ( ) ;
363
362
let concurrency_limits = ConcurrencyLimits {
@@ -394,8 +393,8 @@ async fn long_queue() {
394
393
/// If a download errors with [`FailureAction::DropPeer`], make sure that the peer is not dropped
395
394
/// while other transfers are still running.
396
395
#[ tokio:: test]
396
+ #[ traced_test]
397
397
async fn fail_while_running ( ) {
398
- let _guard = iroh_test:: logging:: setup ( ) ;
399
398
let dialer = dialer:: TestingDialer :: default ( ) ;
400
399
let getter = getter:: TestingGetter :: default ( ) ;
401
400
let ( downloader, _lp) =
@@ -432,8 +431,8 @@ async fn fail_while_running() {
432
431
}
433
432
434
433
#[ tokio:: test]
434
+ #[ traced_test]
435
435
async fn retry_nodes_simple ( ) {
436
- let _guard = iroh_test:: logging:: setup ( ) ;
437
436
let dialer = dialer:: TestingDialer :: default ( ) ;
438
437
let getter = getter:: TestingGetter :: default ( ) ;
439
438
let ( downloader, _lp) =
@@ -453,8 +452,8 @@ async fn retry_nodes_simple() {
453
452
}
454
453
455
454
#[ tokio:: test]
455
+ #[ traced_test]
456
456
async fn retry_nodes_fail ( ) {
457
- let _guard = iroh_test:: logging:: setup ( ) ;
458
457
let dialer = dialer:: TestingDialer :: default ( ) ;
459
458
let getter = getter:: TestingGetter :: default ( ) ;
460
459
let config = RetryConfig {
@@ -491,8 +490,8 @@ async fn retry_nodes_fail() {
491
490
}
492
491
493
492
#[ tokio:: test]
493
+ #[ traced_test]
494
494
async fn retry_nodes_jump_queue ( ) {
495
- let _guard = iroh_test:: logging:: setup ( ) ;
496
495
let dialer = dialer:: TestingDialer :: default ( ) ;
497
496
let getter = getter:: TestingGetter :: default ( ) ;
498
497
let concurrency_limits = ConcurrencyLimits {
0 commit comments