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())
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)
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())