From b2f2eb3bb19770898a4b2e47d33939a8251be94d Mon Sep 17 00:00:00 2001
From: Julien Viet <julien@julienviet.com>
Date: Wed, 10 Jul 2024 13:49:47 +0200
Subject: [PATCH] Improve Http2Test#testCloseMulti reliability

---
 .../java/io/vertx/tests/http/HttpTest.java    | 20 +++++++++++++------
 1 file changed, 14 insertions(+), 6 deletions(-)

diff --git a/vertx-core/src/test/java/io/vertx/tests/http/HttpTest.java b/vertx-core/src/test/java/io/vertx/tests/http/HttpTest.java
index 600bb6a7253..d6ef879f62d 100644
--- a/vertx-core/src/test/java/io/vertx/tests/http/HttpTest.java
+++ b/vertx-core/src/test/java/io/vertx/tests/http/HttpTest.java
@@ -90,28 +90,36 @@ public void testCloseMulti() throws Exception {
         .onComplete(onSuccess(connections::add));
     }
     waitUntil(() -> connections.size() == 4);
+    Set<String> actual = new HashSet<>();
+    Set<String> expected = new HashSet<>();
     for (int i = 0;i < num;i++) {
       Buffer body = awaitFuture(connections.get(i)
         .request(requestOptions)
         .compose(req -> req.send()
           .compose(HttpClientResponse::body)
         ));
-      assertEquals("Server " + i, body.toString());
+      expected.add("Server " + i);
+      actual.add(body.toString());
     }
+    assertEquals(expected, actual);
     awaitFuture(servers[3].close());
+    int successes = 0;
+    int failures = 0;
     for (int i = 0;i < num;i++) {
       Future<Buffer> fut = connections.get(i)
         .request(requestOptions)
         .compose(req -> req.send()
           .compose(HttpClientResponse::body)
         );
-      if (i < num - 1) {
-        fut.onComplete(onSuccess(body -> complete()));
-      } else {
-        fut.onComplete(onFailure(err -> complete()));
+      try {
+        awaitFuture(fut);
+        successes++;
+      } catch (Exception e) {
+        failures++;
       }
     }
-    await();
+    assertEquals(3, successes);
+    assertEquals(1, failures);
   }
 
   @Rule