retry on kafka connect test setup failure

kafbat · Nov 6, 2024 · 62598e4 · 62598e4
1 parent 8f5915c
commit 62598e4
Showing 1 changed file with 21 additions and 0 deletions.
diff --git a/api/src/test/java/io/kafbat/ui/KafkaConnectServiceTests.java b/api/src/test/java/io/kafbat/ui/KafkaConnectServiceTests.java
@@ -17,6 +17,7 @@
 import java.util.List;
 import java.util.Map;
 import java.util.UUID;
+import java.util.concurrent.TimeUnit;
 import lombok.extern.slf4j.Slf4j;
 import org.junit.jupiter.api.AfterEach;
 import org.junit.jupiter.api.BeforeEach;
@@ -44,6 +45,26 @@ public class KafkaConnectServiceTests extends AbstractIntegrationTest {
 
   @BeforeEach
   public void setUp() {
+
+    int limit = 5;
+    int tries = 0;
+    boolean failed = false;
+
+    do {
+      try {
+        TimeUnit.SECONDS.sleep(1);
+        webTestClient.get()
+            .uri("/api/clusters/{clusterName}/connects/{connectName}/connectors", LOCAL,
+                connectName)
+            .exchange()
+            .expectStatus().isOk();
+      } catch (Exception e) {
+        failed = true;
+        System.out.println("failed to retrieve connectors: " + tries);
+      }
+      tries++;
+    } while (failed == true && tries < limit);
+
     webTestClient.post()
         .uri("/api/clusters/{clusterName}/connects/{connectName}/connectors", LOCAL, connectName)
         .bodyValue(new NewConnectorDTO()