From 6a91dae5a1699ff2a9d64f1e3735c413e015f72d Mon Sep 17 00:00:00 2001
From: Trask Stalnaker <trask.stalnaker@gmail.com>
Date: Mon, 20 Sep 2021 17:21:55 -0700
Subject: [PATCH] Improve reactor test (#4168)

---
 .../reactor/AbstractSubscriptionTest.groovy    | 18 ++++++++++--------
 1 file changed, 10 insertions(+), 8 deletions(-)

diff --git a/instrumentation/reactor-3.1/testing/src/main/groovy/io/opentelemetry/instrumentation/reactor/AbstractSubscriptionTest.groovy b/instrumentation/reactor-3.1/testing/src/main/groovy/io/opentelemetry/instrumentation/reactor/AbstractSubscriptionTest.groovy
index 4070839fe5..53fdcac945 100644
--- a/instrumentation/reactor-3.1/testing/src/main/groovy/io/opentelemetry/instrumentation/reactor/AbstractSubscriptionTest.groovy
+++ b/instrumentation/reactor-3.1/testing/src/main/groovy/io/opentelemetry/instrumentation/reactor/AbstractSubscriptionTest.groovy
@@ -10,21 +10,24 @@ import io.opentelemetry.api.trace.SpanKind
 import io.opentelemetry.instrumentation.test.InstrumentationSpecification
 import reactor.core.publisher.Mono
 
+import java.time.Duration
 import java.util.concurrent.CountDownLatch
 
 abstract class AbstractSubscriptionTest extends InstrumentationSpecification {
 
   def "subscription test"() {
     when:
+    Mono<Connection> connection = Mono.create {
+      it.success(new Connection())
+    }
     CountDownLatch latch = new CountDownLatch(1)
     runWithSpan("parent") {
-      Mono<Connection> connection = Mono.create {
-        it.success(new Connection())
-      }
-      connection.subscribe {
-        it.query()
-        latch.countDown()
-      }
+      connection
+        .delayElement(Duration.ofMillis(1))
+        .subscribe {
+          it.query()
+          latch.countDown()
+        }
     }
     latch.await()
 
@@ -43,7 +46,6 @@ abstract class AbstractSubscriptionTest extends InstrumentationSpecification {
         }
       }
     }
-
   }
 
   static class Connection {