diff --git a/inferred-spans/src/test/java/io/opentelemetry/contrib/inferredspans/internal/SamplingProfilerTest.java b/inferred-spans/src/test/java/io/opentelemetry/contrib/inferredspans/internal/SamplingProfilerTest.java index 1ec74d6b18..6125cbdf01 100644 --- a/inferred-spans/src/test/java/io/opentelemetry/contrib/inferredspans/internal/SamplingProfilerTest.java +++ b/inferred-spans/src/test/java/io/opentelemetry/contrib/inferredspans/internal/SamplingProfilerTest.java @@ -182,7 +182,11 @@ void testProfileTransaction() throws Exception { .timeout(Duration.ofSeconds(10)) .until(() -> setup.profiler.getProfilingSessions() > currentSession); profilingActiveOnThread = setup.profiler.isProfilingActiveOnThread(Thread.currentThread()); + int sessionBeforeInferred = setup.profiler.getProfilingSessions(); aInferred(tracer); + await() + .timeout(Duration.ofSeconds(10)) + .until(() -> setup.profiler.getProfilingSessions() > sessionBeforeInferred); } finally { tx.end(); }