test: stabilize traced native socket TaskBlock assertion

Author: Copilot

ddprof-test/src/test/java/com/datadoghq/profiler/wallclock/NativeSocketTaskBlockTest.java

@@ -211,7 +211,7 @@ public void tracedBlockingSocketReadDoesNotEmitTaskBlock() throws Exception {
 
         stopProfiler();
         assertFalse(
-                verifyEvents("datadog.TaskBlock", false).hasItems(),
+                TaskBlockAssertions.containsSpan(verifyEvents("datadog.TaskBlock", false), 0x5101L),
                 "Traced socket I/O must keep MethodSample wall-clock data instead of emitting TaskBlock");
     }
 

ddprof-test/src/test/java/com/datadoghq/profiler/wallclock/TaskBlockAssertions.java

@@ -39,6 +39,22 @@ static void assertContainsObservedState(IItemCollection events, String observedS
                 "Expected TaskBlock with observedBlockingState=" + observedState);
     }
 
+    static boolean containsSpan(IItemCollection events, long spanId) {
+        for (IItemIterable iterable : events) {
+            IMemberAccessor<IQuantity, IItem> spanAccessor =
+                    AbstractProfilerTest.SPAN_ID.getAccessor(iterable.getType());
+            if (spanAccessor == null) {
+                continue;
+            }
+            for (IItem item : iterable) {
+                if (spanAccessor.getMember(item).longValue() == spanId) {
+                    return true;
+                }
+            }
+        }
+        return false;
+    }
+
     private static boolean contains(
             IItemCollection events, long rootSpanId, long spanId, long blocker, long unblockingSpanId) {
         for (IItemIterable iterable : events) {