diff --git a/dd-java-agent/instrumentation/vertx/src/test/groovy/client/VertxRxCircuitBreakerWebClientTest.groovy b/dd-java-agent/instrumentation/vertx/src/test/groovy/client/VertxRxCircuitBreakerWebClientTest.groovy index b10b93702d..6880b1d87c 100644 --- a/dd-java-agent/instrumentation/vertx/src/test/groovy/client/VertxRxCircuitBreakerWebClientTest.groovy +++ b/dd-java-agent/instrumentation/vertx/src/test/groovy/client/VertxRxCircuitBreakerWebClientTest.groovy @@ -32,23 +32,23 @@ class VertxRxCircuitBreakerWebClientTest extends HttpClientTest headers, Closure callback) { def request = client.request(HttpMethod.valueOf(method), uri.port, uri.host, "$uri") headers.each { request.putHeader(it.key, it.value) } - Future result = breaker.execute { command -> + + def future = new CompletableFuture() + + Future result = breaker.executeCommand({ command -> request.rxSend().doOnSuccess { command.complete(it) }.doOnError { command.fail(it) }.subscribe() - } - - def future = new CompletableFuture() - result.setHandler { + }, { callback?.call() if (it.succeeded()) { future.complete(it.result().statusCode()) } else { future.completeExceptionally(it.cause()) } - } + }) return future.get() } diff --git a/dd-java-agent/instrumentation/vertx/src/test/groovy/server/VertxRxCircuitBreakerHttpServerTest.groovy b/dd-java-agent/instrumentation/vertx/src/test/groovy/server/VertxRxCircuitBreakerHttpServerTest.groovy index 3887d0c3e1..e3d462c8c8 100644 --- a/dd-java-agent/instrumentation/vertx/src/test/groovy/server/VertxRxCircuitBreakerHttpServerTest.groovy +++ b/dd-java-agent/instrumentation/vertx/src/test/groovy/server/VertxRxCircuitBreakerHttpServerTest.groovy @@ -33,10 +33,9 @@ class VertxRxCircuitBreakerHttpServerTest extends VertxHttpServerTest { ) router.route(SUCCESS.path).handler { ctx -> - def result = breaker.execute { future -> + def result = breaker.executeCommand({ future -> future.complete(SUCCESS) - } - result.setHandler { + }, { it -> if (it.failed()) { throw it.cause() } @@ -44,13 +43,12 @@ class VertxRxCircuitBreakerHttpServerTest extends VertxHttpServerTest { controller(endpoint) { ctx.response().setStatusCode(endpoint.status).end(endpoint.body) } - } + }) } router.route(REDIRECT.path).handler { ctx -> - def result = breaker.execute { future -> + def result = breaker.executeCommand({ future -> future.complete(REDIRECT) - } - result.setHandler { + }, { if (it.failed()) { throw it.cause() } @@ -58,13 +56,12 @@ class VertxRxCircuitBreakerHttpServerTest extends VertxHttpServerTest { controller(endpoint) { ctx.response().setStatusCode(endpoint.status).putHeader("location", endpoint.body).end() } - } + }) } router.route(ERROR.path).handler { ctx -> - def result = breaker.execute { future -> + def result = breaker.executeCommand({ future -> future.complete(ERROR) - } - result.setHandler { + }, { if (it.failed()) { throw it.cause() } @@ -72,13 +69,12 @@ class VertxRxCircuitBreakerHttpServerTest extends VertxHttpServerTest { controller(endpoint) { ctx.response().setStatusCode(endpoint.status).end(endpoint.body) } - } + }) } router.route(EXCEPTION.path).handler { ctx -> - def result = breaker.execute { future -> + def result = breaker.executeCommand({ future -> future.fail(new Exception(EXCEPTION.body)) - } - result.setHandler { + }, { try { def cause = it.cause() controller(EXCEPTION) { @@ -87,7 +83,7 @@ class VertxRxCircuitBreakerHttpServerTest extends VertxHttpServerTest { } catch (Exception ex) { ctx.response().setStatusCode(EXCEPTION.status).end(ex.message) } - } + }) } super.@vertx.createHttpServer()