def setUp(self): self.tasks = [ 'demo.demo_task.TestTask', 'demo.demo_task.TestContainerTask', 'demo.demo_task.TestParallelTask' ] self.completion = {} for task in self.tasks: self.completion[task] = None # setup manager with an internal cache we can alter self.manager = TaskManager(None, lazy_init=True) pydra_settings.TASK_DIR_INTERNAL = '/var/lib/pydra/test_tasks_internal' # find at least one task package to use for testing self.package = 'demo' self.package_dir = '%s/%s' % (self.manager.tasks_dir_internal, self.package) self.task_instances = [] for task in self.tasks [:2]: #queued tasks task_instance = TaskInstance() task_instance.task_key=task task_instance.save() self.task_instances.append(task_instance) #running tasks task_instance = TaskInstance() task_instance.task_key=task task_instance.started = datetime.now() task_instance.save() self.task_instances.append(task_instance) #finished tasks task_instance = TaskInstance() task_instance.task_key=task task_instance.started = datetime.now() completed_time = datetime.now() task_instance.completed = completed_time task_instance.save() self.completion[task] = completed_time self.task_instances.append(task_instance) #failed tasks task_instance = TaskInstance() task_instance.task_key=task task_instance.started = datetime.now() task_instance.status = -1 task_instance.save() self.task_instances.append(task_instance)
def setUp(self): TaskManagerTestCaseMixIn.setUp(self) self.completion = {} for task in self.tasks: self.completion[task] = None self.task_instances = [] for task in self.tasks: #queued tasks task_instance = TaskInstance() task_instance.task_key=task task_instance.save() self.task_instances.append(task_instance) #running tasks task_instance = TaskInstance() task_instance.task_key=task task_instance.started = datetime.now() task_instance.save() self.task_instances.append(task_instance) #finished tasks task_instance = TaskInstance() task_instance.task_key=task task_instance.started = datetime.now() completed_time = datetime.now() task_instance.completed = completed_time task_instance.save() self.completion[task] = completed_time self.task_instances.append(task_instance) #failed tasks task_instance = TaskInstance() task_instance.task_key=task task_instance.started = datetime.now() task_instance.status = -1 task_instance.save() self.task_instances.append(task_instance)