diff --git a/dd-java-agent/instrumentation/vertx/vertx-pg-client/vertx-pg-client-4.0/src/test/groovy/VertxPostgresSqlClientForkedTest.groovy b/dd-java-agent/instrumentation/vertx/vertx-pg-client/vertx-pg-client-4.0/src/test/groovy/VertxPostgresSqlClientForkedTest.groovy index d294e0de339..cb6e0b965f0 100644 --- a/dd-java-agent/instrumentation/vertx/vertx-pg-client/vertx-pg-client-4.0/src/test/groovy/VertxPostgresSqlClientForkedTest.groovy +++ b/dd-java-agent/instrumentation/vertx/vertx-pg-client/vertx-pg-client-4.0/src/test/groovy/VertxPostgresSqlClientForkedTest.groovy @@ -93,37 +93,40 @@ class VertxPostgresSqlClientForkedTest extends InstrumentationSpecification { AsyncResult> queryCursorWithHandler(Cursor cursor) { def latch = new CountDownLatch(1) - AsyncResult> result = null + AsyncResult> asyncResult = null cursor.read(0) { rowSetAR -> runUnderTrace("handler") { - result = rowSetAR + asyncResult = rowSetAR } latch.countDown() } assert latch.await(10, TimeUnit.SECONDS) - return result + assert asyncResult?.succeeded() : "Failed to read cursor: ${asyncResult?.cause()}" + return asyncResult } SqlConnection connection(Pool pool) { def latch = new CountDownLatch(1) - SqlConnection result = null + AsyncResult asyncResult = null pool.getConnection({ connectionAR -> - result = connectionAR.result() + asyncResult = connectionAR latch.countDown() }) assert latch.await(10, TimeUnit.SECONDS) - return result + assert asyncResult?.succeeded() : "Failed to get connection: ${asyncResult?.cause()}" + return asyncResult.result() } PreparedStatement prepare(SqlConnection connection, String sql) { def latch = new CountDownLatch(1) - PreparedStatement result = null + AsyncResult asyncResult = null connection.prepare(sql, { statementAR -> - result = statementAR.result() + asyncResult = statementAR latch.countDown() }) assert latch.await(10, TimeUnit.SECONDS) - return result + assert asyncResult?.succeeded() : "Failed to prepare statement: ${asyncResult?.cause()}" + return asyncResult.result() } void checkDBSpan(TraceAssert ta, DDSpan parent, String resource, String operation, TestDBInfo info, boolean prepared = false) {