def test_task_failure(self, jm): task = mock.MagicMock() task.request.parent_id = None task.job_manager = jm(**self.headers) exc, tb = mock.MagicMock(), mock.MagicMock() exc.__str__.return_value = 'MOCKEXCEPTION' with mock.patch('girder_worker.app.tb') as traceback: traceback.format_tb.return_value = 'TRACEBACK' gw_task_failure(sender=task, exception=exc, traceback=tb) task.job_manager.write.assert_called_once_with( 'MagicMock: MOCKEXCEPTION\nTRACEBACK') task.job_manager.updateStatus(JobStatus.ERROR)
def test_gw_task_failure_with_builtin_tasks_should_noop(name): with mock.patch('girder_worker.app._update_status') as us: task = mock.MagicMock() task.name = name gw_task_failure(sender=task) us.assert_not_called()