def setUp(self):
self.connectDB()
self.driver = Driver(self.engine)
class DriverTest(DBTest):
def setUp(self):
self.connectDB()
self.driver = Driver(self.engine)
def testInstall(self):
self.assert_(self.driver.uninstall())
self.assert_(self.driver.install())
self.assert_(self.driver.uninstall())
self.assert_(self.driver.install())
def testDriver(self):
self.assert_(self.driver is not None)
# Create a workflow.
file = os.path.join(os.path.dirname(__file__), 'parallel_split.xml')
workflow_info = WorkflowInfo('my/workflow', file = file)
self.assertRaises(WorkflowServerException,
self.driver.create_job,
workflow_info)
self.driver.save_workflow_info(workflow_info)
self.assert_(workflow_info.id >= 0)
# Instantiate the workflow.
job_info = self.driver.create_job(workflow_info)
self.assert_(job_info.id >= 0)
# Retrieve a list of tasks.
task_info_list = self.driver.get_task_info(job_id = job_info.id)
self.assert_(len(task_info_list) == 10)
task_info_list = self.driver.get_task_info(job_id = job_info.id,
status = TaskInfo.WAITING)
self.assert_(len(task_info_list) == 1)
# Execute a few tasks.
self.driver.execute_task(task_info_list[0])
task_info_list = self.driver.get_task_info(job_id = job_info.id,
status = TaskInfo.WAITING)
self.assert_(len(task_info_list) == 1)
