Fix Race in Test

Author: garyrussell

spring-kafka/src/test/java/org/springframework/kafka/listener/KafkaMessageListenerContainerTests.java

@@ -2772,12 +2772,14 @@ public void rePausePartitionAfterRebalance() throws Exception {
 			return null;
 		}).given(consumer).pause(any());
 		given(consumer.paused()).willReturn(pausedParts);
+		CountDownLatch firstPoll = new CountDownLatch(1);
 		given(consumer.poll(any(Duration.class))).willAnswer(i -> {
 			if (paused.get()) {
 				pauseLatch1.countDown();
 				// hold up the consumer thread while we revoke/assign partitions on the test thread
 				suspendConsumerThread.await(10, TimeUnit.SECONDS);
 			}
+			firstPoll.countDown();
 			Thread.sleep(50);
 			return ConsumerRecords.empty();
 		});
@@ -2801,6 +2803,7 @@ public void rePausePartitionAfterRebalance() throws Exception {
 				new KafkaMessageListenerContainer<>(cf, containerProps);
 		container.start();
 		InOrder inOrder = inOrder(consumer);
+		assertThat(firstPoll.await(10, TimeUnit.SECONDS)).isNotNull();
 		container.pausePartition(tp0);
 		container.pausePartition(tp1);
 		assertThat(pauseLatch1.await(10, TimeUnit.SECONDS)).isTrue();