def save(self, commit=True): instance = super().save() youtube_dl.apply_async([instance.pk]) return instance
def test_task_cancellation_attr_error(self, mock_deepcopy, mock_remove): mock_deepcopy.side_effect = SystemExit() youtube_dl.apply_async([self.download_task.id]) mock_remove.assert_not_called()
def test_task_other_exception(self, mock_download, mock_log): mock_download.side_effect = ValueError() self.assertRaises(ValueError, lambda: youtube_dl.apply_async([self.download_task.id])) self.assertEqual(self.download_task.tracebacks.count(), 1)
def test_database_exception(self, mock_log): self.assertRaises(ObjectDoesNotExist, lambda: youtube_dl.apply_async([0])) mock_log.exception.assert_called_with('Exception while updating DownloadTask. id=0')