class TestDownloadFilenameOutputManager(BaseDownloadOutputManagerTest):
def setUp(self):
super(TestDownloadFilenameOutputManager, self).setUp()
self.download_output_manager = DownloadFilenameOutputManager(
self.osutil,
self.transfer_coordinator,
io_executor=self.io_executor)
def test_is_compatible(self):
self.assertTrue(
self.download_output_manager.is_compatible(self.filename))
def test_get_download_task_tag(self):
self.assertIsNone(self.download_output_manager.get_download_task_tag())
def test_get_fileobj_for_io_writes(self):
with self.download_output_manager.get_fileobj_for_io_writes(
self.future) as f:
# Ensure it is a file like object returned
self.assertTrue(hasattr(f, 'read'))
self.assertTrue(hasattr(f, 'seek'))
# Make sure the name of the file returned is not the same as the
# final filename as we should be writing to a temporary file.
self.assertNotEqual(f.name, self.filename)
def test_get_final_io_task(self):
ref_contents = b'my_contents'
with self.download_output_manager.get_fileobj_for_io_writes(
self.future) as f:
temp_filename = f.name
# Write some data to test that the data gets moved over to the
# final location.
f.write(ref_contents)
final_task = self.download_output_manager.get_final_io_task()
# Make sure it is the appropriate task.
self.assertIsInstance(final_task, IORenameFileTask)
final_task()
# Make sure the temp_file gets removed
self.assertFalse(os.path.exists(temp_filename))
# Make sure what ever was written to the temp file got moved to
# the final filename
with open(self.filename, 'rb') as f:
self.assertEqual(f.read(), ref_contents)
def test_can_queue_file_io_task(self):
fileobj = WriteCollector()
self.download_output_manager.queue_file_io_task(fileobj=fileobj,
data='foo',
offset=0)
self.download_output_manager.queue_file_io_task(fileobj=fileobj,
data='bar',
offset=3)
self.io_executor.shutdown()
self.assertEqual(fileobj.writes, [(0, 'foo'), (3, 'bar')])
def setUp(self):
super(TestDownloadFilenameOutputManager, self).setUp()
self.download_output_manager = DownloadFilenameOutputManager(
self.osutil,
self.transfer_coordinator,
io_executor=self.io_executor)