def test__getPilotOptions(mocker):
""" Testing SiteDirector()._getPilotOptions()
"""
mocker.patch("DIRAC.WorkloadManagementSystem.Agent.SiteDirector.AgentModule.__init__")
mocker.patch("DIRAC.WorkloadManagementSystem.Agent.SiteDirector.gConfig.getValue", side_effect=mockGCReply)
mocker.patch("DIRAC.WorkloadManagementSystem.Agent.SiteDirector.Operations", side_effect=mockOPS)
mocker.patch("DIRAC.WorkloadManagementSystem.Agent.SiteDirector.gProxyManager.requestToken", side_effect=mockPMReply)
mocker.patch("DIRAC.WorkloadManagementSystem.Agent.SiteDirector.AgentModule", side_effect=mockAM)
sd = SiteDirector()
sd.log = gLogger
sd.am_getOption = mockAM
sd.log.setLevel('DEBUG')
sd.queueDict = {'aQueue': {'CEName': 'aCE',
'QueueName': 'aQueue',
'ParametersDict': {'CPUTime': 12345,
'Community': 'lhcb',
'OwnerGroup': ['lhcb_user'],
'Setup': 'LHCb-Production',
'Site': 'LCG.CERN.cern',
'SubmitPool': ''}}}
res = sd._getPilotOptions('aQueue', 10)
assert res[0] == ['-S TestSetup', '-V 123', '-l 123', '-r 1,2,3', '-g 123',
'-o /Security/ProxyToken=token', '-M 1', '-C T,e,s,t,S,e,t,u,p',
'-e 1,2,3', '-N aCE', '-Q aQueue', '-n LCG.CERN.cern']
assert res[1] == 1
def test__getPilotOptions(self, _patch1, _patch2, _patch3, _patch4):
""" Testing SiteDirector()._getPilotOptions()
"""
sd = SiteDirector()
sd.log = gLogger
sd.am_getOption = mockAM
sd.log.setLevel('DEBUG')
sd.queueDict = {
'aQueue': {
'CEName': 'aCE',
'QueueName': 'aQueue',
'ParametersDict': {
'CPUTime': 12345,
'Community': 'lhcb',
'OwnerGroup': ['lhcb_user'],
'Setup': 'LHCb-Production',
'Site': 'LCG.CERN.cern',
'SubmitPool': ''
}
}
}
res = sd._getPilotOptions('aQueue', 10)
self.assertEqual(res[0], [
'-S TestSetup', '-V 123', '-l 123', '-r 1,2,3', '-g 123',
'-o /Security/ProxyToken=token', '-M 1', '-C T,e,s,t,S,e,t,u,p',
'-e 1,2,3', '-N aCE', '-Q aQueue', '-n LCG.CERN.cern'
])
self.assertEqual(res[1], 1)
def _getPilotOptions( self, queue, pilotsToSubmit, **kwargs ):
""" Prepare pilot options
"""
pilotOptions, pilotsToSubmit = OriginalSiteDirector._getPilotOptions(self, queue, pilotsToSubmit)
# Get the module specific options
if self.extraOptions:
pilotOptions.append( self.extraOptions )
processors = kwargs.pop('processors', 1)
if processors > 0:
pilotOptions.append( '--maxNumberOfProcessors %u' % processors )
return [pilotOptions, pilotsToSubmit]
def test__getPilotOptions(self, _patch1, _patch2, _patch3):
sd = SiteDirector()
sd.log = gLogger
sd.am_getOption = mockAM
sd.log.setLevel('DEBUG')
sd.queueDict = {'aQueue': {'ParametersDict': {'CPUTime': 12345}}}
sd.queueDict['aQueue'] = {}
sd.queueDict['aQueue']['ParametersDict'] = {}
res = sd._getPilotOptions('aQueue', 10)
#FIXME: incomplete
self.assertEqual(res, [None, None])
def test__getPilotOptions(mocker):
""" Testing SiteDirector()._getPilotOptions()
"""
mocker.patch(
"DIRAC.WorkloadManagementSystem.Agent.SiteDirector.AgentModule.__init__"
)
mocker.patch(
"DIRAC.WorkloadManagementSystem.Agent.SiteDirector.gConfig.getValue",
side_effect=mockGCReply)
mocker.patch(
"DIRAC.WorkloadManagementSystem.Agent.SiteDirector.Operations",
side_effect=mockOPS)
mocker.patch(
"DIRAC.WorkloadManagementSystem.Agent.SiteDirector.gProxyManager.requestToken",
side_effect=mockPMReply)
mocker.patch(
"DIRAC.WorkloadManagementSystem.Agent.SiteDirector.AgentModule",
side_effect=mockAM)
sd = SiteDirector()
sd.log = gLogger
sd.am_getOption = mockAM
sd.log.setLevel('DEBUG')
sd.queueDict = {
'aQueue': {
'CEName': 'aCE',
'QueueName': 'aQueue',
'ParametersDict': {
'CPUTime': 12345,
'Community': 'lhcb',
'OwnerGroup': ['lhcb_user'],
'Setup': 'LHCb-Production',
'Site': 'LCG.CERN.cern',
'SubmitPool': ''
}
}
}
res = sd._getPilotOptions('aQueue', 10)
assert res[0] == [
'-S TestSetup', '-V 123', '-l 123', '-r 1,2,3', '-g 123',
'-o /Security/ProxyToken=token', '-M 1', '-C T,e,s,t,S,e,t,u,p',
'-e 1,2,3', '-N aCE', '-Q aQueue', '-n LCG.CERN.cern'
]
assert res[1] == 1
def _getPilotOptions(self, queue, pilotsToSubmit):
""" Adding LHCb specific options
"""
pilotOptions, newPilotsToSubmit = DIRACSiteDirector._getPilotOptions(
self, queue, pilotsToSubmit)
lhcbPilotCommands = [
'LHCbGetPilotVersion', 'CheckWorkerNode', 'LHCbInstallDIRAC',
'LHCbConfigureBasics', 'CheckCECapabilities',
'CheckWNCapabilities', 'LHCbConfigureSite',
'LHCbConfigureArchitecture', 'LHCbConfigureCPURequirements',
'LaunchAgent'
]
pilotOptions.append('-E LHCbPilot')
pilotOptions.append('-X %s' % ','.join(lhcbPilotCommands))
if self.lbRunOnly:
pilotOptions.append('-o lbRunOnly')
return [pilotOptions, newPilotsToSubmit]
def test__getPilotOptions(mocker):
""" Testing SiteDirector()._getPilotOptions()
"""
mocker.patch("DIRAC.WorkloadManagementSystem.Agent.SiteDirector.AgentModule.__init__")
mocker.patch("DIRAC.WorkloadManagementSystem.Agent.SiteDirector.gConfig.getValue", side_effect=mockGCReply)
mocker.patch("DIRAC.WorkloadManagementSystem.Agent.SiteDirector.Operations", side_effect=mockOPS)
mocker.patch("DIRAC.WorkloadManagementSystem.Agent.SiteDirector.gProxyManager.requestToken", side_effect=mockPMReply)
mocker.patch("DIRAC.WorkloadManagementSystem.Agent.SiteDirector.AgentModule", side_effect=mockAM)
sd = SiteDirector()
sd.log = gLogger
sd.am_getOption = mockAM
sd.log.setLevel('DEBUG')
sd.queueDict = {'aQueue': {'CEName': 'aCE',
'QueueName': 'aQueue',
'ParametersDict': {'CPUTime': 12345,
'Community': 'lhcb',
'OwnerGroup': ['lhcb_user'],
'Setup': 'LHCb-Production',
'Site': 'LCG.CERN.cern'}}}
res = sd._getPilotOptions('aQueue')
assert set(['-S TestSetup', '-V 123', '-l 123', '-n LCG.CERN.cern']) <= set(res)