From 21c3009ac12f79d01daa073ed104956e3f7ff04a Mon Sep 17 00:00:00 2001 From: Kvicii <42023367+Kvicii@users.noreply.github.com> Date: Thu, 3 Feb 2022 17:59:12 +0800 Subject: [PATCH] KAFKA-13583; Fix FetchRequestBetweenDifferentIbpTest flaky tests (#11699) Co-authored-by: Kvicii Reviewers: David Jacot --- .../kafka/server/FetchRequestBetweenDifferentIbpTest.scala | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/core/src/test/scala/integration/kafka/server/FetchRequestBetweenDifferentIbpTest.scala b/core/src/test/scala/integration/kafka/server/FetchRequestBetweenDifferentIbpTest.scala index 37ac4a29ea9..d10bafde69a 100644 --- a/core/src/test/scala/integration/kafka/server/FetchRequestBetweenDifferentIbpTest.scala +++ b/core/src/test/scala/integration/kafka/server/FetchRequestBetweenDifferentIbpTest.scala @@ -73,7 +73,7 @@ class FetchRequestBetweenDifferentIbpTest extends BaseRequestTest { producer.send(record2) consumer.assign(asList(new TopicPartition(topic, 0), new TopicPartition(topic, 1))) - val count = consumer.poll(Duration.ofMillis(1500)).count() + consumer.poll(Duration.ofMillis(1500)).count() + val count = consumer.poll(Duration.ofMillis(5000)).count() + consumer.poll(Duration.ofMillis(5000)).count() assertEquals(2, count) } @@ -109,7 +109,7 @@ class FetchRequestBetweenDifferentIbpTest extends BaseRequestTest { consumer.assign(asList(new TopicPartition(topic, 0), new TopicPartition(topic, 1))) - val count = consumer.poll(Duration.ofMillis(1500)).count() + consumer.poll(Duration.ofMillis(1500)).count() + val count = consumer.poll(Duration.ofMillis(5000)).count() + consumer.poll(Duration.ofMillis(5000)).count() assertEquals(2, count) // Make controller version2 @@ -128,7 +128,7 @@ class FetchRequestBetweenDifferentIbpTest extends BaseRequestTest { // Assign this new topic in addition to the old topics. consumer.assign(asList(new TopicPartition(topic, 0), new TopicPartition(topic, 1), new TopicPartition(topic2, 0))) - val count2 = consumer.poll(Duration.ofMillis(1500)).count() + consumer.poll(Duration.ofMillis(1500)).count() + val count2 = consumer.poll(Duration.ofMillis(5000)).count() + consumer.poll(Duration.ofMillis(5000)).count() assertEquals(2, count2) }