def __init__(self, package='WMCore.DataStructs',
subscription=None,
generators=[],
limit = None):
"""
__init__
Create the DAOs
"""
myThread = threading.currentThread()
JobFactory.__init__(self, package = 'WMCore.WMBS',
subscription = subscription,
generators = generators,
limit = limit)
self.daoFactory = DAOFactory(package = "WMCore.WMBS",
logger = myThread.logger,
dbinterface = myThread.dbi)
self.getParentInfoAction = self.daoFactory(classname = "Files.GetParentInfo")
return
def __init__(self,
package='WMCore.DataStructs',
subscription=None,
generators=[],
limit=None):
"""
__init__
Create the DAOs
"""
myThread = threading.currentThread()
JobFactory.__init__(self,
package='WMCore.WMBS',
subscription=subscription,
generators=generators,
limit=limit)
self.daoFactory = DAOFactory(package="WMCore.WMBS",
logger=myThread.logger,
dbinterface=myThread.dbi)
self.getParentInfoAction = self.daoFactory(
classname="Files.GetParentInfo")
return
def testProductionRunNumber(self):
"""
_testProductionRunNumber_
Verify that jobs created by production subscritpions have the correct
run number is their job mask. Also verify that non-production
subscriptions don't have modified run numbers.
"""
testWorkflow = Workflow(spec="spec.pkl",
owner="Steve",
name="TestWorkflow",
task="TestTask")
testFileset = Fileset(name="TestFileset")
testFile = File(lfn="someLFN")
testFileset.addFile(testFile)
testFileset.commit()
testSubscription = Subscription(fileset=testFileset,
workflow=testWorkflow,
split_algo="FileBased",
type="Production")
myJobFactory = JobFactory(subscription=testSubscription)
testJobGroups = myJobFactory()
self.assertTrue(len(testJobGroups) > 0)
for testJobGroup in testJobGroups:
self.assertTrue(len(testJobGroup.jobs) > 0)
for job in testJobGroup.jobs:
self.assertEqual(job["mask"]["FirstRun"], 1,
"Error: First run is wrong.")
self.assertEqual(job["mask"]["LastRun"], 1,
"Error: Last run is wrong.")
testSubscription = Subscription(fileset=testFileset,
workflow=testWorkflow,
split_algo="FileBased",
type="Processing")
myJobFactory = JobFactory(subscription=testSubscription)
testJobGroups = myJobFactory()
for testJobGroup in testJobGroups:
for job in testJobGroup.jobs:
self.assertEqual(job["mask"]["FirstRun"], None,
"Error: First run is wrong.")
self.assertEqual(job["mask"]["LastRun"], None,
"Error: Last run is wrong.")
return
def testMetaData(self):
"""
_testMetaData_
Make sure that the workflow name, task, owner and white and black lists
make it into each job object.
"""
testWorkflow = Workflow(spec="spec.pkl",
owner="Steve",
name="TestWorkflow",
task="TestTask")
testFileset = Fileset(name="TestFileset")
testFile = File(lfn="someLFN")
testFileset.addFile(testFile)
testFileset.commit()
testSubscription = Subscription(fileset=testFileset,
workflow=testWorkflow,
split_algo="FileBased")
myJobFactory = JobFactory(subscription=testSubscription)
testJobGroups = myJobFactory(siteWhitelist=["site1"],
siteBlacklist=["site2"])
self.assertTrue(len(testJobGroups) > 0)
for testJobGroup in testJobGroups:
self.assertTrue(len(testJobGroup.jobs) > 0)
for job in testJobGroup.jobs:
self.assertEqual(job["task"], "TestTask",
"Error: Task is wrong.")
self.assertEqual(job["workflow"], "TestWorkflow",
"Error: Workflow is wrong.")
self.assertEqual(job["owner"], "Steve",
"Error: Owner is wrong.")
self.assertEqual(job["siteWhitelist"], ["site1"],
"Error: Site white list is wrong.")
self.assertEqual(job["siteBlacklist"], ["site2"],
"Error: Site black list is wrong.")
return