From d537dd609f5aaabc7abcabf1ab0dcdb4540c2bd9 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Cenk=20G=C3=BCndo=C4=9Fan?= <mail-github@cgundogan.de>
Date: Sun, 12 Nov 2017 12:59:45 +0100
Subject: [PATCH] testrunner: refactor exception printing

---
 dist/tools/testrunner/testrunner.py | 16 ++++++++++------
 1 file changed, 10 insertions(+), 6 deletions(-)

diff --git a/dist/tools/testrunner/testrunner.py b/dist/tools/testrunner/testrunner.py
index 5b2b73e20c..471c93253d 100755
--- a/dist/tools/testrunner/testrunner.py
+++ b/dist/tools/testrunner/testrunner.py
@@ -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
-- 
GitLab