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,21 +10,24 @@ 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:
Mono<Connection> connection = Mono.create {
it.success(new Connection())
}
CountDownLatch latch = new CountDownLatch(1) CountDownLatch latch = new CountDownLatch(1)
runWithSpan("parent") { runWithSpan("parent") {
Mono<Connection> connection = Mono.create { connection
it.success(new Connection()) .delayElement(Duration.ofMillis(1))
} .subscribe {
connection.subscribe { it.query()
it.query() latch.countDown()
latch.countDown() }
}
} }
latch.await() latch.await()
@ -43,7 +46,6 @@ abstract class AbstractSubscriptionTest extends InstrumentationSpecification {
} }
} }
} }
} }
static class Connection { static class Connection {