StarMirror/py-libp2p
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Suppress all exceptions in clean up.

Browse Source
This commit is contained in:
mhchia 2019-09-09 22:52:47 +08:00
parent e5eb01d22b
commit a45eb76421
No known key found for this signature in database
GPG Key ID: 389EFBEA1362589A
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
tests/utils.py
View File

@ -17,6 +17,7 @@ async def connect(node1, node2):
await node1.connect(info) await node1.connect(info)
# FIXME: Should be deprecated, since it also kills the main task.
async def cleanup(): async def cleanup():
pending = asyncio.all_tasks() pending = asyncio.all_tasks()
for task in pending: for task in pending:
@ -24,7 +25,9 @@ async def cleanup():
# Now we should await task to execute it's cancellation. # Now we should await task to execute it's cancellation.
# Cancelled task raises asyncio.CancelledError that we can suppress: # Cancelled task raises asyncio.CancelledError that we can suppress:
with suppress(asyncio.CancelledError): # NOTE: Changed from `asyncio.CancelledError` to `Exception`, to suppress all exceptions
# including the one in `run_until_complete`.
with suppress(Exception):
await task await task