INT-4088 ZookeeperLeaderTests: fix race condition

Two initiators for the same path, same `SmartLifecycleRoleController` and, finally, same `adapter`.
So, one initiator after `yield()` stops the `adapter` and at the same time another starts it.

Since there is no barrier in between events and assertion, we end up with an early "re-granting".

* Add `CountDownLatch yieldBarrier` to `countDown()` after performing second `adapter.isRunning()` assert
* `LeaderEventPublisher` waits for the `yieldBarrier` after the first `OnRevokedEvent`

Author: artembilan

spring-integration-zookeeper/src/test/java/org/springframework/integration/zookeeper/event/ZookeeperLeaderTests.java

@@ -24,6 +24,7 @@
 
 import java.util.Collections;
 import java.util.concurrent.BlockingQueue;
+import java.util.concurrent.CountDownLatch;
 import java.util.concurrent.LinkedBlockingQueue;
 import java.util.concurrent.TimeUnit;
 
@@ -62,6 +63,8 @@ public class ZookeeperLeaderTests extends ZookeeperTestSupport {
 	private final SmartLifecycleRoleController controller = new SmartLifecycleRoleController(
 			Collections.singletonList("sitest"), Collections.<SmartLifecycle>singletonList(this.adapter));
 
+	private final CountDownLatch yieldBarrier = new CountDownLatch(1);
+
 	@Test
 	public void testLeader() throws Exception {
 		assertFalse(this.adapter.isRunning());
@@ -87,6 +90,8 @@ public void testLeader() throws Exception {
 
 		assertFalse(this.adapter.isRunning());
 
+		this.yieldBarrier.countDown();
+
 		event = this.events.poll(30, TimeUnit.SECONDS);
 		assertNotNull(event);
 		assertThat(event, instanceOf(OnGrantedEvent.class));
@@ -105,13 +110,25 @@ public void testLeader() throws Exception {
 	private LeaderEventPublisher publisher() {
 		return new DefaultLeaderEventPublisher(new ApplicationEventPublisher() {
 
+			volatile boolean onRevokedEventHappened;
+
 			@Override
 			public void publishEvent(Object event) {
 			}
 
 			@Override
 			public void publishEvent(ApplicationEvent event) {
 				AbstractLeaderEvent leadershipEvent = (AbstractLeaderEvent) event;
+				if (this.onRevokedEventHappened) {
+					try {
+						yieldBarrier.await(10, TimeUnit.SECONDS);
+					}
+					catch (InterruptedException e) {
+						Thread.currentThread().interrupt();
+						throw new RuntimeException(e);
+					}
+				}
+				onRevokedEventHappened = event instanceof OnRevokedEvent;
 				controller.onApplicationEvent((AbstractLeaderEvent) event);
 				events.add(leadershipEvent);
 			}