mirror of https://github.com/apache/kafka.git
KAFKA-6145: KIP-441: fix flaky shouldEnforceRebalance test in StreamThreadTest (#8452)
Reviewers: Boyang Chen <boyang@confluent.io>, John Roesler <vvcephei@apache.org>
This commit is contained in:
parent
bf6dffe93b
commit
0470e2bc95
|
@ -508,13 +508,16 @@ public class StreamThreadTest {
|
||||||
10 * 1000,
|
10 * 1000,
|
||||||
"Thread never started.");
|
"Thread never started.");
|
||||||
|
|
||||||
|
TestUtils.retryOnExceptionWithTimeout(
|
||||||
|
() -> EasyMock.verify(mockConsumer)
|
||||||
|
);
|
||||||
|
|
||||||
thread.shutdown();
|
thread.shutdown();
|
||||||
TestUtils.waitForCondition(
|
TestUtils.waitForCondition(
|
||||||
() -> thread.state() == StreamThread.State.DEAD,
|
() -> thread.state() == StreamThread.State.DEAD,
|
||||||
10 * 1000,
|
10 * 1000,
|
||||||
"Thread never shut down.");
|
"Thread never shut down.");
|
||||||
|
|
||||||
EasyMock.verify(mockConsumer);
|
|
||||||
}
|
}
|
||||||
|
|
||||||
private static class EasyMockConsumerClientSupplier extends MockClientSupplier {
|
private static class EasyMockConsumerClientSupplier extends MockClientSupplier {
|
||||||
|
|
Loading…
Reference in New Issue