From 61102b5ac7be23b9eabdc005ee0451559f326c54 Mon Sep 17 00:00:00 2001
From: jay vyas <jay@apache.org>
Date: Wed, 9 Mar 2016 16:32:32 -0500
Subject: [PATCH] Concurrent liveness test to gaurantee test finishes in 2
 minutes with meaningfull results

---
 docs/user-guide/liveness/exec-liveness.yaml |  2 +-
 docs/user-guide/liveness/http-liveness.yaml |  2 +-
 test/e2e/examples.go                        | 33 ++++++++++++++++-----
 3 files changed, 28 insertions(+), 9 deletions(-)

diff --git a/docs/user-guide/liveness/exec-liveness.yaml b/docs/user-guide/liveness/exec-liveness.yaml
index 7b2af47c030..461691df221 100644
--- a/docs/user-guide/liveness/exec-liveness.yaml
+++ b/docs/user-guide/liveness/exec-liveness.yaml
@@ -18,4 +18,4 @@ spec:
         - /tmp/health
       initialDelaySeconds: 15
       timeoutSeconds: 1
-    name: liveness
+    name: liveness-exec
diff --git a/docs/user-guide/liveness/http-liveness.yaml b/docs/user-guide/liveness/http-liveness.yaml
index c2c6065450f..e7196c65593 100644
--- a/docs/user-guide/liveness/http-liveness.yaml
+++ b/docs/user-guide/liveness/http-liveness.yaml
@@ -18,4 +18,4 @@ spec:
             value: Awesome
       initialDelaySeconds: 15
       timeoutSeconds: 1
-    name: liveness
+    name: liveness-http
diff --git a/test/e2e/examples.go b/test/e2e/examples.go
index 85166054548..9b5708f2030 100644
--- a/test/e2e/examples.go
+++ b/test/e2e/examples.go
@@ -22,6 +22,7 @@ import (
 	"os"
 	"path/filepath"
 	"strings"
+	"sync"
 	"time"
 
 	"k8s.io/kubernetes/pkg/api"
@@ -312,24 +313,42 @@ var _ = Describe("[Feature:Example]", func() {
 
 			runKubectlOrDie("create", "-f", execYaml, nsFlag)
 			runKubectlOrDie("create", "-f", httpYaml, nsFlag)
+
+			// Since both containers start rapidly, we can easily run this test in parallel.
+			var wg sync.WaitGroup
+			passed := true
 			checkRestart := func(podName string, timeout time.Duration) {
 				err := waitForPodRunningInNamespace(c, podName, ns)
 				Expect(err).NotTo(HaveOccurred())
-
 				for t := time.Now(); time.Since(t) < timeout; time.Sleep(poll) {
 					pod, err := c.Pods(ns).Get(podName)
 					expectNoError(err, fmt.Sprintf("getting pod %s", podName))
-					restartCount := api.GetExistingContainerStatus(pod.Status.ContainerStatuses, "liveness").RestartCount
-					Logf("Pod: %s   restart count:%d", podName, restartCount)
-					if restartCount > 0 {
+					stat := api.GetExistingContainerStatus(pod.Status.ContainerStatuses, podName)
+					Logf("Pod: %s, restart count:%d", stat.Name, stat.RestartCount)
+					if stat.RestartCount > 0 {
+						Logf("Saw %v restart, succeeded...", podName)
+						wg.Done()
 						return
 					}
 				}
-				Failf("Pod %s was not restarted", podName)
+				Logf("Failed waiting for %v restart! ", podName)
+				passed = false
+				wg.Done()
 			}
+
 			By("Check restarts")
-			checkRestart("liveness-exec", time.Minute)
-			checkRestart("liveness-http", time.Minute)
+
+			// Start the "actual test", and wait for both pods to complete.
+			// If 2 fail: Something is broken with the test (or maybe even with liveness).
+			// If 1 fails: Its probably just an error in the examples/ files themselves.
+			wg.Add(2)
+			for _, c := range []string{"liveness-http", "liveness-exec"} {
+				go checkRestart(c, 2*time.Minute)
+			}
+			wg.Wait()
+			if !passed {
+				Failf("At least one liveness example failed.  See the logs above.")
+			}
 		})
 	})