Merge pull request #1026 from rohanKanojia/issue975

fabric8io · Mar 13, 2018 · 30af800 · 30af800
2 parents 25ba0c6 + 3902fd7
commit 30af800
Showing 1 changed file with 44 additions and 0 deletions.
diff --git a/kubernetes-itests/src/test/java/io/fabric8/kubernetes/PodIT.java b/kubernetes-itests/src/test/java/io/fabric8/kubernetes/PodIT.java
@@ -20,6 +20,9 @@
 import io.fabric8.kubernetes.api.model.PodBuilder;
 import io.fabric8.kubernetes.api.model.PodList;
 import io.fabric8.kubernetes.client.KubernetesClient;
+import io.fabric8.kubernetes.client.dsl.ExecListener;
+import io.fabric8.kubernetes.client.dsl.ExecWatch;
+import okhttp3.Response;
 import org.apache.commons.lang.RandomStringUtils;
 import org.arquillian.cube.kubernetes.api.Session;
 import org.arquillian.cube.kubernetes.impl.requirement.RequiresKubernetes;
@@ -32,6 +35,10 @@
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
 
+import java.io.ByteArrayOutputStream;
+import java.util.concurrent.CountDownLatch;
+import java.util.concurrent.TimeUnit;
+
 import static junit.framework.TestCase.assertNotNull;
 import static org.assertj.core.api.Assertions.assertThat;
 import static org.junit.Assert.assertEquals;
@@ -120,6 +127,43 @@ public void delete() {
     assertTrue(client.pods().inNamespace(currentNamespace).delete(pod1));
   }
 
+  @Test
+  public void log() throws InterruptedException {
+    client.pods().inNamespace(currentNamespace).withName(pod1.getMetadata().getName()).waitUntilReady(10, TimeUnit.MINUTES);
+    String log = client.pods().inNamespace(currentNamespace).withName(pod1.getMetadata().getName()).getLog();
+    assertNotNull(log);
+  }
+
+  @Test
+  public void exec() throws InterruptedException {
+    client.pods().inNamespace(currentNamespace).withName(pod1.getMetadata().getName()).waitUntilReady(10, TimeUnit.MINUTES);
+    final CountDownLatch execLatch = new CountDownLatch(1);
+    ByteArrayOutputStream out = new ByteArrayOutputStream();
+    ExecWatch execWatch = client.pods().inNamespace(currentNamespace).withName(pod1.getMetadata().getName())
+      .writingOutput(out).withTTY().usingListener(new ExecListener() {
+        @Override
+        public void onOpen(Response response) {
+          logger.info("Shell was opened");
+        }
+
+        @Override
+        public void onFailure(Throwable throwable, Response response) {
+          logger.info("Shell barfed");
+          execLatch.countDown();
+        }
+
+        @Override
+        public void onClose(int i, String s) {
+          logger.info("Shell closed");
+          execLatch.countDown();
+        }
+      }).exec("date");
+
+    execLatch.await(5, TimeUnit.MINUTES);
+    assertNotNull(execWatch);
+    assertNotNull(out.toString());
+  }
+
   @After
   public void cleanup() throws InterruptedException {
     client.pods().inNamespace(currentNamespace).withName(pod1.getMetadata().getName()).delete();