Improve racy test

eclipse-vertx · Oct 11, 2023 · d04371e · d04371e
1 parent b7b8015
commit d04371e
Showing 1 changed file with 17 additions and 17 deletions.
diff --git a/src/test/java/io/vertx/core/http/HttpTest.java b/src/test/java/io/vertx/core/http/HttpTest.java
@@ -1706,16 +1706,16 @@ private void testStatusCode(int code, String statusMessage) {
   }
 
   @Test
-  public void testResponseTrailersPutAll() {
+  public void testResponseTrailersPutAll() throws Exception {
     testResponseTrailers(false);
   }
 
   @Test
-  public void testResponseTrailersPutIndividually() {
+  public void testResponseTrailersPutIndividually() throws Exception {
     testResponseTrailers(true);
   }
 
-  private void testResponseTrailers(boolean individually) {
+  private void testResponseTrailers(boolean individually) throws Exception {
     MultiMap trailers = randomMultiMap(10);
 
     server.requestHandler(req -> {
@@ -1730,21 +1730,21 @@ private void testResponseTrailers(boolean individually) {
       req.response().end();
     });
 
-    server.listen(testAddress, onSuccess(s -> {
-      client.request(requestOptions).onComplete(onSuccess(req -> {
-        req.send(onSuccess(resp -> {
-          resp.endHandler(v -> {
-            assertEquals(trailers.size(), resp.trailers().size());
-            for (Map.Entry<String, String> entry : trailers) {
-              assertEquals(entry.getValue(), resp.trailers().get(entry.getKey()));
-              assertEquals(entry.getValue(), resp.getTrailer(entry.getKey()));
-            }
-            testComplete();
-          });
-        }));
-      }));
+    startServer(testAddress);
+    client.request(requestOptions).compose(req -> req
+      .send()
+      .compose(resp -> resp
+        .end()
+        .map(v -> {
+          assertEquals(trailers.size(), resp.trailers().size());
+          for (Map.Entry<String, String> entry : trailers) {
+            assertEquals(entry.getValue(), resp.trailers().get(entry.getKey()));
+            assertEquals(entry.getValue(), resp.getTrailer(entry.getKey()));
+          }
+          return v;
+        }))).onComplete(onSuccess(v -> {
+      testComplete();
     }));
-
     await();
   }