def testRunImplNotTriggerSameTaskTwice(self, mock_trigger, _): master_name = 'm' builder_name = 'b' build_number = 123 step_name = 's' test_name = 't' commit_position = 1000 isolate_sha = 'sha1' iterations = 50 timeout_seconds = 1200 run_flake_swarming_task_input = RunFlakeSwarmingTaskInput( builder_name=builder_name, commit_position=commit_position, isolate_sha=isolate_sha, iterations=iterations, master_name=master_name, reference_build_number=build_number, step_name=step_name, test_name=test_name, timeout_seconds=timeout_seconds) pipeline_job = RunFlakeSwarmingTaskPipeline(run_flake_swarming_task_input) pipeline_job.RunImpl(run_flake_swarming_task_input) self.assertFalse(mock_trigger.called)
def testRunImplTriggerTaskFailed(self, mocked_pipeline_id, mock_trigger, _): master_name = 'm' builder_name = 'b' build_number = 123 step_name = 's' test_name = 't' commit_position = 1000 isolate_sha = 'sha1' iterations = 50 timeout_seconds = 1200 pipeline_id = 'pipeline-id' run_flake_swarming_task_input = RunFlakeSwarmingTaskInput( builder_name=builder_name, commit_position=commit_position, isolate_sha=isolate_sha, iterations=iterations, master_name=master_name, reference_build_number=build_number, step_name=step_name, test_name=test_name, timeout_seconds=timeout_seconds) mock_trigger.return_value = None mocked_pipeline_id.__get__ = mock.Mock(return_value=pipeline_id) pipeline_job = RunFlakeSwarmingTaskPipeline(run_flake_swarming_task_input) with self.assertRaises(pipeline.Retry): pipeline_job.RunImpl(run_flake_swarming_task_input) mock_trigger.assert_called_once_with( master_name=master_name, builder_name=builder_name, reference_build_number=build_number, step_name=step_name, test_name=test_name, isolate_sha=isolate_sha, iterations=iterations, timeout_seconds=timeout_seconds, runner_id=pipeline_id)