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