Improve logging in tests

informalsystems · Jan 10, 2025 · 83c5eed · 83c5eed
1 parent 47f994c
commit 83c5eed
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 3 deletions.
diff --git a/code/crates/engine/src/wal.rs b/code/crates/engine/src/wal.rs
@@ -185,6 +185,11 @@ where
     type Arguments = Args<Codec>;
     type State = State<Ctx>;
 
+    #[tracing::instrument(
+        name = "wal.pre_start",
+        parent = &self.span,
+        skip_all,
+    )]
     async fn pre_start(
         &self,
         _myself: WalRef<Ctx>,
@@ -196,7 +201,7 @@ where
         let (tx, rx) = mpsc::channel(100);
 
         // Spawn a system thread to perform blocking WAL operations.
-        let handle = self::thread::spawn(tracing::Span::current(), log, args.codec, rx);
+        let handle = self::thread::spawn(self.span.clone(), log, args.codec, rx);
 
         Ok(State {
             height: Ctx::Height::default(),
@@ -224,6 +229,12 @@ where
         Ok(())
     }
 
+    #[tracing::instrument(
+        name = "wal.post_stop",
+        parent = &self.span,
+        skip_all,
+        fields(height = %state.height),
+    )]
     async fn post_stop(
         &self,
         _: WalRef<Ctx>,

diff --git a/code/crates/test/framework/src/lib.rs b/code/crates/test/framework/src/lib.rs
@@ -636,9 +636,9 @@ pub fn init_logging(test_module: &str) {
         .any(|(k, v)| std::env::var(k).as_deref() == Ok(v));
 
     let directive = if enable_debug {
-        format!("{test_module}=debug,ractor=error,debug")
+        format!("{test_module}=debug,informalsystems_malachitebft=trace,informalsystems_malachitebft_discovery=error,libp2p=warn,ractor=warn")
     } else {
-        format!("{test_module}=debug,ractor=error,warn")
+        format!("{test_module}=debug,informalsystems_malachitebft=info,informalsystems_malachitebft_discovery=error,libp2p=warn,ractor=warn")
     };
 
     let filter = EnvFilter::builder().parse(directive).unwrap();