def test_sets_exception_on_error(self):
exception = ValueError()
self.client.delete_object.side_effect = exception
DeleteSourceObjectSubscriber(self.client).on_done(self.future)
self.client.delete_object.assert_called_once_with(Bucket=self.bucket,
Key=self.key)
self.future.set_exception.assert_called_once_with(exception)
def _add_additional_subscribers(self, subscribers, fileinfo):
subscribers.append(ProvideSizeSubscriber(fileinfo.size))
subscribers.append(DirectoryCreatorSubscriber())
subscribers.append(ProvideLastModifiedTimeSubscriber(
fileinfo.last_update, self._result_queue))
if self._cli_params.get('is_move', False):
subscribers.append(DeleteSourceObjectSubscriber(
fileinfo.source_client))
def test_with_request_payer(self):
self.future.meta.call_args.extra_args = {'RequestPayer': 'requester'}
DeleteSourceObjectSubscriber(self.client).on_done(self.future)
self.client.delete_object.assert_called_once_with(
Bucket=self.bucket, Key=self.key, RequestPayer='requester')
def test_deletes_object(self):
DeleteSourceObjectSubscriber(self.client).on_done(self.future)
self.client.delete_object.assert_called_once_with(Bucket=self.bucket,
Key=self.key)
self.future.set_exception.assert_not_called()