def testTwoTaskTree(self):
workload = TestSpecs.twoTaskTree()
tasks = workload.listAllTaskNames()
parenttask = workload.getTask("FirstTask")
findWorkload = getWorkloadFromTask(parenttask)
self.assertEqual(id(findWorkload.data), id(workload.data))
self.assertTrue( isinstance( workload, WMCore.WMSpec.WMWorkload.WMWorkloadHelper ) )
self.assertTrue( isinstance( parenttask, WMCore.WMSpec.WMTask.WMTaskHelper ) )
task = workload.getTask("SecondTask")
#print task.data
self.assertTrue( isinstance( task, WMCore.WMSpec.WMTask.WMTaskHelper ) )
steps = task.listAllStepNames()
# there should be a way to do this with iteration that would be neater
step = task.getStep(steps[0])
self.assertTrue( isinstance( step, WMCore.WMSpec.WMStep.WMStepHelper ) )
step2= task.getStep(steps[1])
self.assertTrue( isinstance( step2, WMCore.WMSpec.WMStep.WMStepHelper ) )
step3= task.getStep(steps[2])
self.assertTrue( isinstance( step3, WMCore.WMSpec.WMStep.WMStepHelper ) )
step4= task.getStep(steps[3])
self.assertTrue( isinstance( step4, WMCore.WMSpec.WMStep.WMStepHelper ) )
self.assertEqual( id(getTaskFromStep(step)), id(getTaskFromStep(step2)) )
def testTwoTaskTree(self):
workload = TestSpecs.twoTaskTree()
tasks = workload.listAllTaskNames()
parenttask = workload.getTask("FirstTask")
findWorkload = getWorkloadFromTask(parenttask)
self.assertEqual(id(findWorkload.data), id(workload.data))
self.assertTrue(
isinstance(workload, WMCore.WMSpec.WMWorkload.WMWorkloadHelper))
self.assertTrue(
isinstance(parenttask, WMCore.WMSpec.WMTask.WMTaskHelper))
task = workload.getTask("SecondTask")
#print task.data
self.assertTrue(isinstance(task, WMCore.WMSpec.WMTask.WMTaskHelper))
steps = task.listAllStepNames()
# there should be a way to do this with iteration that would be neater
step = task.getStep(steps[0])
self.assertTrue(isinstance(step, WMCore.WMSpec.WMStep.WMStepHelper))
step2 = task.getStep(steps[1])
self.assertTrue(isinstance(step2, WMCore.WMSpec.WMStep.WMStepHelper))
step3 = task.getStep(steps[2])
self.assertTrue(isinstance(step3, WMCore.WMSpec.WMStep.WMStepHelper))
step4 = task.getStep(steps[3])
self.assertTrue(isinstance(step4, WMCore.WMSpec.WMStep.WMStepHelper))
self.assertEqual(id(getTaskFromStep(step)), id(getTaskFromStep(step2)))
def testOneTaskTwoSTep(self):
workload = TestSpecs.oneTaskTwoStep()
tasks = workload.listAllTaskNames()
task = workload.getTask(tasks[0])
findWorkload = getWorkloadFromTask(task)
self.assertEqual(id(findWorkload.data), id(workload.data))
self.assertTrue( isinstance( workload, WMCore.WMSpec.WMWorkload.WMWorkloadHelper ) )
self.assertTrue( isinstance( task, WMCore.WMSpec.WMTask.WMTaskHelper ) )
steps = task.listAllStepNames()
step = task.getStep(steps[0])
self.assertTrue( isinstance( step, WMCore.WMSpec.WMStep.WMStepHelper ) )
step2= task.getStep(steps[1])
self.assertTrue( isinstance( step2, WMCore.WMSpec.WMStep.WMStepHelper ) )
self.assertEqual( id(getTaskFromStep(step)), id(getTaskFromStep(step2)) )
def testOneTaskTwoSTep(self):
workload = TestSpecs.oneTaskTwoStep()
tasks = workload.listAllTaskNames()
task = workload.getTask(tasks[0])
findWorkload = getWorkloadFromTask(task)
self.assertEqual(id(findWorkload.data), id(workload.data))
self.assertTrue(
isinstance(workload, WMCore.WMSpec.WMWorkload.WMWorkloadHelper))
self.assertTrue(isinstance(task, WMCore.WMSpec.WMTask.WMTaskHelper))
steps = task.listAllStepNames()
step = task.getStep(steps[0])
self.assertTrue(isinstance(step, WMCore.WMSpec.WMStep.WMStepHelper))
step2 = task.getStep(steps[1])
self.assertTrue(isinstance(step2, WMCore.WMSpec.WMStep.WMStepHelper))
self.assertEqual(id(getTaskFromStep(step)), id(getTaskFromStep(step2)))
