def test__submitPilotsToQueue(mocker):
""" Testing SiteDirector()._submitPilotsToQueue()
"""
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.CSGlobals.getSetup", side_effect=mockCSGlobalReply)
mocker.patch("DIRAC.WorkloadManagementSystem.Agent.SiteDirector.Resources.getCompatiblePlatforms",
side_effect=mockResourcesReply)
mocker.patch("DIRAC.WorkloadManagementSystem.Agent.SiteDirector.AgentModule", side_effect=mockAM)
sd = SiteDirector()
sd.log = gLogger
sd.am_getOption = mockAM
sd.log.setLevel('DEBUG')
sd.rpcMatcher = MagicMock()
sd.rssClient = MagicMock()
sd.workingDirectory = ''
sd.queueDict = {'aQueue': {'Site': 'LCG.CERN.cern',
'CEName': 'aCE',
'CEType': 'SSH',
'QueueName': 'aQueue',
'ParametersDict': {'CPUTime': 12345,
'Community': 'lhcb',
'OwnerGroup': ['lhcb_user'],
'Setup': 'LHCb-Production',
'Site': 'LCG.CERN.cern',
'SubmitPool': ''}}}
sd.queueSlots = {'aQueue': {'AvailableSlots': 10}}
res = sd._submitPilotsToQueue(1, MagicMock(), 'aQueue')
assert res['OK'] is True
assert res['Value'][0] == 0
def test__submitPilotsToQueue(mocker):
""" Testing SiteDirector()._submitPilotsToQueue()
"""
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.CSGlobals.getSetup", side_effect=mockCSGlobalReply)
mocker.patch("DIRAC.WorkloadManagementSystem.Agent.SiteDirector.AgentModule", side_effect=mockAM)
sd = SiteDirector()
sd.log = gLogger
sd.am_getOption = mockAM
sd.log.setLevel('DEBUG')
sd.rpcMatcher = MagicMock()
sd.rssClient = MagicMock()
sd.workingDirectory = ''
sd.queueDict = {'aQueue': {'Site': 'LCG.CERN.cern',
'CEName': 'aCE',
'CEType': 'SSH',
'QueueName': 'aQueue',
'ParametersDict': {'CPUTime': 12345,
'Community': 'lhcb',
'OwnerGroup': ['lhcb_user'],
'Setup': 'LHCb-Production',
'Site': 'LCG.CERN.cern'}}}
sd.queueSlots = {'aQueue': {'AvailableSlots': 10}}
res = sd._submitPilotsToQueue(1, MagicMock(), 'aQueue')
assert res['OK'] is True
def test__submitPilotsToQueue(self, _patch1, _patch2, _patch3, _patch4):
""" Testing SiteDirector()._submitPilotsToQueue()
"""
sd = SiteDirector()
sd.log = gLogger
sd.am_getOption = mockAM
sd.log.setLevel('DEBUG')
sd.rpcMatcher = MagicMock()
sd.rssClient = MagicMock()
sd.workingDirectory = ''
sd.queueDict = {
'aQueue': {
'Site': 'LCG.CERN.cern',
'CEName': 'aCE',
'CEType': 'SSH',
'QueueName': 'aQueue',
'ParametersDict': {
'CPUTime': 12345,
'Community': 'lhcb',
'OwnerGroup': ['lhcb_user'],
'Setup': 'LHCb-Production',
'Site': 'LCG.CERN.cern',
'SubmitPool': ''
}
}
}
sd.queueSlots = {'aQueue': {'AvailableSlots': 10}}
res = sd._submitPilotsToQueue(1, MagicMock(), 'aQueue')
self.assertTrue(res['OK'])
self.assertEqual(res['Value'][0], 0)
def sd(mocker):
"""mocker for SiteDirector"""
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.rpcMatcher = MagicMock()
sd.rssClient = MagicMock()
sd.workingDirectory = ""
sd.queueDict = {
"aQueue": {
"Site": "LCG.CERN.cern",
"CEName": "aCE",
"CEType": "SSH",
"QueueName": "aQueue",
"ParametersDict": {
"CPUTime": 12345,
"Community": "lhcb",
"OwnerGroup": ["lhcb_user"],
"Setup": "LHCb-Production",
"Site": "LCG.CERN.cern",
},
}
}
return sd