Show only pods which are created by plugin on plugin status report page (#5) (#16)

Author: GaneshSPatil

src/main/java/cd/go/contrib/elasticagent/model/KubernetesCluster.java

@@ -39,7 +39,11 @@ public KubernetesCluster(KubernetesClient client) throws ParseException {
 
     private void fetchPods(KubernetesClient dockerClient) throws ParseException {
         final Map<String, KubernetesNode> dockerNodeMap = nodes.stream().distinct().collect(toMap(KubernetesNode::getName, node -> node));
-        final List<Pod> pods = dockerClient.pods().inNamespace(Constants.KUBERNETES_NAMESPACE_KEY).list().getItems();
+
+        final List<Pod> pods = dockerClient.pods().inNamespace(Constants.KUBERNETES_NAMESPACE_KEY)
+                .withLabel(Constants.CREATED_BY_LABEL_KEY, Constants.PLUGIN_ID)
+                .list().getItems();
+
         LOG.info("Running pods " + pods.size());
 
         for (Pod pod : pods) {

src/test/java/cd/go/contrib/elasticagent/executors/StatusReportExecutorTest.java

@@ -65,6 +65,7 @@ public void shouldBuildStatusReportView() throws Exception {
         when(kubernetesClient.nodes()).thenReturn(nodes);
 
         when(pods.inNamespace(Constants.KUBERNETES_NAMESPACE_KEY)).thenReturn(pods);
+        when(pods.withLabel(Constants.CREATED_BY_LABEL_KEY, Constants.PLUGIN_ID)).thenReturn(pods);
         when(pods.list()).thenReturn(new PodList());
         when(kubernetesClient.pods()).thenReturn(pods);
 
@@ -79,4 +80,4 @@ public void shouldBuildStatusReportView() throws Exception {
         assertThat(response.responseCode(), is(200));
         assertThat(response.responseBody(), is("{\"view\":\"status-report\"}"));
     }
-}
+}

src/test/java/cd/go/contrib/elasticagent/model/KubernetesClusterTest.java

@@ -17,19 +17,13 @@
 package cd.go.contrib.elasticagent.model;
 
 import cd.go.contrib.elasticagent.Constants;
-import io.fabric8.kubernetes.api.model.Node;
 import io.fabric8.kubernetes.api.model.NodeList;
 import io.fabric8.kubernetes.api.model.PodList;
 import io.fabric8.kubernetes.client.KubernetesClient;
 import io.fabric8.kubernetes.client.dsl.internal.NodeOperationsImpl;
 import io.fabric8.kubernetes.client.dsl.internal.PodOperationsImpl;
 import org.junit.Test;
 
-import java.util.Arrays;
-import java.util.List;
-
-import static org.hamcrest.Matchers.hasSize;
-import static org.junit.Assert.assertThat;
 import static org.mockito.Mockito.*;
 
 public class KubernetesClusterTest {
@@ -44,6 +38,7 @@ public void shouldCreateKubernetesClusterObject() throws Exception {
         when(kubernetesClient.nodes()).thenReturn(nodes);
 
         when(pods.inNamespace(Constants.KUBERNETES_NAMESPACE_KEY)).thenReturn(pods);
+        when(pods.withLabel(Constants.CREATED_BY_LABEL_KEY, Constants.PLUGIN_ID)).thenReturn(pods);
         when(pods.list()).thenReturn(new PodList());
         when(kubernetesClient.pods()).thenReturn(pods);
 
@@ -52,4 +47,4 @@ public void shouldCreateKubernetesClusterObject() throws Exception {
         verify(kubernetesClient, times(1)).nodes();
         verify(kubernetesClient, times(1)).pods();
     }
-}
+}