Add a test for server RST flood protection

eclipse-vertx · Oct 12, 2023 · 28da3ef · 28da3ef
1 parent d04371e
commit 28da3ef
Showing 1 changed file with 25 additions and 0 deletions.
diff --git a/src/test/java/io/vertx/core/http/Http2Test.java b/src/test/java/io/vertx/core/http/Http2Test.java
@@ -1047,4 +1047,29 @@ public void testConnectionCloseEvictsConnectionFromThePoolBeforeStreamsAreClosed
     }));
     await();
   }
+
+  @Test
+  public void testRstFloodProtection() throws Exception {
+    server.requestHandler(req -> {
+    });
+    startServer(testAddress);
+    int num = HttpServerOptions.DEFAULT_HTTP2_RST_FLOOD_MAX_RST_FRAME_PER_WINDOW + 1;
+    for (int i = 0;i < num;i++) {
+      int val = i;
+      client.request(requestOptions, onSuccess(req -> {
+        if (val == 0) {
+          req
+            .connection()
+            .goAwayHandler(ga -> {
+              assertEquals(11, ga.getErrorCode()); // Enhance your calm
+              testComplete();
+            });
+        }
+        req.end().onComplete(onSuccess(v -> {
+          req.reset();
+        }));
+      }));
+    }
+    await();
+  }
 }