Python SiteDirector.matcherClient Examples

Example #1

def test__ifAndWhereToSubmit(mocker, mockMatcherReturnValue, expected,
                             anyExpected, sitesExpected):
    """ Testing SiteDirector()._ifAndWhereToSubmit()
  """
    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.matcherClient = MagicMock()
    sd.matcherClient.getMatchingTaskQueues.return_value = mockMatcherReturnValue
    res = sd._ifAndWhereToSubmit()
    assert res[0] == expected
    if res[0]:
        assert res == (expected, anyExpected, sitesExpected, set())

Example #2

 def test__ifAndWhereToSubmit(self, _patch1, _patch2, _patch3, _patch4):
     """ Testing SiteDirector()._ifAndWhereToSubmit()
 """
     sd = SiteDirector()
     sd.log = gLogger
     sd.am_getOption = mockAM
     sd.log.setLevel('DEBUG')
     sd.matcherClient = MagicMock()
     submit, _anySite, _jobSites, _testSites = sd._ifAndWhereToSubmit()
     self.assertTrue(submit)

Example #3

File: Test_Agent_SiteDirector.py Project: DIRACGrid/DIRAC

def test__ifAndWhereToSubmit(mocker, mockMatcherReturnValue, expected, anyExpected, sitesExpected):
  """ Testing SiteDirector()._ifAndWhereToSubmit()
  """
  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.matcherClient = MagicMock()
  sd.matcherClient.getMatchingTaskQueues.return_value = mockMatcherReturnValue
  res = sd._ifAndWhereToSubmit()
  assert res[0] == expected
  if res[0]:
    assert res == (expected, anyExpected, sitesExpected, set())