def test__requestProxyFromProxyManager(mocker, mockGCReplyInput, mockPMReplyInput, expected): """ Testing JobAgent()._requestProxyFromProxyManager() """ mockGCReply.return_value = mockGCReplyInput mockPMReply.return_value = mockPMReplyInput mocker.patch( "DIRAC.WorkloadManagementSystem.Agent.JobAgent.AgentModule.__init__") mocker.patch("DIRAC.WorkloadManagementSystem.Agent.JobAgent.AgentModule", side_effect=mockAM) mocker.patch( "DIRAC.WorkloadManagementSystem.Agent.JobAgent.gConfig.getValue", side_effect=mockGCReply) module_str = "DIRAC.WorkloadManagementSystem.Agent.JobAgent.gProxyManager.getPayloadProxyFromDIRACGroup" mocker.patch(module_str, side_effect=mockPMReply) jobAgent = JobAgent('Test', 'Test1') ownerDN = 'DIRAC' ownerGroup = 'DIRAC' jobAgent.log = gLogger jobAgent.log.setLevel('DEBUG') result = jobAgent._requestProxyFromProxyManager(ownerDN, ownerGroup) assert result['OK'] == expected['OK'] if result['OK']: assert result['Value'] == expected['Value'] else: assert result['Message'] == expected['Message']
def test__requestProxyFromProxyManager(mocker, mockGCReply, mockPMReply, expected): """Testing JobAgent()._requestProxyFromProxyManager()""" mocker.patch( "DIRAC.WorkloadManagementSystem.Agent.JobAgent.AgentModule.__init__") mocker.patch("DIRAC.WorkloadManagementSystem.Agent.JobAgent.AgentModule") mocker.patch( "DIRAC.WorkloadManagementSystem.Agent.JobAgent.gConfig.getValue", return_value=mockGCReply) module_str = "DIRAC.WorkloadManagementSystem.Agent.JobAgent.gProxyManager.getPayloadProxyFromDIRACGroup" mocker.patch(module_str, return_value=mockPMReply) jobAgent = JobAgent("Test", "Test1") ownerDN = "DIRAC" ownerGroup = "DIRAC" jobAgent.log = gLogger jobAgent.log.setLevel("DEBUG") result = jobAgent._requestProxyFromProxyManager(ownerDN, ownerGroup) assert result["OK"] == expected["OK"] if result["OK"]: assert result["Value"] == expected["Value"] else: assert result["Message"] == expected["Message"]