def testOnSwarmingTaskTimeout(self, mocked_parse, mocked_result):
error = SwarmingTaskError.GenerateError(
swarming_task_error.RUNNER_TIMEOUT)
task_id = 'task_id'
task_data = {
'state': 'COMPLETE',
}
mocked_result.return_value = task_data, 'content', None
commit_position = 1000
isolate_sha = 'sha1'
iterations = 50
timeout_seconds = 1200
parameters = RunFlakeSwarmingTaskInput(builder_name='b',
commit_position=commit_position,
isolate_sha=isolate_sha,
iterations=iterations,
master_name='m',
reference_build_number=123,
step_name='s',
test_name='t',
timeout_seconds=timeout_seconds)
flake_swarming.OnSwarmingTaskTimeout(parameters, task_id)
mocked_parse.assert_called_once_with(task_data, 'content', error,
parameters)
def OnTimeout(self, pipeline_parameters, callback_parameters):
# TODO(crbug.com/835066): Capture metrics for pipeline timeouts.
super(RunFlakeSwarmingTaskPipeline,
self).OnTimeout(pipeline_parameters, callback_parameters)
task_id = callback_parameters.get('task_id')
return flake_swarming.OnSwarmingTaskTimeout(pipeline_parameters,
task_id)
def testOnSwarmingTaskTimeoutNoData(self, mocked_result):
error = SwarmingTaskError(code=1000, message='Unknown error')
mocked_result.return_value = None, None, error
task_id = 'task_id'
expected_result = FlakeSwarmingTaskOutput(task_id=task_id,
started_time=None,
completed_time=None,
iterations=None,
error=error,
pass_count=None)
self.assertEqual(expected_result,
flake_swarming.OnSwarmingTaskTimeout(None, task_id))
def testOnSwarmingTaskTimeoutNoTaskId(self, mocked_result):
error = SwarmingTaskError(
code=350, message='Runner to run swarming task timed out')
mocked_result.return_value = None, None, error
expected_result = FlakeSwarmingTaskOutput(task_id=None,
started_time=None,
completed_time=None,
iterations=None,
error=error,
pass_count=None)
self.assertEqual(expected_result,
flake_swarming.OnSwarmingTaskTimeout(None, None))