From 2e4ff38f268bd8af7a4cd62df2332be72a5af64b Mon Sep 17 00:00:00 2001
From: Chris Egerton <chrise@aiven.io>
Date: Tue, 11 Jun 2024 11:42:01 -0400
Subject: [PATCH] MINOR: Wait for embedded clusters to start before using them
 in Connect OffsetsApiIntegrationTest

---
 .../connect/integration/OffsetsApiIntegrationTest.java   | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/connect/runtime/src/test/java/org/apache/kafka/connect/integration/OffsetsApiIntegrationTest.java b/connect/runtime/src/test/java/org/apache/kafka/connect/integration/OffsetsApiIntegrationTest.java
index dc507b68df7c6..2da52cf9abd05 100644
--- a/connect/runtime/src/test/java/org/apache/kafka/connect/integration/OffsetsApiIntegrationTest.java
+++ b/connect/runtime/src/test/java/org/apache/kafka/connect/integration/OffsetsApiIntegrationTest.java
@@ -143,6 +143,15 @@ private static EmbeddedConnectCluster createOrReuseConnectWithWorkerProps(Map<St
 
             result.start();
 
+            try {
+                result.assertions().assertExactlyNumWorkersAreUp(
+                        NUM_WORKERS,
+                        "Workers did not complete startup in time"
+                );
+            } catch (InterruptedException e) {
+                throw new RuntimeException("Interrupted while awaiting cluster startup", e);
+            }
+
             return result;
         });
     }