testrunner: refactor exception printing

This commit is contained in:
Cenk Gündoğan 2017-11-12 12:59:45 +01:00
parent bd52d33274
commit d537dd609f

View File

@ -21,6 +21,14 @@ RIOTBASE = os.environ['RIOTBASE'] or \
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.filename.startswith(PEXPECT_PATH)
)[-1]
return pos.line, \
os.path.relpath(os.path.abspath(pos.filename), RIOTBASE), \
pos.lineno
def run(testfunc, timeout=10, echo=True, traceback=False):
env = os.environ.copy()
child = pexpect.spawnu("make term", env=env, timeout=timeout)
@ -40,13 +48,9 @@ def run(testfunc, timeout=10, echo=True, traceback=False):
try:
testfunc(child)
except pexpect.TIMEOUT:
timeouted_at = list_until(extract_tb(sys.exc_info()[2]),
lambda frame:
frame.filename.startswith(PEXPECT_PATH))[-1]
line, filename, lineno = find_exc_origin(sys.exc_info()[2])
print("Timeout in expect script at \"%s\" (%s:%d)" %
(timeouted_at.line,
os.path.relpath(os.path.abspath(timeouted_at.filename), RIOTBASE),
timeouted_at.lineno))
(line, filename, lineno))
if traceback:
print_tb(sys.exc_info()[2])
return 1