From 4975e5d9d9b59460f4a6d88046945536706c6469 Mon Sep 17 00:00:00 2001
From: Yu-Ju Hong <yjhong@google.com>
Date: Wed, 27 Jan 2016 15:20:06 -0800
Subject: [PATCH] e2e reboot: prints stats and logs for not ready pods

---
 test/e2e/reboot.go | 44 ++++++++++++++++++++++++--------------------
 1 file changed, 24 insertions(+), 20 deletions(-)

diff --git a/test/e2e/reboot.go b/test/e2e/reboot.go
index 1fe8558a3bd..0ee03a778b6 100644
--- a/test/e2e/reboot.go
+++ b/test/e2e/reboot.go
@@ -25,6 +25,7 @@ import (
 	client "k8s.io/kubernetes/pkg/client/unversioned"
 	"k8s.io/kubernetes/pkg/fields"
 	"k8s.io/kubernetes/pkg/labels"
+	"k8s.io/kubernetes/pkg/util/sets"
 
 	. "github.com/onsi/ginkgo"
 	. "github.com/onsi/gomega"
@@ -157,7 +158,7 @@ func testReboot(c *client.Client, rebootCmd string) {
 	}
 }
 
-func printStatusAndLogsForNotReadyPods(c *client.Client, oldPods, newPods []*api.Pod) {
+func printStatusAndLogsForNotReadyPods(c *client.Client, ns string, podNames []string, pods []*api.Pod) {
 	printFn := func(id, log string, err error, previous bool) {
 		prefix := "Retrieving log for container"
 		if previous {
@@ -169,25 +170,27 @@ func printStatusAndLogsForNotReadyPods(c *client.Client, oldPods, newPods []*api
 			Logf("%s %s:\n%s\n", prefix, id, log)
 		}
 	}
-	for _, oldPod := range oldPods {
-		for _, p := range newPods {
-			if p.Namespace != oldPod.Namespace || p.Name != oldPod.Name {
-				continue
+	podNameSet := sets.NewString(podNames...)
+	for _, p := range pods {
+		if p.Namespace != ns {
+			continue
+		}
+		if !podNameSet.Has(p.Name) {
+			continue
+		}
+		if ok, _ := podRunningReady(p); ok {
+			continue
+		}
+		Logf("Status for not ready pod %s/%s: %+v", p.Namespace, p.Name, p.Status)
+		// Print the log of the containers if pod is not running and ready.
+		for _, container := range p.Status.ContainerStatuses {
+			cIdentifer := fmt.Sprintf("%s/%s/%s", p.Namespace, p.Name, container.Name)
+			log, err := getPodLogs(c, p.Namespace, p.Name, container.Name)
+			printFn(cIdentifer, log, err, false)
+			// Get log from the previous container.
+			if container.RestartCount > 0 {
+				printFn(cIdentifer, log, err, true)
 			}
-			if ok, _ := podRunningReady(p); !ok {
-				Logf("Status for not ready pod %s/%s: %+v", p.Namespace, p.Name, p.Status)
-				// Print the log of the containers if pod is not running and ready.
-				for _, container := range p.Status.ContainerStatuses {
-					cIdentifer := fmt.Sprintf("%s/%s/%s", p.Namespace, p.Name, container.Name)
-					log, err := getPodLogs(c, p.Namespace, p.Name, container.Name)
-					printFn(cIdentifer, log, err, false)
-					// Get log from the previous container.
-					if container.RestartCount > 0 {
-						printFn(cIdentifer, log, err, true)
-					}
-				}
-			}
-			break
 		}
 	}
 }
@@ -242,6 +245,7 @@ func rebootNode(c *client.Client, provider, name, rebootCmd string) bool {
 	// For each pod, we do a sanity check to ensure it's running / healthy
 	// now, as that's what we'll be checking later.
 	if !checkPodsRunningReady(c, ns, podNames, podReadyBeforeTimeout) {
+		printStatusAndLogsForNotReadyPods(c, ns, podNames, pods)
 		return false
 	}
 
@@ -265,7 +269,7 @@ func rebootNode(c *client.Client, provider, name, rebootCmd string) bool {
 	// running / healthy.
 	if !checkPodsRunningReady(c, ns, podNames, rebootPodReadyAgainTimeout) {
 		newPods := ps.List()
-		printStatusAndLogsForNotReadyPods(c, pods, newPods)
+		printStatusAndLogsForNotReadyPods(c, ns, podNames, newPods)
 		return false
 	}