Fix generator wrapper of concurrency guard (#995)

neo4j · Nov 28, 2023 · 3d1dc5b · 3d1dc5b
1 parent 484bfdc
commit 3d1dc5b
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git a/src/neo4j/_async/_debug/_concurrency_check.py b/src/neo4j/_async/_debug/_concurrency_check.py
@@ -131,6 +131,8 @@ async def inner(*args, **kwargs):
                     if acquired:
                         try:
                             item = await iter_.__anext__()
+                        except StopAsyncIteration:
+                            return
                         finally:
                             async with self.__tracebacks_lock:
                                 self.__tracebacks.pop()

diff --git a/src/neo4j/_sync/_debug/_concurrency_check.py b/src/neo4j/_sync/_debug/_concurrency_check.py
@@ -131,6 +131,8 @@ def inner(*args, **kwargs):
                     if acquired:
                         try:
                             item = iter_.__next__()
+                        except StopIteration:
+                            return
                         finally:
                             with self.__tracebacks_lock:
                                 self.__tracebacks.pop()