def create_task_status_update(task_id,
agent_id,
status,
when,
exit_code=None,
reason=None,
source=None,
message=None,
data=None):
"""Creates a job model for unit testing
:param task_id: The unique ID of the task
:type task_id: string
:param agent_id: The agent ID for the task
:type agent_id: string
:param status: The status of the task
:type status: string
:param when: The timestamp of the update
:type when: :class:`datetime.datetime`
:param exit_code: The task's exit code
:type exit_code: int
:param reason: The reason
:type reason: string
:param source: The source
:type source: string
:param message: The message
:type message: string
:param data: The data dict
:type data: dict
:returns: The task status update
:rtype: :class:`job.tasks.update.TaskStatusUpdate`
"""
if data is None:
data = {}
task_update_model = TaskUpdate()
task_update_model.task_id = task_id
task_update_model.timestamp = when
task_update_model.message = message
task_update_model.source = source
task_update_model.reason = reason
update = TaskStatusUpdate(task_update_model, agent_id, data)
update.status = status
if exit_code is not None:
update.exit_code = exit_code
return update
def create_task_update_model(status):
"""Creates and returns a task update model for the given Mesos task status
:param status: The task status in TaskStatus JSON format
:type status: dict
:returns: The task update model
:rtype: :class:`job.models.TaskUpdate`
"""
task_update = TaskUpdate()
task_update.task_id = get_status_task_id(status)
task_update.status = get_status_state(status)
task_update.timestamp = get_status_timestamp(status)
task_update.source = get_status_source(status)
task_update.reason = get_status_reason(status)
task_update.message = get_status_message(status)
return task_update
def create_task_update_model(status):
"""Creates and returns a task update model for the given Mesos task status
:param status: The task status
:type status: :class:`mesos_pb2.TaskStatus`
:returns: The task update model
:rtype: :class:`job.models.TaskUpdate`
"""
task_update = TaskUpdate()
task_update.task_id = get_status_task_id(status)
task_update.job_exe_id = RunningJobExecution.get_job_exe_id(task_update.task_id)
task_update.status = get_status_state(status)
task_update.timestamp = get_status_timestamp(status)
task_update.source = get_status_source(status)
task_update.reason = get_status_reason(status)
task_update.message = get_status_message(status)
return task_update
def create_task_update_model(status):
"""Creates and returns a task update model for the given Mesos task status
:param status: The task status
:type status: :class:`mesos_pb2.TaskStatus`
:returns: The task update model
:rtype: :class:`job.models.TaskUpdate`
"""
task_update = TaskUpdate()
task_update.task_id = get_status_task_id(status)
task_update.job_exe_id = RunningJobExecution.get_job_exe_id(
task_update.task_id)
task_update.status = get_status_state(status)
task_update.timestamp = get_status_timestamp(status)
task_update.source = get_status_source(status)
task_update.reason = get_status_reason(status)
task_update.message = get_status_message(status)
return task_update