From 8b9164b01ece5f523c7f3f278d13810db8b1d4b0 Mon Sep 17 00:00:00 2001
From: Soby Chacko <soby.chacko@broadcom.com>
Date: Thu, 28 Mar 2024 13:03:33 -0400
Subject: [PATCH] Addressing PR review

---
 .../kafka/listener/KafkaMessageListenerContainerTests.java    | 4 +---
 1 file changed, 1 insertion(+), 3 deletions(-)

diff --git a/spring-kafka/src/test/java/org/springframework/kafka/listener/KafkaMessageListenerContainerTests.java b/spring-kafka/src/test/java/org/springframework/kafka/listener/KafkaMessageListenerContainerTests.java
index 12054db09f..9368233073 100644
--- a/spring-kafka/src/test/java/org/springframework/kafka/listener/KafkaMessageListenerContainerTests.java
+++ b/spring-kafka/src/test/java/org/springframework/kafka/listener/KafkaMessageListenerContainerTests.java
@@ -4252,9 +4252,7 @@ public void invokeBatchInterceptorSuccessFailureOnRetry() throws Exception {
 		ConsumerRecords<Integer, String> consumerRecords = new ConsumerRecords<>(records);
 		AtomicInteger invocation = new AtomicInteger(0);
 		given(consumer.poll(any(Duration.class))).willAnswer(i -> {
-			Thread.sleep(50);
-			if (invocation.get() == 0) {
-				invocation.getAndIncrement();
+			if (invocation.getAndIncrement() == 0) {
 				return consumerRecords;
 			}
 			else {