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))