def testCallbackImplFailedRun(self, _):
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)
result = pipeline_job.CallbackImpl(run_flake_swarming_task_input,
{'task_id': 'task_id'})
self.assertEqual(('Error getting swarming task result: m', None), result)
def testCallbackImplNoTaskID(self, mocked_pipeline_id, mock_fn):
mocked_pipeline_id.__get__ = mock.Mock(return_value='pipeline-id')
pipeline_input = RunFlakeSwarmingTaskInput.FromSerializable({})
p = RunFlakeSwarmingTaskPipeline(pipeline_input)
result = p.CallbackImpl(pipeline_input, {})
self.assertEqual(('Task_id not found for pipeline pipeline-id', None),
result)
self.assertFalse(mock_fn.called)
def testCallbackImplCompleted(self, mocked_output):
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)
flake_swarming_task_output = FlakeSwarmingTaskOutput(
completed_time=datetime(2018, 4, 1, 0, 1, 0),
error=None,
iterations=iterations,
pass_count=iterations,
started_time=datetime(2018, 4, 1, 0, 0, 0),
task_id='task_id')
mocked_output.return_value = flake_swarming_task_output
pipeline_job = RunFlakeSwarmingTaskPipeline(run_flake_swarming_task_input)
result = pipeline_job.CallbackImpl(run_flake_swarming_task_input,
{'task_id': 'task_id'})
self.assertEqual((None, flake_swarming_task_output), result)