def test_schedule_exception_handler__no_raise(self): # no raise process_id = uniqid() schedule_id = '%s%s' % (uniqid(), uniqid()) with schedule.schedule_exception_handler(process_id, schedule_id): pass Status.objects.filter.assert_not_called() PipelineProcess.objects.get.assert_not_called() schedule.delete_parent_data.assert_not_called()
def test_schedule_exception_handler__raise_and_not_find_corresponding_status(self): e = Exception() process = MockPipelineProcess() process_id = uniqid() schedule_id = '%s%s' % (uniqid(), uniqid()) with mock.patch(PIPELINE_PROCESS_GET, mock.MagicMock(return_value=process)): with schedule.schedule_exception_handler(process_id, schedule_id): raise e Status.objects.filter.assert_called_once_with(id=schedule_id[:32], version=schedule_id[32:]) PipelineProcess.objects.get.assert_not_called() process.exit_gracefully.assert_not_called() schedule.delete_parent_data.assert_called_once_with(schedule_id)