github/kubernetes
1
0
Fork 0
mirror of https://github.com/k3s-io/kubernetes.git synced 2025-08-03 17:30:00 +00:00
Code Issues Projects Releases Wiki Activity

make pods e2e retry getting logs from web-sockets

Browse Source 
Signed-off-by: Vishnu kannan <vishnuk@google.com>
This commit is contained in:
Vishnu kannan 2016-08-30 17:11:48 -07:00
parent f839c5ffdf
commit 8d74371537
1 changed files with 21 additions and 18 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
test/e2e/common/pods.go
View File

@ -491,6 +491,7 @@ var _ = framework.KubeDescribe("Pods", func() {
defer ws.Close() defer ws.Close()
buf := &bytes.Buffer{} buf := &bytes.Buffer{}
Eventually(func() error {
for { for {
var msg []byte var msg []byte
if err := websocket.Message.Receive(ws, &msg); err != nil { if err := websocket.Message.Receive(ws, &msg); err != nil {
@ -508,11 +509,13 @@ var _ = framework.KubeDescribe("Pods", func() {
buf.Write(msg[1:]) buf.Write(msg[1:])
} }
if buf.Len() == 0 { if buf.Len() == 0 {
framework.Failf("Unexpected output from server") return fmt.Errorf("Unexpected output from server")
} }
if !strings.Contains(buf.String(), "nameserver") { 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() { It("should support retrieving logs from the container over websockets", func() {