diff --git a/vertx-pg-client/src/test/java/io/vertx/pgclient/PreparedStatementTestBase.java b/vertx-pg-client/src/test/java/io/vertx/pgclient/PreparedStatementTestBase.java index 461b4be86..96bfe746b 100644 --- a/vertx-pg-client/src/test/java/io/vertx/pgclient/PreparedStatementTestBase.java +++ b/vertx-pg-client/src/test/java/io/vertx/pgclient/PreparedStatementTestBase.java @@ -489,24 +489,25 @@ private void testInferDataType42P18(TestContext ctx, Class type, T value, } private void testInferDataType42P18(TestContext ctx, Class type, T value, String suffix1, String suffix2) { + PgConnection.connect(vertx, options()).onComplete(ctx.asyncAssertSuccess(conn -> { + conn + .preparedQuery("SELECT CONCAT('HELLO ', $1)").execute(Tuple.of(value)) + .map(rows -> rows.iterator().next().getString(0)) + .eventually(() -> conn.close()) + .onComplete(ctx.asyncAssertSuccess(str -> { + ctx.assertEquals("HELLO " + suffix1, str); + })); + })); Object array = Array.newInstance(type, 1); Array.set(array, 0, value); - PgConnection.connect(vertx, options(), ctx.asyncAssertSuccess(conn -> { + PgConnection.connect(vertx, options()).onComplete(ctx.asyncAssertSuccess(conn -> { conn - .preparedQuery("SELECT CONCAT('HELLO ', $1)").execute(Tuple.of(value)) - .map(result1 -> { - Row row1 = result1.iterator().next(); - ctx.assertEquals("HELLO " + suffix1, row1.getString(0)); - return ""; - }) - .compose(v -> conn.preparedQuery("SELECT CONCAT('HELLO ', $1)").execute(Tuple.of(array))) - .map(result2 -> { - Row row2 = result2.iterator().next(); - String v = row2.getString(0); - ctx.assertEquals("HELLO " + suffix2, row2.getString(0)); - return ""; - }) - .eventually(v -> conn.close()); + .preparedQuery("SELECT CONCAT('HELLO ', $1)").execute(Tuple.of(array)) + .map(rows -> rows.iterator().next().getString(0)) + .eventually(() -> conn.close()) + .onComplete(ctx.asyncAssertSuccess(str -> { + ctx.assertEquals("HELLO " + suffix2, str); + })); })); }