Test topic client using multiple brokers

bakdata · Jan 30, 2025 · 6886227 · 6886227
1 parent 4df0519
commit 6886227
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 5 deletions.
diff --git a/streams-bootstrap-core/build.gradle.kts b/streams-bootstrap-core/build.gradle.kts
@@ -43,7 +43,6 @@ dependencies {
         name = "kafka",
         version = testContainersVersion
     )
-    testFixturesImplementation(group = "org.assertj", name = "assertj-core", version = assertJVersion)
     testImplementation(group = "io.confluent", name = "kafka-streams-avro-serde", version = confluentVersion)
     val log4jVersion: String by project
     testImplementation(group = "org.apache.logging.log4j", name = "log4j-slf4j2-impl", version = log4jVersion)

diff --git a/...s-bootstrap-core/src/testFixtures/java/com/bakdata/kafka/ApacheKafkaContainerCluster.java b/...s-bootstrap-core/src/testFixtures/java/com/bakdata/kafka/ApacheKafkaContainerCluster.java
@@ -24,7 +24,6 @@
 
 package com.bakdata.kafka;
 
-import static org.assertj.core.api.Assertions.assertThat;
 import static org.awaitility.Awaitility.await;
 
 import java.time.Duration;
@@ -98,7 +97,7 @@ public void start() {
 
         await()
                 .atMost(Duration.ofSeconds(120))
-                .untilAsserted(() -> {
+                .until(() -> {
                     final Container.ExecResult result = this.brokers.stream()
                             .findFirst()
                             .get()
@@ -111,8 +110,12 @@ public void start() {
                             );
                     final String brokers = result.getStdout().replace("\n", "");
 
-                    assertThat(brokers).asInt().isEqualTo(this.brokersNum);
-                });
+                    return Integer.parseInt(brokers);
+                }, readyBrokers -> readyBrokers == this.brokersNum);
+        await()
+                .pollDelay(Duration.ofSeconds(10))
+                .atMost(Duration.ofSeconds(11))
+                .until(() -> true);
     }
 
     @Override