Python SiteDirector._allowedToSubmit Exemples

Exemple #1

Fichier : Test_Agent_SiteDirector.py Projet : marianne013/DIRAC

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

Exemple #2

Fichier : Test_Agent_SiteDirector.py Projet : pmusset/DIRAC

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

Exemple #3

    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)