Clean up

barseghyanartur · Dec 24, 2024 · 5ac587f · 5ac587f
1 parent 83ecb14
commit 5ac587f
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/src/faker_file/tests/sftp_server.py b/src/faker_file/tests/sftp_server.py
@@ -4,6 +4,7 @@
 import tempfile
 import threading
 from asyncio import Semaphore
+from contextlib import suppress
 from typing import Type
 
 import asyncssh
@@ -85,10 +86,8 @@ async def start_server_async(
     )
 
     async with server:
-        try:
+        with suppress(asyncio.CancelledError):
             await server.wait_closed()
-        except asyncio.CancelledError:
-            pass
 
 
 def start_server(host: str = SFTP_HOST, port: int = SFTP_PORT) -> None: