def setUp(self):
super(TestDownloadSpecialFilenameOutputManager, self).setUp()
self.osutil = AlwaysIndicatesSpecialFileOSUtils()
self.download_output_manager = DownloadSpecialFilenameOutputManager(
self.osutil,
self.transfer_coordinator,
io_executor=self.io_executor)
class TestDownloadSpecialFilenameOutputManager(BaseDownloadOutputManagerTest):
def setUp(self):
super(TestDownloadSpecialFilenameOutputManager, self).setUp()
self.osutil = AlwaysIndicatesSpecialFileOSUtils()
self.download_output_manager = DownloadSpecialFilenameOutputManager(
self.osutil,
self.transfer_coordinator,
io_executor=self.io_executor)
def test_is_compatible_for_special_file(self):
self.assertTrue(
self.download_output_manager.is_compatible(
self.filename, AlwaysIndicatesSpecialFileOSUtils()))
def test_is_not_compatible_for_non_special_file(self):
self.assertFalse(
self.download_output_manager.is_compatible(self.filename,
OSUtils()))
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'))
# Make sure the name of the file returned is the same as the
# final filename as we should not be writing to a temporary file.
self.assertEqual(f.name, self.filename)
def test_get_final_io_task(self):
self.assertIsInstance(self.download_output_manager.get_final_io_task(),
IOCloseTask)
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')])
class TestDownloadSpecialFilenameOutputManager(BaseDownloadOutputManagerTest):
def setUp(self):
super(TestDownloadSpecialFilenameOutputManager, self).setUp()
self.osutil = AlwaysIndicatesSpecialFileOSUtils()
self.download_output_manager = DownloadSpecialFilenameOutputManager(
self.osutil, self.transfer_coordinator,
io_executor=self.io_executor)
def test_is_compatible_for_special_file(self):
self.assertTrue(
self.download_output_manager.is_compatible(
self.filename, AlwaysIndicatesSpecialFileOSUtils()))
def test_is_not_compatible_for_non_special_file(self):
self.assertFalse(
self.download_output_manager.is_compatible(
self.filename, OSUtils()))
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'))
# Make sure the name of the file returned is the same as the
# final filename as we should not be writing to a temporary file.
self.assertEqual(f.name, self.filename)
def test_get_final_io_task(self):
self.assertIsInstance(
self.download_output_manager.get_final_io_task(), IOCloseTask)
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(TestDownloadSpecialFilenameOutputManager, self).setUp()
self.osutil = AlwaysIndicatesSpecialFileOSUtils()
self.download_output_manager = DownloadSpecialFilenameOutputManager(
self.osutil, self.transfer_coordinator,
io_executor=self.io_executor)