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

add some unit tests for BlobFile.close()

Browse source
This commit is contained in:
Kay Kurokawa 2017-09-13 14:01:53 -04:00 committed by Jack Robison
parent e92321a9c1
commit f816cc1637
No known key found for this signature in database
GPG key ID: 284699E7404E3CFF
1 changed files with 21 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

21
tests/unit/core/test_HashBlob.py
View file

@ -30,6 +30,7 @@ class BlobFileTest(unittest.TestCase):
writer, finished_d = blob_file.open_for_writing(peer=1)
writer.write(self.fake_content)
writer.close()
out = yield finished_d
self.assertTrue(isinstance(out,HashBlob))
self.assertTrue(out.verified)
@ -83,6 +84,26 @@ class BlobFileTest(unittest.TestCase):
writer.write(content)
yield self.assertFailure(finished_d, InvalidDataError)
@defer.inlineCallbacks
def test_close_on_incomplete_write(self):
# write all but 1 byte of data,
blob_file = BlobFile(self.blob_dir, self.fake_content_hash, self.fake_content_len)
writer, finished_d = blob_file.open_for_writing(peer=1)
writer.write(self.fake_content[:self.fake_content_len-1])
writer.close()
yield self.assertFailure(finished_d, DownloadCanceledError)
# writes after close will throw a ValueError exception
with self.assertRaises(ValueError):
writer.write(self.fake_content)
# another call to close will do nothing
writer.close()
# file should not exist, since we did not finish write
blob_file_2 = BlobFile(self.blob_dir, self.fake_content_hash, self.fake_content_len)
out = blob_file_2.open_for_reading()
self.assertEqual(None, out)
@defer.inlineCallbacks
def test_multiple_writers(self):