Avoid an infinite loop in console test

tcalmant · Aug 16, 2017 · 199d6c9 · 199d6c9
1 parent 6c0fc06
commit 199d6c9
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/tests/shell/test_console.py b/tests/shell/test_console.py
@@ -68,7 +68,12 @@ def test_echo(self):
             # Wait for prompt
             got = ""
             while ps1 not in got:
-                got += to_str(process.stdout.read(1))
+                char = to_str(process.stdout.read(1))
+                if not char:
+                    self.fail("Can't read from stdout ({})"
+                              .format(process.returncode))
+                else:
+                    got += char
 
             try:
                 # Try echoing