Try to suppress asyncio.CancelledError in a different way in test_streaming.py.

This commit is contained in:
Jonathan Moody 2022-07-09 06:12:32 -04:00 committed by Lex Berezhny
parent d0b5a0a8fd
commit c2d2080034

View file

@ -3,6 +3,7 @@ import hashlib
import aiohttp
import aiohttp.web
import asyncio
import contextlib
from lbry.file.source import ManagedDownloadSource
from lbry.utils import aiohttp_request
@ -353,7 +354,8 @@ class RangeRequests(CommandTestCase):
path = stream.full_path
self.assertIsNotNone(path)
if wait_for_start_writing:
await stream.started_writing.wait()
with contextlib.suppress(asyncio.CancelledError):
await stream.started_writing.wait()
self.assertTrue(os.path.isfile(path))
await self.daemon.file_manager.stop()
# while stopped, we get no response to query and no file is present
@ -365,7 +367,8 @@ class RangeRequests(CommandTestCase):
self.assertIsNotNone(stream.full_path)
self.assertEqual(stream.full_path, path)
if wait_for_start_writing:
await stream.started_writing.wait()
with contextlib.suppress(asyncio.CancelledError):
await stream.started_writing.wait()
self.assertTrue(os.path.isfile(path))
async def test_file_save_stop_before_finished_streaming_only_wait_for_start(self):