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)