From 1d2a3fd9cb597434d21603f7b948fc0c282f7c40 Mon Sep 17 00:00:00 2001
From: Chia-Ping Tsai <chia7712@gmail.com>
Date: Wed, 23 Dec 2020 10:04:36 +0800
Subject: [PATCH] KAFKA-10815 EosTestDriver#verifyAllTransactionFinished should
 break loop if all partitions are verified (#9706)

Reviewers: Matthias J. Sax <mjsax@apache.org>
---
 .../test/java/org/apache/kafka/streams/tests/EosTestDriver.java | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/streams/src/test/java/org/apache/kafka/streams/tests/EosTestDriver.java b/streams/src/test/java/org/apache/kafka/streams/tests/EosTestDriver.java
index 45843aac11a25..34b3baca4487f 100644
--- a/streams/src/test/java/org/apache/kafka/streams/tests/EosTestDriver.java
+++ b/streams/src/test/java/org/apache/kafka/streams/tests/EosTestDriver.java
@@ -584,7 +584,7 @@ private static void verifyAllTransactionFinished(final KafkaConsumer<byte[], byt
 
         final long maxWaitTime = System.currentTimeMillis() + MAX_IDLE_TIME_MS;
         try (final KafkaConsumer<byte[], byte[]> consumerUncommitted = new KafkaConsumer<>(consumerProps)) {
-            while (System.currentTimeMillis() < maxWaitTime) {
+            while (!partitions.isEmpty() && System.currentTimeMillis() < maxWaitTime) {
                 consumer.seekToEnd(partitions);
                 final Map<TopicPartition, Long> topicEndOffsets = consumerUncommitted.endOffsets(partitions);