change generator exit test

dlt-hub · Jan 30, 2024 · 8cf3c3c · 8cf3c3c
1 parent 2a168e5
commit 8cf3c3c
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/tests/extract/test_extract_pipe.py b/tests/extract/test_extract_pipe.py
@@ -718,6 +718,8 @@ async def long_gen():
             close_pipe_yielding = False
         except GeneratorExit:
             close_pipe_got_exit = True
+        except asyncio.CancelledError:
+            close_pipe_got_exit = True
 
     # execute in a thread
     async def raise_gen(item: int):