From 9ddc4e6d12e972ab32c05db681f1e434b3b46dd6 Mon Sep 17 00:00:00 2001
From: Marius Bogoevici <mbogoevici@pivotal.io>
Date: Thu, 22 Oct 2015 15:03:43 -0400
Subject: [PATCH] Check if leader election is complete during tests

---
 .../springframework/integration/kafka/util/TopicUtils.java    | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/spring-integration-kafka/src/main/java/org/springframework/integration/kafka/util/TopicUtils.java b/spring-integration-kafka/src/main/java/org/springframework/integration/kafka/util/TopicUtils.java
index b41b9e8b4d..19c3dfe8bd 100644
--- a/spring-integration-kafka/src/main/java/org/springframework/integration/kafka/util/TopicUtils.java
+++ b/spring-integration-kafka/src/main/java/org/springframework/integration/kafka/util/TopicUtils.java
@@ -19,6 +19,7 @@ package org.springframework.integration.kafka.util;
 import java.util.List;
 import java.util.Properties;
 
+import kafka.common.LeaderNotAvailableException;
 import org.I0Itec.zkclient.ZkClient;
 import org.apache.commons.logging.Log;
 import org.apache.commons.logging.LogFactory;
@@ -142,6 +143,9 @@ public class TopicUtils {
 						if (partitionMetadata.errorCode() != ErrorMapping.NoError()) {
 							throw (Exception) ErrorMapping.exceptionFor(partitionMetadata.errorCode());
 						}
+						if (partitionMetadata.leader() == null) {
+							throw new LeaderNotAvailableException();
+						}
 					}
 					return topicMetadata;
 				}