From 5de42f79a3a0c6f25974bf270f77bd7ed7648bc7 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 9f2472026d6de..18822d3b08ba6 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
@@ -598,7 +598,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);