def test_task_cancellation_success(self, mock_deepcopy, mock_remove):
mock_deepcopy.side_effect = SystemExit()
youtube_dl._last_status = {
'tmpfilename': 'some_tmpfilename'
}
youtube_dl(self.download_task.id)
del youtube_dl._last_status
mock_remove.assert_called_with('some_tmpfilename')
def test_task_cancellation_other_exception(self, mock_deepcopy, mock_remove, mock_log):
mock_deepcopy.side_effect = SystemExit()
mock_remove.side_effect = ValueError()
youtube_dl._last_status = {
'tmpfilename': 'some_tmpfilename'
}
self.assertRaises(ValueError, lambda: youtube_dl(self.download_task.id))
del youtube_dl._last_status
mock_log.exception.assert_called_with('Exception while removing temporary file.'
' id=1, tempfilename=some_tmpfilename')
