80 lines
2.3 KiB
Python
80 lines
2.3 KiB
Python
|
#!/usr/bin/env python3
|
||
|
|
|
||
|
|
import sys, os
|
||
|
|
import serial
|
||
|
|
import re
|
||
|
|
|
||
|
|
from await_flipper import flp_serial_by_name
|
||
|
|
|
||
|
|
|
||
|
|
LEAK_THRESHOLD = 3000 # added until units are fixed
|
||
|
|
|
||
|
|
|
||
|
|
def main():
|
||
|
|
flp_serial = flp_serial_by_name(sys.argv[1])
|
||
|
|
|
||
|
|
if flp_serial == "":
|
||
|
|
print("Name or serial port is invalid")
|
||
|
|
sys.exit(1)
|
||
|
|
|
||
|
|
with serial.Serial(flp_serial, timeout=1) as flipper:
|
||
|
|
flipper.baudrate = 230400
|
||
|
|
flipper.flushOutput()
|
||
|
|
flipper.flushInput()
|
||
|
|
|
||
|
|
flipper.timeout = 300
|
||
|
|
|
||
|
|
flipper.read_until(b">: ").decode("utf-8")
|
||
|
|
flipper.write(b"unit_tests\r")
|
||
|
|
data = flipper.read_until(b">: ").decode("utf-8")
|
||
|
|
|
||
|
|
lines = data.split("\r\n")
|
||
|
|
|
||
|
|
tests_re = r"Failed tests: \d{0,}"
|
||
|
|
time_re = r"Consumed: \d{0,}"
|
||
|
|
leak_re = r"Leaked: \d{0,}"
|
||
|
|
status_re = r"Status: \w{3,}"
|
||
|
|
|
||
|
|
tests_pattern = re.compile(tests_re)
|
||
|
|
time_pattern = re.compile(time_re)
|
||
|
|
leak_pattern = re.compile(leak_re)
|
||
|
|
status_pattern = re.compile(status_re)
|
||
|
|
|
||
|
|
tests, time, leak, status = None, None, None, None
|
||
|
|
|
||
|
|
for line in lines:
|
||
|
|
print(line)
|
||
|
|
if not tests:
|
||
|
|
tests = re.match(tests_pattern, line)
|
||
|
|
if not time:
|
||
|
|
time = re.match(time_pattern, line)
|
||
|
|
if not leak:
|
||
|
|
leak = re.match(leak_pattern, line)
|
||
|
|
if not status:
|
||
|
|
status = re.match(status_pattern, line)
|
||
|
|
|
||
|
|
if leak is None or time is None or leak is None or status is None:
|
||
|
|
print("Failed to get data. Or output is corrupt")
|
||
|
|
sys.exit(1)
|
||
|
|
|
||
|
|
leak = int(re.findall(r"[- ]\d+", leak.group(0))[0])
|
||
|
|
status = re.findall(r"\w+", status.group(0))[1]
|
||
|
|
tests = int(re.findall(r"\d+", tests.group(0))[0])
|
||
|
|
time = int(re.findall(r"\d+", time.group(0))[0])
|
||
|
|
|
||
|
|
if tests > 0 or leak > LEAK_THRESHOLD or status != "PASSED":
|
||
|
|
print(f"Got {tests} failed tests.")
|
||
|
|
print(f"Leaked {leak} bytes.")
|
||
|
|
print(f"Status by flipper: {status}")
|
||
|
|
print(f"Time elapsed {time/1000} seconds.")
|
||
|
|
sys.exit(1)
|
||
|
|
|
||
|
|
print(
|
||
|
|
f"Tests ran successfully! Time elapsed {time/1000} seconds. Passed {tests} tests."
|
||
|
|
)
|
||
|
|
sys.exit(0)
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
main()
|