diff --git a/streams/src/test/java/org/apache/kafka/streams/processor/internals/StreamThreadTest.java b/streams/src/test/java/org/apache/kafka/streams/processor/internals/StreamThreadTest.java index dbd43488bfe..d0488f62d5b 100644 --- a/streams/src/test/java/org/apache/kafka/streams/processor/internals/StreamThreadTest.java +++ b/streams/src/test/java/org/apache/kafka/streams/processor/internals/StreamThreadTest.java @@ -508,13 +508,16 @@ public class StreamThreadTest { 10 * 1000, "Thread never started."); + TestUtils.retryOnExceptionWithTimeout( + () -> EasyMock.verify(mockConsumer) + ); + thread.shutdown(); TestUtils.waitForCondition( () -> thread.state() == StreamThread.State.DEAD, 10 * 1000, "Thread never shut down."); - EasyMock.verify(mockConsumer); } private static class EasyMockConsumerClientSupplier extends MockClientSupplier {