def test__allowedToSubmit(mocker): """ Testing SiteDirector()._allowedToSubmit() """ mocker.patch("DIRAC.WorkloadManagementSystem.Agent.SiteDirector.AgentModule.__init__") 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': {'Site': 'LCG.CERN.cern', 'CEName': 'aCE', 'QueueName': 'aQueue', 'ParametersDict': {'CPUTime': 12345, 'Community': 'lhcb', 'OwnerGroup': ['lhcb_user'], 'Setup': 'LHCb-Production', 'Site': 'LCG.CERN.cern', 'SubmitPool': ''}}} submit = sd._allowedToSubmit('aQueue', True, set(['LCG.CERN.cern']), set()) assert submit is False sd.siteMaskList = ['LCG.CERN.cern', 'DIRAC.CNAF.it'] submit = sd._allowedToSubmit('aQueue', True, set(['LCG.CERN.cern']), set()) assert submit is True sd.rssFlag = True submit = sd._allowedToSubmit('aQueue', True, set(['LCG.CERN.cern']), set()) assert submit is False sd.ceMaskList = ['aCE', 'anotherCE'] submit = sd._allowedToSubmit('aQueue', True, set(['LCG.CERN.cern']), set()) assert submit is True
def test__allowedToSubmit(mocker): """ Testing SiteDirector()._allowedToSubmit() """ mocker.patch("DIRAC.WorkloadManagementSystem.Agent.SiteDirector.AgentModule.__init__") 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': {'Site': 'LCG.CERN.cern', 'CEName': 'aCE', 'QueueName': 'aQueue', 'ParametersDict': {'CPUTime': 12345, 'Community': 'lhcb', 'OwnerGroup': ['lhcb_user'], 'Setup': 'LHCb-Production', 'Site': 'LCG.CERN.cern'}}} submit = sd._allowedToSubmit('aQueue', True, set(['LCG.CERN.cern']), set()) assert submit is False sd.siteMaskList = ['LCG.CERN.cern', 'DIRAC.CNAF.it'] submit = sd._allowedToSubmit('aQueue', True, set(['LCG.CERN.cern']), set()) assert submit is True sd.rssFlag = True submit = sd._allowedToSubmit('aQueue', True, set(['LCG.CERN.cern']), set()) assert submit is False sd.ceMaskList = ['aCE', 'anotherCE'] submit = sd._allowedToSubmit('aQueue', True, set(['LCG.CERN.cern']), set()) assert submit is True
def test__allowedToSubmit(self, _patch1): """ Testing SiteDirector()._allowedToSubmit() """ sd = SiteDirector() sd.log = gLogger sd.am_getOption = mockAM sd.log.setLevel('DEBUG') sd.queueDict = { 'aQueue': { 'Site': 'LCG.CERN.cern', 'CEName': 'aCE', 'QueueName': 'aQueue', 'ParametersDict': { 'CPUTime': 12345, 'Community': 'lhcb', 'OwnerGroup': ['lhcb_user'], 'Setup': 'LHCb-Production', 'Site': 'LCG.CERN.cern', 'SubmitPool': '' } } } submit = sd._allowedToSubmit('aQueue', True, set(['LCG.CERN.cern']), set()) self.assertFalse(submit) sd.siteMaskList = ['LCG.CERN.cern', 'DIRAC.CNAF.it'] submit = sd._allowedToSubmit('aQueue', True, set(['LCG.CERN.cern']), set()) self.assertTrue(submit) sd.rssFlag = True submit = sd._allowedToSubmit('aQueue', True, set(['LCG.CERN.cern']), set()) self.assertFalse(submit) sd.ceMaskList = ['aCE', 'anotherCE'] submit = sd._allowedToSubmit('aQueue', True, set(['LCG.CERN.cern']), set()) self.assertTrue(submit)