assure we flush the stdin/stderr

Signed-off-by: Sorin Sbarnea <[email protected]>
pycontribs · Feb 18, 2018 · 5f6afd4 · 5f6afd4
1 parent 44e21c6
commit 5f6afd4
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/rtox/rtox.py b/rtox/rtox.py
@@ -52,6 +52,13 @@ def run(self, command):
 
                 time.sleep(0.1)
 
+            if channel.recv_ready():
+                length = len(channel.in_buffer)
+                sys.stdout.write(channel.recv(length))
+            if channel.recv_stderr_ready():
+                length = len(channel.in_stderr_buffer)
+                sys.stderr.write(channel.recv_stderr(length))
+
             return channel.recv_exit_status()
         except KeyboardInterrupt:
             channel.close()