commit 9e55c0001f9658b131ebbceccddf082787d162c9
author Yuexi Ma <yuexima@google.com> Tue Nov 19 15:39:35 2024 -0800
committer Yuexi Ma <yuexima@google.com> Wed Nov 20 10:15:26 2024 -0800
tree 421f1ca06109c41ff0c6f37b6a269be6611d4ad1
parent b1f6d8d8eac7bc91a11d06f7c1ce5776b3d0e29a

Intercept stderr as well during rolling output

Test: atest unittests
Change-Id: Ic9c10b056f45e46fb991c11979f6a85cd8f0b879

atest/atest_utils.py

diff --git a/atest/atest_utils.py b/atest/atest_utils.py
index 2bc306d..d8266b8 100644
--- a/atest/atest_utils.py
+++ b/atest/atest_utils.py
@@ -327,6 +327,7 @@
     io_output.flush()
 
   original_stdout = sys.stdout
+  original_stderr = sys.stderr
 
   lock = threading.Lock()
 
@@ -351,6 +352,7 @@
       original_stdout.flush()
 
   sys.stdout = SafeStdout()
+  sys.stderr = sys.stdout
 
   for line in iter(io_input.readline, ''):
     if not line:
@@ -376,6 +378,7 @@
     io_output.flush()
 
   sys.stdout = original_stdout
+  sys.stderr = original_stderr
 
   io_input.close()