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

fix test

Browse source
This commit is contained in:
Jack Robison 2019-08-12 16:29:22 -04:00
parent 8871a611f9
commit a288adbfc6
No known key found for this signature in database
GPG key ID: DF25C68FE0239BB2
1 changed files with 23 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

29
lbry/tests/unit/blob/test_blob_file.py
View file

@ -10,6 +10,22 @@ from lbry.blob.blob_manager import BlobManager
from lbry.blob.blob_file import BlobFile, BlobBuffer, AbstractBlob from lbry.blob.blob_file import BlobFile, BlobBuffer, AbstractBlob
class MockTransport(asyncio.Transport):
def __init__(self):
self.closed = asyncio.Event()
self.paused = asyncio.Event()
self._extra = {}
def close(self):
self.closed.set()
def pause_reading(self) -> None:
self.paused.set()
def resume_reading(self) -> None:
self.paused.clear()
class TestBlob(AsyncioTestCase): class TestBlob(AsyncioTestCase):
blob_hash = "7f5ab2def99f0ddd008da71db3a3772135f4002b19b7605840ed1034c8955431bd7079549e65e6b2a3b9c17c773073ed" blob_hash = "7f5ab2def99f0ddd008da71db3a3772135f4002b19b7605840ed1034c8955431bd7079549e65e6b2a3b9c17c773073ed"
blob_bytes = b'1' * ((2 * 2 ** 20) - 1) blob_bytes = b'1' * ((2 * 2 ** 20) - 1)
@ -41,7 +57,7 @@ class TestBlob(AsyncioTestCase):
async def _test_close_writers_on_finished(self, blob_class=AbstractBlob, blob_directory=None): async def _test_close_writers_on_finished(self, blob_class=AbstractBlob, blob_directory=None):
blob = self._get_blob(blob_class, blob_directory=blob_directory) blob = self._get_blob(blob_class, blob_directory=blob_directory)
writers = [blob.get_blob_writer('1.2.3.4', port) for port in range(5)] writers = [blob.get_blob_writer(MockTransport()) for _ in range(5)]
self.assertEqual(5, len(blob.writers)) self.assertEqual(5, len(blob.writers))
# test that writing too much causes the writer to fail with InvalidDataError and to be removed # test that writing too much causes the writer to fail with InvalidDataError and to be removed
@ -137,14 +153,15 @@ class TestBlob(AsyncioTestCase):
blob_directory = tempfile.mkdtemp() blob_directory = tempfile.mkdtemp()
self.addCleanup(lambda: shutil.rmtree(blob_directory)) self.addCleanup(lambda: shutil.rmtree(blob_directory))
blob = self._get_blob(BlobBuffer, blob_directory=blob_directory) blob = self._get_blob(BlobBuffer, blob_directory=blob_directory)
writer = blob.get_blob_writer('1.1.1.1', 1337) transport = MockTransport()
writer = blob.get_blob_writer(transport)
self.assertEqual(1, len(blob.writers)) self.assertEqual(1, len(blob.writers))
with self.assertRaises(OSError): with self.assertRaises(OSError):
blob.get_blob_writer('1.1.1.1', 1337) blob.get_blob_writer(transport)
writer.close_handle() writer.close_handle()
self.assertTrue(blob.writers[('1.1.1.1', 1337)].closed()) self.assertTrue(blob.writers[(transport)].closed())
writer = blob.get_blob_writer('1.1.1.1', 1337) writer = blob.get_blob_writer(transport)
self.assertEqual(blob.writers[('1.1.1.1', 1337)], writer) self.assertEqual(blob.writers[transport], writer)
writer.close_handle() writer.close_handle()
await asyncio.sleep(0.000000001) # flush callbacks await asyncio.sleep(0.000000001) # flush callbacks
self.assertEqual(0, len(blob.writers)) self.assertEqual(0, len(blob.writers))