trunner: prevent printing escape codes in output from fail

JIRA: CI-460
phoenix-rtos · Jun 12, 2024 · c49e59c · c49e59c
1 parent c28bc0c
commit c49e59c
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/trunner/types.py b/trunner/types.py
@@ -242,10 +242,12 @@ def _failed_traceback(self) -> List[str]:
         return [bold("ASSERTION TRACEBACK (most recent call last):"), "".join(tb_info)]
 
     def _failed_before_buffer(self, dut) -> str:
-        return dut.before.replace("\r", "")
+        before = re.sub(r'\x1b\[[0-9]*[A-Za-z]', '', dut.before)
+        return before.replace("\r", "")
 
     def _failed_buffer(self, dut) -> str:
-        return dut.buffer.replace("\r", "")
+        buffer = re.sub(r'\x1b\[[0-9]*[A-Zaq-z]', '', dut.buffer)
+        return buffer.replace("\r", "")
 
     def fail_pexpect(self, dut, exc):
         self.fail(summary="Pexpect failure")