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)