Finish fix and text for traceback error

HypothesisWorks · Nov 14, 2022 · 6760241 · 6760241
1 parent 7d1c3ec
commit 6760241
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 3 deletions.
diff --git a/hypothesis-python/src/hypothesis/internal/escalation.py b/hypothesis-python/src/hypothesis/internal/escalation.py
@@ -67,12 +67,12 @@ def escalate_hypothesis_internal_error():
     if PREVENT_ESCALATION:
         return
 
-    error_type, e, tb = sys.exc_info()
+    _, e, tb = sys.exc_info()
 
     if getattr(e, "hypothesis_internal_never_escalate", False):
         return
 
-    filepath = traceback.extract_tb(tb)[-1][0]
+    filepath = None if tb is None else traceback.extract_tb(tb)[-1][0]
     if is_hypothesis_file(filepath) and not isinstance(
         e, (HypothesisException,) + HYPOTHESIS_CONTROL_EXCEPTIONS
     ):
@@ -113,7 +113,10 @@ def get_interesting_origin(exception):
     # blocks and understand the __cause__ (`raise x from y`) or __context__ that
     # first raised an exception as well as PEP-654 exception groups.
     tb = get_trimmed_traceback(exception)
-    filename, lineno, *_ = traceback.extract_tb(tb)[-1]
+    if tb is None:
+        filename, lineno = None, None
+    else:
+        filename, lineno, *_ = traceback.extract_tb(tb)[-1]
     return (
         type(exception),
         filename,

diff --git a/hypothesis-python/tests/cover/test_escalation.py b/hypothesis-python/tests/cover/test_escalation.py
@@ -70,3 +70,7 @@ def test_multiplefailures_deprecation():
     with pytest.warns(errors.HypothesisDeprecationWarning):
         exc = errors.MultipleFailures
     assert exc is BaseExceptionGroup
+
+
+def test_handles_null_traceback():
+    esc.get_interesting_origin(Exception())