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)
