make pods e2e retry getting logs from web-sockets

test/e2e/common/pods.go

@@ -491,6 +491,7 @@ var _ = framework.KubeDescribe("Pods", func() {
 		defer ws.Close()
 
 		buf := &bytes.Buffer{}
+		Eventually(func() error {
 			for {
 				var msg []byte
 				if err := websocket.Message.Receive(ws, &msg); err != nil {
@@ -508,11 +509,13 @@ var _ = framework.KubeDescribe("Pods", func() {
 				buf.Write(msg[1:])
 			}
 			if buf.Len() == 0 {
-			framework.Failf("Unexpected output from server")
+				return fmt.Errorf("Unexpected output from server")
 			}
 			if !strings.Contains(buf.String(), "nameserver") {
-			framework.Failf("Expected to find 'nameserver' in %q", buf.String())
+				return fmt.Errorf("Expected to find 'nameserver' in %q", buf.String())
 			}
+			return nil
+		}, time.Minute, 10*time.Second).Should(BeNil())
 	})
 
 	It("should support retrieving logs from the container over websockets", func() {