Browse Source

tests: improve clearing tracebacks from Qubes* objects

Clear also tracebacks of chained exceptions.
Marek Marczykowski-Górecki 6 years ago
parent
commit
35c66987ab
1 changed files with 4 additions and 1 deletions
  1. 4 1
      qubes/tests/__init__.py

+ 4 - 1
qubes/tests/__init__.py

@@ -391,7 +391,10 @@ class QubesTestCase(unittest.TestCase):
                 continue
             if exc_info is None:
                 continue
-            traceback.clear_frames(exc_info[2])
+            ex = exc_info[1]
+            while ex is not None:
+                traceback.clear_frames(ex.__traceback__)
+                ex = ex.__context__
 
     def cleanup_gc(self):
         gc.collect()