platformio/platformio-core
1
0
Fork 1
mirror of https://github.com/platformio/platformio-core.git synced 2025-08-02 11:24:27 +02:00
Code Issues Packages Projects Releases Wiki Activity

Strip ANSI codes from Unity output

Browse Source
This commit is contained in:
Ivan Kravets
2022-05-04 18:56:57 +03:00
parent 517ee6532f
commit 3ed5d41df5
1 changed files with 13 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
platformio/test/runners/unity.py
View File

@@ -21,6 +21,7 @@ import click
from platformio.test.exception import UnitTestSuiteError from platformio.test.exception import UnitTestSuiteError
from platformio.test.runners.base import TestRunnerBase from platformio.test.runners.base import TestRunnerBase
from platformio.util import strip_ansi_codes
class UnityTestRunner(TestRunnerBase): class UnityTestRunner(TestRunnerBase):
@@ -249,6 +250,7 @@ void unityOutputComplete(void) { unittest_uart_end(); }
) )
def on_test_output(self, data): def on_test_output(self, data):
data = strip_ansi_codes(data or "")
if not data.strip(): if not data.strip():
return click.echo(data, nl=False) return click.echo(data, nl=False)
@@ -258,11 +260,18 @@ void unityOutputComplete(void) { unittest_uart_end(); }
# beautify output # beautify output
for line in data.strip().split("\n"): for line in data.strip().split("\n"):
line = line.strip() line = line.strip()
if line.endswith(":PASS"): if line.strip(".").endswith(":PASS"):
click.echo("%s\t[%s]" % (line[:-5], click.style("PASSED", fg="green")))
elif line.endswith(":IGNORE"):
click.echo( click.echo(
"%s\t[%s]" % (line[:-7], click.style("IGNORED", fg="yellow")) "%s\t[%s]"
% (line[: line.rindex(":PASS")], click.style("PASSED", fg="green"))
)
elif line.strip(".").endswith(":IGNORE"):
click.echo(
"%s\t[%s]"
% (
line[: line.rindex(":IGNORE")],
click.style("IGNORED", fg="yellow"),
)
) )
elif ":FAIL" in line: elif ":FAIL" in line:
click.echo("%s\t[%s]" % (line, click.style("FAILED", fg="red"))) click.echo("%s\t[%s]" % (line, click.style("FAILED", fg="red")))