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()
