Leader using KubernetesHelper

spring-cloud-kubernetes-leader/src/main/java/org/springframework/cloud/kubernetes/leader/Leader.java

@@ -25,9 +25,12 @@ public class Leader {
 
 	private final String id;
 
-	public Leader(String role, String id) {
+	private final KubernetesHelper kubernetesHelper;
+
+	public Leader(String role, String id, KubernetesHelper kubernetesHelper) {
 		this.role = role;
 		this.id = id;
+		this.kubernetesHelper = kubernetesHelper;
 	}
 
 	public String getRole() {
@@ -38,4 +41,8 @@ public class Leader {
 		return id;
 	}
 
+	public boolean isValid() {
+		return kubernetesHelper.isPodAlive(id);
+	}
+
 }

spring-cloud-kubernetes-leader/src/test/java/org/springframework/cloud/kubernetes/leader/LeaderTest.java

@@ -0,0 +1,53 @@
+package org.springframework.cloud.kubernetes.leader;
+
+import org.junit.Before;
+import org.junit.Test;
+import org.junit.runner.RunWith;
+import org.mockito.Mock;
+import org.mockito.junit.MockitoJUnitRunner;
+
+import static org.assertj.core.api.Assertions.assertThat;
+import static org.mockito.BDDMockito.given;
+import static org.mockito.Mockito.verify;
+
+/**
+ * @author <a href="mailto:gytis@redhat.com">Gytis Trikleris</a>
+ */
+@RunWith(MockitoJUnitRunner.class)
+public class LeaderTest {
+
+	private static final String ROLE = "test-role";
+
+	private static final String ID = "test-id";
+
+	@Mock
+	private KubernetesHelper mockKubernetesHelper;
+
+	private Leader leader;
+
+	@Before
+	public void before() {
+		leader = new Leader(ROLE, ID, mockKubernetesHelper);
+	}
+
+	@Test
+	public void shouldGetRole() {
+		assertThat(leader.getRole()).isEqualTo(ROLE);
+	}
+
+	@Test
+	public void shouldGetId() {
+		assertThat(leader.getId()).isEqualTo(ID);
+	}
+
+	@Test
+	public void shouldCheckValidity() {
+		given(mockKubernetesHelper.isPodAlive(ID)).willReturn(true);
+
+		boolean result = leader.isValid();
+
+		assertThat(result).isTrue();
+		verify(mockKubernetesHelper).isPodAlive(ID);
+	}
+
+}