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