Improve test

Author: vietj

vertx-core/src/test/java/io/vertx/tests/http/Http2MYRServerTest.java

@@ -81,7 +81,7 @@ private void testMYR(boolean multiplexImplementation) throws Exception {
     AtomicInteger inflightRequests = new AtomicInteger();
     AtomicInteger maxInflightRequests = new AtomicInteger();
     AtomicInteger receivedRstFrames = new AtomicInteger();
-    CompletableFuture<Void> goAway = new CompletableFuture<>();
+    CompletableFuture<Boolean> goAway = new CompletableFuture<>();
 
     server.requestHandler(req -> {
       int val = inflightRequests.incrementAndGet();
@@ -153,7 +153,7 @@ public void onSettingsRead(ChannelHandlerContext ctx, Http2Settings settings) th
 
                   @Override
                   public void onGoAwayRead(ChannelHandlerContext ctx, int lastStreamId, long errorCode, ByteBuf debugData) throws Http2Exception {
-                    goAway.complete(null);
+                    goAway.complete(true);
                   }
                 });
                 return super.build();
@@ -167,7 +167,7 @@ public void onGoAwayRead(ChannelHandlerContext ctx, int lastStreamId, long error
               @Override
               public void exceptionCaught(ChannelHandlerContext ctx, Throwable cause) throws Exception {
                 if (cause instanceof IOException && cause.getMessage().equals("Connection reset by peer")) {
-                  goAway.complete(null);
+                  goAway.complete(false);
                 } else {
                   goAway.completeExceptionally(cause);
                 }
@@ -201,10 +201,13 @@ public ChannelFuture connect(int port, String host, BiConsumer<ChannelHandlerCon
       chctx.flush();
     }).sync();
 
-    goAway.get(20, TimeUnit.SECONDS);
-
     // Check the number of rst frame received before getting a go away
-    assertEquals(receivedRstFrames.get(), maxRstFramePerWindow + 1);
+    if (goAway.get(20, TimeUnit.SECONDS)) {
+      assertEquals(receivedRstFrames.get(), maxRstFramePerWindow + 1);
+    } else {
+      assertTrue(receivedRstFrames.get() < maxRstFramePerWindow + 1);
+    }
+
     assertTrue(maxInflightRequests.get() <= 2 * maxRstFramePerWindow );
   }
 }