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)