def test_deletes_file(self):
with open(self.filename, 'w') as f:
f.write('data')
DeleteSourceFileSubscriber().on_done(self.future)
self.assertFalse(os.path.exists(self.filename))
self.future.set_exception.assert_not_called()
def test_sets_exception_on_error(self):
DeleteSourceFileSubscriber().on_done(self.future)
self.assertFalse(os.path.exists(self.filename))
call_args = self.future.set_exception.call_args[0]
self.assertIsInstance(call_args[0], EnvironmentError)
def _add_additional_subscribers(self, subscribers, fileinfo):
subscribers.append(ProvideSizeSubscriber(fileinfo.size))
if self._should_inject_content_type():
subscribers.append(ProvideUploadContentTypeSubscriber())
if self._cli_params.get('is_move', False):
subscribers.append(DeleteSourceFileSubscriber())