[JAVA-33842] Fix spring-kafka-2 integration test (#16415)

spring-kafka-2/pom.xml

@@ -26,7 +26,10 @@
         <dependency>
             <groupId>org.springframework.kafka</groupId>
             <artifactId>spring-kafka</artifactId>
-            <version>${spring-kafka.version}</version>
+        </dependency>
+        <dependency>
+            <groupId>org.springframework.kafka</groupId>
+            <artifactId>spring-kafka-test</artifactId>
         </dependency>
         <dependency>
             <groupId>org.apache.kafka</groupId>
@@ -46,11 +49,6 @@
             <groupId>org.apache.commons</groupId>
             <artifactId>commons-lang3</artifactId>
         </dependency>
-        <dependency>
-            <groupId>org.springframework.kafka</groupId>
-            <artifactId>spring-kafka-test</artifactId>
-            <scope>test</scope>
-        </dependency>
         <dependency>
             <groupId>org.testcontainers</groupId>
             <artifactId>kafka</artifactId>
@@ -70,9 +68,8 @@
     </dependencies>
 
     <properties>
-        <testcontainers-kafka.version>1.19.3</testcontainers-kafka.version>
+        <testcontainers-kafka.version>1.19.7</testcontainers-kafka.version>
-        <spring-kafka.version>3.1.2</spring-kafka.version>
+        <kafka-streams.version>3.7.0</kafka-streams.version>
-        <kafka-streams.version>3.6.1</kafka-streams.version>
         <start-class>com.baeldung.spring.kafka.dlt.KafkaDltApplication</start-class>
     </properties>
 

spring-kafka-2/src/test/java/com/baeldung/spring/kafka/managingkafkaconsumergroups/ManagingConsumerGroupsIntegrationTest.java

@@ -1,5 +1,11 @@
 package com.baeldung.spring.kafka.managingkafkaconsumergroups;
 
+import static org.junit.jupiter.api.Assertions.assertEquals;
+
+import java.time.Duration;
+import java.util.Objects;
+import java.util.Set;
+
 import org.apache.commons.lang3.RandomUtils;
 import org.junit.jupiter.api.Test;
 import org.springframework.beans.factory.annotation.Autowired;
@@ -8,22 +14,19 @@ import org.springframework.kafka.config.KafkaListenerEndpointRegistry;
 import org.springframework.kafka.core.KafkaTemplate;
 import org.springframework.kafka.listener.MessageListenerContainer;
 import org.springframework.kafka.test.context.EmbeddedKafka;
-import org.springframework.util.CollectionUtils;
-
-import java.util.Objects;
 import org.springframework.test.annotation.DirtiesContext;
 import org.springframework.test.annotation.DirtiesContext.ClassMode;
 import org.springframework.test.context.ActiveProfiles;
-import java.util.Set;
+import org.springframework.util.CollectionUtils;
 
-import static org.junit.jupiter.api.Assertions.assertEquals;
+import lombok.extern.slf4j.Slf4j;
-import static org.junit.jupiter.api.Assertions.assertTrue;
 
 @SpringBootTest(classes = ManagingConsumerGroupsApplicationKafkaApp.class)
 @EmbeddedKafka(partitions = 2, brokerProperties = {"listeners=PLAINTEXT://localhost:9098", "port=9098"}, topics = {"topic1"})
 @DirtiesContext(classMode = ClassMode.BEFORE_CLASS)
 @ActiveProfiles("managed")
-public class ManagingConsumerGroupsIntegrationTest {
+@Slf4j
+class ManagingConsumerGroupsIntegrationTest {
 
     private static final String CONSUMER_1_IDENTIFIER = "org.springframework.kafka.KafkaListenerEndpointContainer#1";
     private static final int TOTAL_PRODUCED_MESSAGES = 5000;
@@ -39,11 +42,11 @@ public class ManagingConsumerGroupsIntegrationTest {
     MessageConsumerService consumerService;
 
     @Test
-    public void givenContinuousMessageFlow_whenAConsumerLeavesTheGroup_thenKafkaTriggersPartitionRebalance() throws InterruptedException {
+    void givenContinuousMessageFlow_whenAConsumerLeavesTheGroup_thenKafkaTriggersPartitionRebalance() throws InterruptedException {
         int currentMessage = 0;
-
         do {
             kafkaTemplate.send("topic-1", RandomUtils.nextDouble(10.0, 20.0));
+            Thread.sleep(0,100);    // Waiting to let the embedded kafka consume the event.
             currentMessage++;
 
             if (currentMessage == MESSAGE_WHERE_CONSUMER_1_LEAVES_GROUP) {
@@ -53,12 +56,13 @@ public class ManagingConsumerGroupsIntegrationTest {
                         .findFirst()
                         .orElse("");
                 MessageListenerContainer container = kafkaListenerEndpointRegistry.getListenerContainer(containerId);
-                Thread.sleep(2000);
                 Objects.requireNonNull(container).stop();
                 kafkaListenerEndpointRegistry.unregisterListenerContainer(containerId);
             }
+            if(currentMessage % 1000 == 0){
+                log.info("Processed {} of {}", currentMessage, TOTAL_PRODUCED_MESSAGES);
+            }
         } while (currentMessage != TOTAL_PRODUCED_MESSAGES);
-        Thread.sleep(2000);
 
         Set<Integer> partitionsConsumedBy1 = consumerService.consumedPartitions.get("consumer-1");
         Set<Integer> partitionsConsumedBy0 = consumerService.consumedPartitions.get("consumer-0");