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)