Improve reactor test (#4168)

This commit is contained in:
Trask Stalnaker 2021-09-20 17:21:55 -07:00 committed by GitHub
parent 2c9d6e313f
commit 6a91dae5a1
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 10 additions and 8 deletions

View File

@ -10,18 +10,21 @@ import io.opentelemetry.api.trace.SpanKind
import io.opentelemetry.instrumentation.test.InstrumentationSpecification import io.opentelemetry.instrumentation.test.InstrumentationSpecification
import reactor.core.publisher.Mono import reactor.core.publisher.Mono
import java.time.Duration
import java.util.concurrent.CountDownLatch import java.util.concurrent.CountDownLatch
abstract class AbstractSubscriptionTest extends InstrumentationSpecification { abstract class AbstractSubscriptionTest extends InstrumentationSpecification {
def "subscription test"() { def "subscription test"() {
when: when:
CountDownLatch latch = new CountDownLatch(1)
runWithSpan("parent") {
Mono<Connection> connection = Mono.create { Mono<Connection> connection = Mono.create {
it.success(new Connection()) it.success(new Connection())
} }
connection.subscribe { CountDownLatch latch = new CountDownLatch(1)
runWithSpan("parent") {
connection
.delayElement(Duration.ofMillis(1))
.subscribe {
it.query() it.query()
latch.countDown() latch.countDown()
} }
@ -43,7 +46,6 @@ abstract class AbstractSubscriptionTest extends InstrumentationSpecification {
} }
} }
} }
} }
static class Connection { static class Connection {