Add ReadTimeout for e2e tests, otherwise this can hang forever.

googleforgames · Sep 20, 2018 · 4b6aa88 · 4b6aa88
1 parent 0ecca13
commit 4b6aa88
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/test/e2e/framework/framework.go b/test/e2e/framework/framework.go
@@ -208,6 +208,11 @@ func PingGameServer(msg, address string) (reply string, err error) {
 		return "", errors.Wrapf(err, "Could not write message %s", msg)
 	}
 	b := make([]byte, 1024)
+
+	err = conn.SetReadDeadline(time.Now().Add(30 * time.Second))
+	if err != nil {
+		return "", err
+	}
 	n, err := conn.Read(b)
 	if err != nil {
 		return "", err