From ca033fe97a978290b3c8434f25c98f351e291259 Mon Sep 17 00:00:00 2001 From: Sagar Rao Date: Tue, 25 Apr 2023 11:11:38 +0530 Subject: [PATCH] KAFKA-14929: Fixing flaky test putTopicStateRetriableFailure --- .../connect/storage/KafkaStatusBackingStoreFormatTest.java | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/connect/runtime/src/test/java/org/apache/kafka/connect/storage/KafkaStatusBackingStoreFormatTest.java b/connect/runtime/src/test/java/org/apache/kafka/connect/storage/KafkaStatusBackingStoreFormatTest.java index 6b509a08baa2c..6d5a4a3c726b5 100644 --- a/connect/runtime/src/test/java/org/apache/kafka/connect/storage/KafkaStatusBackingStoreFormatTest.java +++ b/connect/runtime/src/test/java/org/apache/kafka/connect/storage/KafkaStatusBackingStoreFormatTest.java @@ -48,7 +48,7 @@ import static org.junit.Assert.assertFalse; import static org.junit.Assert.assertNull; import static org.junit.Assert.assertTrue; -import static org.mockito.Mockito.times; +import static org.mockito.Mockito.timeout; import static org.mockito.Mockito.verify; @SuppressWarnings("unchecked") @@ -216,7 +216,7 @@ public void putTopicStateRetriableFailure() { }).when(kafkaBasedLog).send(eq(key), valueCaptor.capture(), any(Callback.class)); store.put(topicStatus); - verify(kafkaBasedLog, times(2)).send(any(), any(), any()); + verify(kafkaBasedLog, timeout(1000).times(2)).send(any(), any(), any()); // check capture state assertEquals(topicStatus, store.parseTopicStatus(valueCaptor.getValue()));