apply a timeout on test wait so if it hangs it fails fast

This commit is contained in:
Victor Shyba 2019-02-13 14:26:43 -03:00 committed by Lex Berezhny
parent 8108aa39c1
commit 119d498bce

View file

@ -98,5 +98,5 @@ class TestStreamDownloader(BlobExchangeTestBase):
self.server.server_protocol_class = ChunkedServerProtocol self.server.server_protocol_class = ChunkedServerProtocol
self.server.start_server(33333, '127.0.0.1') self.server.start_server(33333, '127.0.0.1')
self.assertEqual(0, len(self.client_blob_manager.completed_blob_hashes)) 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)) self.assertEqual(11, len(self.client_blob_manager.completed_blob_hashes))