diff --git a/spring-cloud-sleuth-core/src/test/java/org/springframework/cloud/sleuth/instrument/web/client/integration/WebClientTests.java b/spring-cloud-sleuth-core/src/test/java/org/springframework/cloud/sleuth/instrument/web/client/integration/WebClientTests.java index a856079c7..df95285da 100644 --- a/spring-cloud-sleuth-core/src/test/java/org/springframework/cloud/sleuth/instrument/web/client/integration/WebClientTests.java +++ b/spring-cloud-sleuth-core/src/test/java/org/springframework/cloud/sleuth/instrument/web/client/integration/WebClientTests.java @@ -392,11 +392,13 @@ public class WebClientTests { span.finish(); } - System.out.println("Found spans " + this.reporter.getSpans()); - final Optional clientSpan = this.reporter.getSpans().stream() - .filter(s -> s.kind() == zipkin2.Span.Kind.CLIENT).findFirst(); - then(clientSpan).isPresent(); - then(clientSpan.get().tags()).containsEntry("error", "CANCELLED"); + Awaitility.await().untilAsserted(() -> { + System.out.println("Found spans " + this.reporter.getSpans()); + final Optional clientSpan = this.reporter.getSpans().stream() + .filter(s -> s.kind() == zipkin2.Span.Kind.CLIENT).findFirst(); + then(clientSpan).isPresent(); + then(clientSpan.get().tags()).containsEntry("error", "CANCELLED"); + }); } Object[] parametersForShouldAttachTraceIdWhenCallingAnotherService() {