def setUp(self):
super(TestLocalDeleteRequestSubmitter, self).setUp()
self.transfer_request_submitter = LocalDeleteRequestSubmitter(
self.transfer_manager, self.result_queue, self.cli_params)
self.file_creator = FileCreator()
def setUp(self):
super(TestLocalDeleteRequestSubmitter, self).setUp()
self.transfer_request_submitter = LocalDeleteRequestSubmitter(
self.transfer_manager, self.result_queue, self.cli_params)
self.file_creator = FileCreator()
class TestLocalDeleteRequestSubmitter(BaseTransferRequestSubmitterTest):
def setUp(self):
super(TestLocalDeleteRequestSubmitter, self).setUp()
self.transfer_request_submitter = LocalDeleteRequestSubmitter(
self.transfer_manager, self.result_queue, self.cli_params)
self.file_creator = FileCreator()
def tearDown(self):
super(TestLocalDeleteRequestSubmitter, self).tearDown()
self.file_creator.remove_all()
def test_can_submit(self):
fileinfo = FileInfo(src=self.filename,
dest=None,
operation_name='delete',
src_type='local')
self.assertTrue(self.transfer_request_submitter.can_submit(fileinfo))
fileinfo.operation_name = 'foo'
self.assertFalse(self.transfer_request_submitter.can_submit(fileinfo))
def test_cannot_submit_remote_deletes(self):
fileinfo = FileInfo(src=self.filename,
dest=None,
operation_name='delete',
src_type='s3')
self.assertFalse(self.transfer_request_submitter.can_submit(fileinfo))
def test_submit(self):
full_filename = self.file_creator.create_file(self.filename, 'content')
fileinfo = FileInfo(src=full_filename,
dest=None,
operation_name='delete',
src_type='local')
rval = self.transfer_request_submitter.submit(fileinfo)
self.assertTrue(rval)
queued_result = self.result_queue.get()
self.assertIsInstance(queued_result, QueuedResult)
self.assertEqual(queued_result.transfer_type, 'delete')
self.assertTrue(queued_result.src.endswith(self.filename))
self.assertIsNone(queued_result.dest)
self.assertEqual(queued_result.total_transfer_size, 0)
failure_result = self.result_queue.get()
self.assertIsInstance(failure_result, SuccessResult)
self.assertEqual(failure_result.transfer_type, 'delete')
self.assertTrue(failure_result.src.endswith(self.filename))
self.assertIsNone(failure_result.dest)
self.assertFalse(os.path.exists(full_filename))
def test_submit_with_exception(self):
fileinfo = FileInfo(src=self.filename,
dest=None,
operation_name='delete',
src_type='local')
# The file was never created so it should trigger an exception
# when it is attempted to be deleted in the submitter.
rval = self.transfer_request_submitter.submit(fileinfo)
self.assertTrue(rval)
queued_result = self.result_queue.get()
self.assertIsInstance(queued_result, QueuedResult)
self.assertEqual(queued_result.transfer_type, 'delete')
self.assertTrue(queued_result.src.endswith(self.filename))
self.assertIsNone(queued_result.dest)
self.assertEqual(queued_result.total_transfer_size, 0)
failure_result = self.result_queue.get()
self.assertIsInstance(failure_result, FailureResult)
self.assertEqual(failure_result.transfer_type, 'delete')
self.assertTrue(failure_result.src.endswith(self.filename))
self.assertIsNone(failure_result.dest)
def test_dry_run(self):
self.cli_params['dryrun'] = True
fileinfo = FileInfo(src=self.filename,
src_type='local',
dest=self.filename,
dest_type='local',
operation_name='delete')
self.transfer_request_submitter.submit(fileinfo)
result = self.result_queue.get()
self.assertIsInstance(result, DryRunResult)
self.assertEqual(result.transfer_type, 'delete')
self.assertTrue(result.src.endswith(self.filename))
self.assertIsNone(result.dest)
class TestLocalDeleteRequestSubmitter(BaseTransferRequestSubmitterTest):
def setUp(self):
super(TestLocalDeleteRequestSubmitter, self).setUp()
self.transfer_request_submitter = LocalDeleteRequestSubmitter(
self.transfer_manager, self.result_queue, self.cli_params)
self.file_creator = FileCreator()
def tearDown(self):
super(TestLocalDeleteRequestSubmitter, self).tearDown()
self.file_creator.remove_all()
def test_can_submit(self):
fileinfo = FileInfo(
src=self.filename, dest=None, operation_name='delete',
src_type='local')
self.assertTrue(
self.transfer_request_submitter.can_submit(fileinfo))
fileinfo.operation_name = 'foo'
self.assertFalse(
self.transfer_request_submitter.can_submit(fileinfo))
def test_cannot_submit_remote_deletes(self):
fileinfo = FileInfo(
src=self.filename, dest=None, operation_name='delete',
src_type='s3')
self.assertFalse(
self.transfer_request_submitter.can_submit(fileinfo))
def test_submit(self):
full_filename = self.file_creator.create_file(self.filename, 'content')
fileinfo = FileInfo(
src=full_filename, dest=None, operation_name='delete',
src_type='local')
rval = self.transfer_request_submitter.submit(fileinfo)
self.assertTrue(rval)
queued_result = self.result_queue.get()
self.assertIsInstance(queued_result, QueuedResult)
self.assertEqual(queued_result.transfer_type, 'delete')
self.assertTrue(queued_result.src.endswith(self.filename))
self.assertIsNone(queued_result.dest)
self.assertEqual(queued_result.total_transfer_size, 0)
failure_result = self.result_queue.get()
self.assertIsInstance(failure_result, SuccessResult)
self.assertEqual(failure_result.transfer_type, 'delete')
self.assertTrue(failure_result.src.endswith(self.filename))
self.assertIsNone(failure_result.dest)
self.assertFalse(os.path.exists(full_filename))
def test_submit_with_exception(self):
fileinfo = FileInfo(
src=self.filename, dest=None, operation_name='delete',
src_type='local')
# The file was never created so it should trigger an exception
# when it is attempted to be deleted in the submitter.
rval = self.transfer_request_submitter.submit(fileinfo)
self.assertTrue(rval)
queued_result = self.result_queue.get()
self.assertIsInstance(queued_result, QueuedResult)
self.assertEqual(queued_result.transfer_type, 'delete')
self.assertTrue(queued_result.src.endswith(self.filename))
self.assertIsNone(queued_result.dest)
self.assertEqual(queued_result.total_transfer_size, 0)
failure_result = self.result_queue.get()
self.assertIsInstance(failure_result, FailureResult)
self.assertEqual(failure_result.transfer_type, 'delete')
self.assertTrue(failure_result.src.endswith(self.filename))
self.assertIsNone(failure_result.dest)
def test_dry_run(self):
self.cli_params['dryrun'] = True
fileinfo = FileInfo(
src=self.filename, src_type='local',
dest=self.filename, dest_type='local',
operation_name='delete')
self.transfer_request_submitter.submit(fileinfo)
result = self.result_queue.get()
self.assertIsInstance(result, DryRunResult)
self.assertEqual(result.transfer_type, 'delete')
self.assertTrue(result.src.endswith(self.filename))
self.assertIsNone(result.dest)