def test__updatePilotStatus(mocker, pilotRefs, pilotDict, pilotCEDict, expected): """ Testing SiteDirector()._updatePilotStatus() """ 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) mocker.patch( "DIRAC.WorkloadManagementSystem.Agent.SiteDirector.pilotAgentsDB", side_effect=mockPilotAgentsDB) sd = SiteDirector() sd.log = gLogger sd.am_getOption = mockAM sd.log.setLevel('DEBUG') sd.rpcMatcher = MagicMock() sd.rssClient = MagicMock() res = sd._updatePilotStatus(pilotRefs, pilotDict, pilotCEDict) assert res == expected
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(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(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 test__updatePilotStatus(mocker, pilotRefs, pilotDict, pilotCEDict, expected): """ Testing SiteDirector()._updatePilotStatus() """ 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) mocker.patch("DIRAC.WorkloadManagementSystem.Agent.SiteDirector.pilotAgentsDB", side_effect=mockPilotAgentsDB) sd = SiteDirector() sd.log = gLogger sd.am_getOption = mockAM sd.log.setLevel('DEBUG') sd.rpcMatcher = MagicMock() sd.rssClient = MagicMock() res = sd._updatePilotStatus(pilotRefs, pilotDict, pilotCEDict) assert res == expected
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