LBRYCommunity/lbry-sdk
1
0
Fork 1
Code Issues Pull requests Projects Releases 282 Packages Wiki Activity

improve test

Browse source
This commit is contained in:
Jack Robison 2019-05-10 14:57:04 -04:00
parent 648dbcc36b
commit ad2da24d64
No known key found for this signature in database
GPG key ID: DF25C68FE0239BB2
1 changed files with 6 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
tests/unit/stream/test_stream_manager.py
View file

@ -311,10 +311,14 @@ class TestStreamManager(BlobExchangeTestBase):
async def test_non_head_data_timeout(self): async def test_non_head_data_timeout(self):
await self.setup_stream_manager() await self.setup_stream_manager()
with open(os.path.join(self.server_dir, self.sd_hash), 'r') as sdf: with open(os.path.join(self.server_dir, self.sd_hash), 'r') as sdf:
head_blob_hash = json.loads(sdf.read())['blobs'][-2]['blob_hash'] last_blob_hash = json.loads(sdf.read())['blobs'][-2]['blob_hash']
self.server_blob_manager.delete_blob(head_blob_hash) self.server_blob_manager.delete_blob(last_blob_hash)
self.client_config.blob_download_timeout = 0.1 self.client_config.blob_download_timeout = 0.1
stream = await self.stream_manager.download_stream_from_uri(self.uri, self.exchange_rate_manager) stream = await self.stream_manager.download_stream_from_uri(self.uri, self.exchange_rate_manager)
await stream.started_writing.wait()
self.assertEqual('running', stream.status)
self.assertIsNotNone(stream.full_path)
self.assertGreater(stream.written_bytes, 0)
await stream.finished_write_attempt.wait() await stream.finished_write_attempt.wait()
self.assertEqual('stopped', stream.status) self.assertEqual('stopped', stream.status)
self.assertIsNone(stream.full_path) self.assertIsNone(stream.full_path)