def test_during_first_execution(self):
task_id = "task_id"
mock_evg_api = MagicMock()
mock_evg_api.task_by_id.return_value.execution = 0
validate_service = under_test.GenTaskValidationService(mock_evg_api)
self.assertTrue(validate_service.should_task_be_generated(task_id))
mock_evg_api.task_by_id.assert_called_with(task_id,
fetch_all_executions=True)
def test_after_multiple_failed_execution(self):
mock_evg_api = MagicMock()
task_id = "task_id"
task = mock_evg_api.task_by_id.return_value
task.execution = 5
task.get_execution.return_value.is_success.return_value = False
validate_service = under_test.GenTaskValidationService(mock_evg_api)
self.assertTrue(validate_service.should_task_be_generated(task_id))
mock_evg_api.task_by_id.assert_called_with(task_id,
fetch_all_executions=True)