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)