Extend tests for stackoverflow profile

distributed/profile.py

@@ -160,14 +160,15 @@ def process(
         # which can cause recursion errors later on since this can generate
         # deeply nested dictionaries
         depth = min(250, sys.getrecursionlimit() // 4)
-    if depth <= 0:
-        return None
+
     if any(frame.f_code.co_filename.endswith(o) for o in omit):
         return None
 
     prev = frame.f_back
-    if prev is not None and (
-        stop is None or not prev.f_code.co_filename.endswith(stop)
+    if (
+        depth > 0
+        and prev is not None
+        and (stop is None or not prev.f_code.co_filename.endswith(stop))
     ):
         new_state = process(prev, frame, state, stop=stop, depth=depth - 1)
         if new_state is None:

distributed/tests/test_profile.py

@@ -356,23 +356,19 @@ def test_call_stack_f_lineno(f_lasti: int, f_lineno: int) -> None:
 
 
 def test_stack_overflow():
-    old = sys.getrecursionlimit()
-    sys.setrecursionlimit(200)
-    try:
-        state = create()
-        frame = None
-
-        def f(i):
-            if i == 0:
-                nonlocal frame
-                frame = sys._current_frames()[threading.get_ident()]
-                return
-            else:
-                return f(i - 1)
-
-        f(sys.getrecursionlimit() - 100)
-        process(frame, None, state)
-        merge(state, state, state)
-
-    finally:
-        sys.setrecursionlimit(old)
+    state = create()
+    frame = None
+
+    def f(i):
+        if i == 0:
+            nonlocal frame
+            frame = sys._current_frames()[threading.get_ident()]
+            return
+        else:
+            return f(i - 1)
+
+    f(sys.getrecursionlimit() - 200)
+    process(frame, None, state)
+    assert state["children"]
+    assert state["count"]
+    assert merge(state, state, state)