v0.1.2-pre.002-fix

crates/ksp-logging-lib/tests/callsite.rs

@@ -1,5 +1,7 @@
 // file: crates/ksp-logging-lib/tests/callsite.rs
-// version: 1
+// version: 2
+
+//! Integration tests for KSP logging callsite and async span instrumentation behavior.
 
 const TEST_TARGET: &str = "ksp-logging-lib";
 
@@ -137,7 +139,7 @@ fn span_macro_preserves_consumer_callsite() {
 }
 
 #[test]
-fn async_instrumentation_enters_and_exits_span_during_poll() {
+fn async_instrumentation_enters_and_exits_span_during_poll_and_drop() {
     let captured = std::sync::Arc::new(std::sync::Mutex::new(std::vec::Vec::new()));
     let enters = std::sync::Arc::new(std::sync::atomic::AtomicU64::new(0));
     let exits = std::sync::Arc::new(std::sync::atomic::AtomicU64::new(0));
@@ -150,8 +152,12 @@ fn async_instrumentation_enters_and_exits_span_during_poll() {
         let mut context = std::task::Context::from_waker(waker);
         let poll = std::future::Future::poll(future.as_mut(), &mut context);
         assert_eq!(poll, std::task::Poll::Ready(42_u32));
+        assert_eq!(enters.load(std::sync::atomic::Ordering::Relaxed), 1);
+        assert_eq!(exits.load(std::sync::atomic::Ordering::Relaxed), 1);
+        std::mem::drop(future);
+        assert_eq!(enters.load(std::sync::atomic::Ordering::Relaxed), 2);
+        assert_eq!(exits.load(std::sync::atomic::Ordering::Relaxed), 2);
         return;
     });
-    assert_eq!(enters.load(std::sync::atomic::Ordering::Relaxed), 1);
-    assert_eq!(exits.load(std::sync::atomic::Ordering::Relaxed), 1);
+    assert_eq!(enters.load(std::sync::atomic::Ordering::Relaxed), exits.load(std::sync::atomic::Ordering::Relaxed));
 }

crates/ksp-logging-lib/tests/public_api.rs

@@ -1,5 +1,7 @@
 // file: crates/ksp-logging-lib/tests/public_api.rs
-// version: 1
+// version: 2
+
+//! Integration tests for the public crate-root surface of `ksp-logging-lib`.
 
 const TEST_TARGET: &str = "ksp-logging-lib";