diff --git a/tests/unit/stream/test_downloader.py b/tests/unit/stream/test_downloader.py index 3acd08717..d97444c0c 100644 --- a/tests/unit/stream/test_downloader.py +++ b/tests/unit/stream/test_downloader.py @@ -98,5 +98,5 @@ class TestStreamDownloader(BlobExchangeTestBase): self.server.server_protocol_class = ChunkedServerProtocol self.server.start_server(33333, '127.0.0.1') self.assertEqual(0, len(self.client_blob_manager.completed_blob_hashes)) - await self._test_transfer_stream(10) + await asyncio.wait_for(self._test_transfer_stream(10), timeout=2) self.assertEqual(11, len(self.client_blob_manager.completed_blob_hashes))