def create_scheduled_task(instance, start_time): task = ScheduledTask() task.assignedTask = AssignedTask() task.assignedTask.taskId = "task_%s" % instance task.assignedTask.slaveId = "random_machine_id" task.assignedTask.slaveHost = "junk.nothing" task.assignedTask.task = TaskConfig() task.assignedTask.task.owner = Identity(role="nobody") task.assignedTask.task.environment = "prod" task.assignedTask.task.jobName = "flibber" task.assignedTask.task.isService = False task.assignedTask.task.numCpus = 2 task.assignedTask.task.ramMb = 2048 task.assignedTask.task.diskMb = 4096 task.assignedTask.task.priority = 7 task.assignedTask.task.maxTaskFailures = 3 task.assignedTask.task.production = False task.assignedTask.task.requestedPorts = ["http"] task.assignedTask.assignedPorts = {"http": 1001} task.assignedTask.instanceId = 0 task.status = 2 task.failureCount = instance + 4 task.taskEvents = create_task_events(start_time) task.ancestorId = "random_task_ancestor%s" % instance return task
def create_scheduled_tasks(cls): tasks = [] for name in ['foo', 'bar', 'baz']: task = ScheduledTask() task.failure_count = 0 task.assignedTask = AssignedTask() task.assignedTask.taskId = 1287391823 task.assignedTask.slaveHost = 'slavehost' task.assignedTask.task = TaskConfig() task.assignedTask.task.maxTaskFailures = 1 task.assignedTask.task.executorConfig = ExecutorConfig() task.assignedTask.task.executorConfig.data = Mock() task.assignedTask.task.metadata = [] task.assignedTask.task.job = JobKey(role=cls.TEST_ROLE, environment=cls.TEST_ENV, name=name) task.assignedTask.task.owner = Identity(role=cls.TEST_ROLE) task.assignedTask.task.environment = cls.TEST_ENV task.assignedTask.task.jobName = name task.assignedTask.task.numCpus = 2 task.assignedTask.task.ramMb = 2 task.assignedTask.task.diskMb = 2 task.assignedTask.instanceId = 4237894 task.assignedTask.assignedPorts = {} task.status = ScheduleStatus.RUNNING event = TaskEvent() event.timestamp = 28234726395 event.status = ScheduleStatus.RUNNING event.message = "Hi there" task.taskEvents = [event] tasks.append(task) return tasks