from WMQuality.Emulators.DBSClient.DBSReader import DBSReader as MockDBSReader from WMQuality.Emulators.SiteDBClient.SiteDB import SiteDBJSON as fakeSiteDB from WMCore.WMSpec.StdSpecs.ReReco import rerecoWorkload, \ getTestArguments as getRerecoArgs from WMQuality.Emulators.WMSpecGenerator.Samples.TestMonteCarloWorkload \ import monteCarloWorkload, getMCArgs from WMQuality.Emulators import EmulatorSetup from WMQuality.TestInitCouchApp import TestInitCouchApp from WMCore.BossAir.BossAirAPI import BossAirAPI from WMCore.Configuration import loadConfigurationFile from WMCore.ResourceControl.ResourceControl import ResourceControl rerecoArgs = getRerecoArgs() mcArgs = getMCArgs() def getFirstTask(wmspec): """Return the 1st top level task""" # http://www.logilab.org/ticket/8774 # pylint: disable-msg=E1101,E1103 return wmspec.taskIterator().next() class WMBSHelperTest(unittest.TestCase): def setUp(self): """ _setUp_
from WMQuality.Emulators.DBSClient.DBSReader import DBSReader as MockDBSReader from WMQuality.Emulators.SiteDBClient.SiteDB import SiteDBJSON as fakeSiteDB from WMCore.WMSpec.StdSpecs.ReReco import rerecoWorkload, \ getTestArguments as getRerecoArgs from WMQuality.Emulators.WMSpecGenerator.Samples.TestMonteCarloWorkload \ import monteCarloWorkload, getMCArgs from WMQuality.Emulators import EmulatorSetup from WMQuality.TestInitCouchApp import TestInitCouchApp from WMCore.BossAir.BossAirAPI import BossAirAPI from WMCore.Configuration import loadConfigurationFile from WMCore.ResourceControl.ResourceControl import ResourceControl rerecoArgs = getRerecoArgs() mcArgs = getMCArgs() def getFirstTask(wmspec): """Return the 1st top level task""" # http://www.logilab.org/ticket/8774 # pylint: disable-msg=E1101,E1103 return wmspec.taskIterator().next() class WMBSHelperTest(unittest.TestCase): def setUp(self): """ _setUp_ """ self.testInit = TestInitCouchApp(__file__)
from WMCore_t.WorkQueue_t.WorkQueueTestCase import WorkQueueTestCase from WMCore_t.WMSpec_t.samples.MultiTaskProductionWorkload \ import workload as MultiTaskProductionWorkload from WMCore.Services.EmulatorSwitch import EmulatorHelper from WMCore.ACDC.DataCollectionService import DataCollectionService from WMCore.DataStructs.Run import Run from WMCore.Services.UUID import makeUUID from WMCore.WMBS.Job import Job from WMCore.DataStructs.File import File as WMFile from WMCore.WMSpec.WMWorkload import WMWorkload, WMWorkloadHelper from WMCore.ResourceControl.ResourceControl import ResourceControl from WMCore.Lexicon import sanitizeURL # NOTE: All queues point to the same database backend # Thus total element counts etc count elements in all queues rerecoArgs = getRerecoArgs() mcArgs = getMCArgs() parentProcArgs = getRerecoArgs() parentProcArgs.update(IncludeParents = "True") def rerecoWorkload(workloadName, arguments): wmspec = rerecoWMSpec(workloadName, arguments) #wmspec.setStartPolicy("DatasetBlock") return wmspec def getFirstTask(wmspec): """Return the 1st top level task""" # http://www.logilab.org/ticket/8774 # pylint: disable-msg=E1101,E1103 return wmspec.taskIterator().next()