MINOR: modified ProducerIdExpirationTest to hopefully be less flaky (a…

…pache#12824) ProducerIdExpirationTest has failed on a few PR runs. Using a waitUntil instead. Reviewers: Luke Chen <[email protected]>
AnonHxy · Nov 5, 2022 · 9467fc3 · 9467fc3
1 parent bc780c7
commit 9467fc3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/core/src/test/scala/integration/kafka/api/ProducerIdExpirationTest.scala b/core/src/test/scala/integration/kafka/api/ProducerIdExpirationTest.scala
@@ -113,7 +113,7 @@ class ProducerIdExpirationTest extends KafkaServerTestHarness {
     producer.flush()
 
     // Ensure producer IDs are added.
-    assertEquals(1, producerState.size)
+    TestUtils.waitUntilTrue(() => producerState.size == 1, "Producer IDs were not added.")
 
     producer.abortTransaction()