fix: retry when failed

Author: ReaJason

integration-test/src/test/java/com/reajason/javaweb/integration/memshell/tomcat/Tomcat10WebSocketBypassNginxTest.java

@@ -12,6 +12,7 @@
 import lombok.extern.slf4j.Slf4j;
 import org.apache.commons.lang3.tuple.Pair;
 import org.junit.jupiter.api.Test;
+import org.junitpioneer.jupiter.RetryingTest;
 import org.objectweb.asm.Opcodes;
 import org.testcontainers.containers.ComposeContainer;
 import org.testcontainers.junit.jupiter.Container;
@@ -37,7 +38,7 @@ public class Tomcat10WebSocketBypassNginxTest {
             new ComposeContainer(new File("docker-compose/tomcat/docker-compose-10.1-jre11-nginx.yaml"))
                     .withExposedService("nginx", 80);
 
-    @Test
+    @RetryingTest(3)
     public void testWs() {
         String url = getUrl();
         String server = ServerType.TOMCAT;

integration-test/src/test/java/com/reajason/javaweb/integration/memshell/tomcat/Tomcat8WebSocketBypassNginxTest.java

@@ -11,7 +11,7 @@
 import com.reajason.javaweb.packer.Packers;
 import lombok.extern.slf4j.Slf4j;
 import org.apache.commons.lang3.tuple.Pair;
-import org.junit.jupiter.api.Test;
+import org.junitpioneer.jupiter.RetryingTest;
 import org.objectweb.asm.Opcodes;
 import org.testcontainers.containers.ComposeContainer;
 import org.testcontainers.junit.jupiter.Container;
@@ -37,7 +37,15 @@ public class Tomcat8WebSocketBypassNginxTest {
             new ComposeContainer(new File("docker-compose/tomcat/docker-compose-8-jre8-nginx.yaml"))
                     .withExposedService("nginx", 80);
 
-    @Test
+    public static String getUrl() {
+        String host = compose.getServiceHost("nginx", 80);
+        int port = compose.getServicePort("nginx", 80);
+        String url = "http://" + host + ":" + port + "/app";
+        log.info("container started, app url is : {}", url);
+        return url;
+    }
+
+    @RetryingTest(3)
     public void testWs() {
         String url = getUrl();
         String server = ServerType.TOMCAT;
@@ -59,12 +67,4 @@ public void testWs() {
         String response = BlockingJavaWebSocketClient.sendRequestWaitResponseWithHeader(shellUrl, "id", commandConfig.getHeaderName(), commandConfig.getHeaderValue());
         assertTrue(response.contains("uid="));
     }
-
-    public static String getUrl() {
-        String host = compose.getServiceHost("nginx", 80);
-        int port = compose.getServicePort("nginx", 80);
-        String url = "http://" + host + ":" + port + "/app";
-        log.info("container started, app url is : {}", url);
-        return url;
-    }
 }