112 lines
3 KiB
Python
112 lines
3 KiB
Python
#!/usr/bin/env python
|
|
# -*- coding: utf-8 -*-
|
|
|
|
# This script reads specified serial port for test output. If the test is
|
|
# successfully finished the script returns 0 otherwise -1. The script also
|
|
# checks for time-out. The timeout value is received from the test itself
|
|
# during test initialization.
|
|
#
|
|
# The script runs as many instances as many serial devices provided.
|
|
#
|
|
# Usage:
|
|
# > run_test.py /dev/ttyUSB0
|
|
# or
|
|
# > run_test.py /dev/ttyUSB0 /dev/ttyUSB1 /dev/ttyUSB2
|
|
# if three devices are connected and should be tested simultaneously
|
|
#
|
|
|
|
|
|
import serial
|
|
import re
|
|
import sys
|
|
import time
|
|
import threading
|
|
|
|
|
|
class Test(threading.Thread):
|
|
def __init__(self, port, baudrate):
|
|
threading.Thread.__init__(self)
|
|
print("init: {}".format(port))
|
|
self._port = serial.Serial(port, baudrate, timeout=1)
|
|
self._timeout = None
|
|
self.result = None
|
|
|
|
def wait_start(self):
|
|
skip_lines = 100
|
|
while skip_lines > 0:
|
|
self._port.write('START\n'.encode())
|
|
l = self._port.readline()
|
|
if 'TEST_INIT: TIMEOUT=' in l:
|
|
res = re.findall(r'\d+', l)
|
|
if len(res) != 1:
|
|
print("Invalid timeout format")
|
|
self.result = False
|
|
return
|
|
else:
|
|
self._timeout = int(res[0])
|
|
break
|
|
else:
|
|
sys.stdout.write(l)
|
|
skip_lines -= 1
|
|
if self._timeout is None:
|
|
print("Device is not responding")
|
|
self.result = False
|
|
else:
|
|
print("Timeout: {}".format(self._timeout))
|
|
self.result = True
|
|
|
|
def test(self):
|
|
end_time = time.time() + self._timeout
|
|
result = None
|
|
while end_time > time.time():
|
|
l = self._port.readline()
|
|
if 'TEST_FINISH: ' in l:
|
|
if 'SUCCESS' in l:
|
|
print("Test finished successfully")
|
|
result = True
|
|
break
|
|
elif 'FAIL' in l:
|
|
print("Test failed")
|
|
result = False
|
|
else:
|
|
print("Unknown status")
|
|
result = False
|
|
else:
|
|
if l is not None:
|
|
sys.stdout.write(l)
|
|
if result is None:
|
|
print("Test timeout")
|
|
return result
|
|
|
|
def run(self):
|
|
self.wait_start()
|
|
if self.result is not None:
|
|
self.result = self.test()
|
|
|
|
def __del__(self):
|
|
self._port.close()
|
|
|
|
|
|
def main():
|
|
if len(sys.argv) <= 1:
|
|
print("Need at least one serial device to test a device")
|
|
exit(-1)
|
|
|
|
tasks = []
|
|
for dev in sys.argv[1:]:
|
|
t = Test(dev, 115200)
|
|
t.start()
|
|
tasks.append(t)
|
|
|
|
result = True
|
|
for t in tasks:
|
|
t.join()
|
|
result = result and t.result
|
|
|
|
if result:
|
|
print("Test: SUCCESS")
|
|
else:
|
|
print("Test: FAIL")
|
|
|
|
if __name__ == "__main__":
|
|
main()
|