@@ -22,7 +22,7 @@ use iroh_docs::{
2222} ;
2323use rand:: { CryptoRng , Rng , SeedableRng } ;
2424use tracing:: { debug, error_span, info, Instrument } ;
25- use tracing_subscriber :: { prelude :: * , EnvFilter } ;
25+ use tracing_test :: traced_test ;
2626mod util;
2727use util:: { Builder , Node } ;
2828
@@ -73,8 +73,8 @@ macro_rules! match_event {
7373
7474/// This tests the simplest scenario: A node connects to another node, and performs sync.
7575#[ tokio:: test]
76+ #[ traced_test]
7677async fn sync_simple ( ) -> Result < ( ) > {
77- setup_logging ( ) ;
7878 let mut rng = test_rng ( b"sync_simple" ) ;
7979 let nodes = spawn_nodes ( 2 , & mut rng) . await ?;
8080 let clients = nodes. iter ( ) . map ( |node| node. client ( ) ) . collect :: < Vec < _ > > ( ) ;
@@ -133,9 +133,9 @@ async fn sync_simple() -> Result<()> {
133133
134134/// Test subscribing to replica events (without sync)
135135#[ tokio:: test]
136+ #[ traced_test]
136137async fn sync_subscribe_no_sync ( ) -> Result < ( ) > {
137138 let mut rng = test_rng ( b"sync_subscribe" ) ;
138- setup_logging ( ) ;
139139 let node = spawn_node ( 0 , & mut rng) . await ?;
140140 let client = node. client ( ) ;
141141 let doc = client. docs ( ) . create ( ) . await ?;
@@ -152,12 +152,12 @@ async fn sync_subscribe_no_sync() -> Result<()> {
152152}
153153
154154#[ tokio:: test]
155+ #[ traced_test]
155156async fn sync_gossip_bulk ( ) -> Result < ( ) > {
156157 let n_entries: usize = std:: env:: var ( "N_ENTRIES" )
157158 . map ( |x| x. parse ( ) . expect ( "N_ENTRIES must be a number" ) )
158159 . unwrap_or ( 100 ) ;
159160 let mut rng = test_rng ( b"sync_gossip_bulk" ) ;
160- setup_logging ( ) ;
161161
162162 let nodes = spawn_nodes ( 2 , & mut rng) . await ?;
163163 let clients = nodes. iter ( ) . map ( |node| node. client ( ) ) . collect :: < Vec < _ > > ( ) ;
@@ -242,10 +242,10 @@ async fn sync_gossip_bulk() -> Result<()> {
242242
243243/// This tests basic sync and gossip with 3 peers.
244244#[ tokio:: test]
245+ #[ traced_test]
245246#[ ignore = "flaky" ]
246247async fn sync_full_basic ( ) -> testresult:: TestResult < ( ) > {
247248 let mut rng = test_rng ( b"sync_full_basic" ) ;
248- setup_logging ( ) ;
249249 let mut nodes = spawn_nodes ( 2 , & mut rng) . await ?;
250250 let mut clients = nodes
251251 . iter ( )
@@ -424,9 +424,9 @@ async fn sync_full_basic() -> testresult::TestResult<()> {
424424}
425425
426426#[ tokio:: test]
427+ #[ traced_test]
427428async fn sync_open_close ( ) -> Result < ( ) > {
428429 let mut rng = test_rng ( b"sync_subscribe_stop_close" ) ;
429- setup_logging ( ) ;
430430 let node = spawn_node ( 0 , & mut rng) . await ?;
431431 let client = node. client ( ) ;
432432
@@ -448,9 +448,9 @@ async fn sync_open_close() -> Result<()> {
448448}
449449
450450#[ tokio:: test]
451+ #[ traced_test]
451452async fn sync_subscribe_stop_close ( ) -> Result < ( ) > {
452453 let mut rng = test_rng ( b"sync_subscribe_stop_close" ) ;
453- setup_logging ( ) ;
454454 let node = spawn_node ( 0 , & mut rng) . await ?;
455455 let client = node. client ( ) ;
456456
@@ -487,9 +487,9 @@ async fn sync_subscribe_stop_close() -> Result<()> {
487487}
488488
489489#[ tokio:: test]
490+ #[ traced_test]
490491#[ cfg( feature = "test-utils" ) ]
491492async fn test_sync_via_relay ( ) -> Result < ( ) > {
492- let _guard = iroh_test:: logging:: setup ( ) ;
493493 let ( relay_map, _relay_url, _guard) = iroh:: test_utils:: run_relay_server ( ) . await ?;
494494
495495 let node1 = Node :: memory ( )
@@ -581,11 +581,11 @@ async fn test_sync_via_relay() -> Result<()> {
581581}
582582
583583#[ tokio:: test]
584+ #[ traced_test]
584585#[ cfg( feature = "test-utils" ) ]
585586#[ ignore = "flaky" ]
586587async fn sync_restart_node ( ) -> Result < ( ) > {
587588 let mut rng = test_rng ( b"sync_restart_node" ) ;
588- setup_logging ( ) ;
589589 let ( relay_map, _relay_url, _guard) = iroh:: test_utils:: run_relay_server ( ) . await ?;
590590
591591 let discovery_server = iroh:: test_utils:: DnsPkarrServer :: run ( ) . await ?;
@@ -752,7 +752,6 @@ async fn test_download_policies() -> Result<()> {
752752 const EXPECTED_B_DOWNLOADED : usize = 3 ;
753753
754754 let mut rng = test_rng ( b"sync_download_policies" ) ;
755- setup_logging ( ) ;
756755 let nodes = spawn_nodes ( 2 , & mut rng) . await ?;
757756 let clients = nodes. iter ( ) . map ( |node| node. client ( ) ) . collect :: < Vec < _ > > ( ) ;
758757
@@ -860,9 +859,9 @@ async fn test_download_policies() -> Result<()> {
860859
861860/// Test sync between many nodes with propagation through sync reports.
862861#[ tokio:: test( flavor = "multi_thread" ) ]
862+ #[ traced_test]
863863#[ ignore = "flaky" ]
864864async fn sync_big ( ) -> Result < ( ) > {
865- setup_logging ( ) ;
866865 let mut rng = test_rng ( b"sync_big" ) ;
867866 let n_nodes = std:: env:: var ( "NODES" )
868867 . map ( |v| v. parse ( ) . expect ( "NODES must be a number" ) )
@@ -982,6 +981,7 @@ async fn sync_big() -> Result<()> {
982981}
983982
984983#[ tokio:: test]
984+ #[ traced_test]
985985#[ cfg( feature = "test-utils" ) ]
986986async fn test_list_docs_stream ( ) -> testresult:: TestResult < ( ) > {
987987 let node = Node :: memory ( )
@@ -1156,6 +1156,7 @@ impl PartialEq<ExpectedEntry> for (Entry, Bytes) {
11561156}
11571157
11581158#[ tokio:: test]
1159+ #[ traced_test]
11591160async fn doc_delete ( ) -> Result < ( ) > {
11601161 let node = Node :: memory ( )
11611162 . gc_interval ( Some ( Duration :: from_millis ( 100 ) ) )
@@ -1185,9 +1186,9 @@ async fn doc_delete() -> Result<()> {
11851186}
11861187
11871188#[ tokio:: test]
1189+ #[ traced_test]
11881190async fn sync_drop_doc ( ) -> Result < ( ) > {
11891191 let mut rng = test_rng ( b"sync_drop_doc" ) ;
1190- setup_logging ( ) ;
11911192 let node = spawn_node ( 0 , & mut rng) . await ?;
11921193 let client = node. client ( ) ;
11931194
@@ -1241,14 +1242,6 @@ async fn get_latest(
12411242 Ok ( content. to_vec ( ) )
12421243}
12431244
1244- fn setup_logging ( ) {
1245- tracing_subscriber:: registry ( )
1246- . with ( tracing_subscriber:: fmt:: layer ( ) . with_writer ( std:: io:: stderr) )
1247- . with ( EnvFilter :: from_default_env ( ) )
1248- . try_init ( )
1249- . ok ( ) ;
1250- }
1251-
12521245async fn next < T : std:: fmt:: Debug > ( mut stream : impl Stream < Item = Result < T > > + Unpin ) -> T {
12531246 let event = stream
12541247 . next ( )
0 commit comments