mirror of
https://github.com/RIOT-OS/RIOT.git
synced 2024-12-29 04:50:03 +01:00
410778303e
The object-like access to frame information in the traceback was only introduced in Python 3.5. Before that version it was a 4-tuple [[1]]. The indexed way to access the frame seems to be upwards-compatible for newer versions (tested with python 3.5, maybe some of the Arch crew can test with even newer versions), so I used that one. [1]: https://docs.python.org/3.4/library/traceback.html#traceback.extract_tb
74 lines
2.3 KiB
Python
Executable File
74 lines
2.3 KiB
Python
Executable File
# Copyright (C) 2017 Cenk Gündoğan <cenk.guendogan@haw-hamburg.de>
|
|
# 2016 Kaspar Schleiser <kaspar@schleiser.de>
|
|
# 2014 Martine Lenders <mlenders@inf.fu-berlin.de>
|
|
#
|
|
# This file is subject to the terms and conditions of the GNU Lesser
|
|
# General Public License v2.1. See the file LICENSE in the top level
|
|
# directory for more details.
|
|
|
|
import os
|
|
import signal
|
|
import sys
|
|
import subprocess
|
|
import time
|
|
from traceback import extract_tb, print_tb
|
|
|
|
import pexpect
|
|
|
|
PEXPECT_PATH = os.path.dirname(pexpect.__file__)
|
|
RIOTBASE = os.environ['RIOTBASE'] or \
|
|
os.path.abspath(os.path.join(os.path.dirname(__file__), "..", "..", ".."))
|
|
|
|
def list_until(l, cond):
|
|
return l[:([i for i, e in enumerate(l) if cond(e)][0])]
|
|
|
|
def find_exc_origin(exc_info):
|
|
pos = list_until(extract_tb(exc_info),
|
|
lambda frame: frame[0].startswith(PEXPECT_PATH)
|
|
)[-1]
|
|
return pos[3], \
|
|
os.path.relpath(os.path.abspath(pos[0]), RIOTBASE), \
|
|
pos[1]
|
|
|
|
def run(testfunc, timeout=10, echo=True, traceback=False):
|
|
env = os.environ.copy()
|
|
child = pexpect.spawnu("make term", env=env, timeout=timeout)
|
|
|
|
# on many platforms, the termprog needs a short while to be ready...
|
|
time.sleep(3)
|
|
|
|
if echo:
|
|
child.logfile = sys.stdout
|
|
|
|
try:
|
|
subprocess.check_output(('make', 'reset'), env=env,
|
|
stderr=subprocess.PIPE)
|
|
except subprocess.CalledProcessError:
|
|
# make reset yields error on some boards even if successful
|
|
pass
|
|
try:
|
|
testfunc(child)
|
|
except pexpect.TIMEOUT:
|
|
line, filename, lineno = find_exc_origin(sys.exc_info()[2])
|
|
print("Timeout in expect script at \"%s\" (%s:%d)" %
|
|
(line, filename, lineno))
|
|
if traceback:
|
|
print_tb(sys.exc_info()[2])
|
|
return 1
|
|
except pexpect.EOF:
|
|
line, filename, lineno = find_exc_origin(sys.exc_info()[2])
|
|
print("Unexpected end of file in expect script at \"%s\" (%s:%d)" %
|
|
(line, filename, lineno))
|
|
if traceback:
|
|
print_tb(sys.exc_info()[2])
|
|
return 1
|
|
finally:
|
|
print("")
|
|
try:
|
|
os.killpg(os.getpgid(child.pid), signal.SIGKILL)
|
|
except ProcessLookupError:
|
|
print("Process already stopped")
|
|
|
|
child.close()
|
|
return 0
|