chore: Fix tests logging setup (#5693)

Signed-off-by: Graham King <grahamk@nvidia.com>

lib/llm/src/kv_router/publisher.rs

@@ -1890,7 +1890,6 @@ mod test_integration_publisher {
     use crate::kv_router::protocols::ActiveLoad;
     use dynamo_runtime::distributed_test_utils::create_test_drt_async;
     use dynamo_runtime::transports::event_plane::EventSubscriber;
-    use futures::StreamExt;
 
     #[tokio::test]
     #[ignore] // Mark as ignored as requested, because CI's integrations still don't have NATS

lib/runtime/src/pipeline/network/ingress/shared_tcp_endpoint.rs

@@ -673,10 +673,7 @@ mod tests {
     #[tokio::test]
     async fn test_graceful_shutdown_waits_for_inflight_tcp_requests() {
         // Initialize tracing for test debugging
-        let _ = tracing_subscriber::fmt()
-            .with_test_writer()
-            .with_max_level(tracing::Level::DEBUG)
-            .try_init();
+        crate::logging::init();
 
         let cancellation_token = CancellationToken::new();
         let bind_addr: SocketAddr = "127.0.0.1:0".parse().unwrap();
@@ -877,10 +874,7 @@ mod tests {
 
     #[tokio::test]
     async fn test_worker_pool_bounds_concurrency() {
-        let _ = tracing_subscriber::fmt()
-            .with_test_writer()
-            .with_max_level(tracing::Level::DEBUG)
-            .try_init();
+        crate::logging::init();
 
         // Use a small pool size for testing
         let pool_size = 3;