Python JobAgent.ceName示例

示例#1

def test_submitJob(mocker, mockJWInput, expected):
    """ Testing JobAgent()._submitJob()
  """

    mockJW.return_value = mockJWInput

    mocker.patch(
        "DIRAC.WorkloadManagementSystem.Agent.JobAgent.AgentModule.__init__")
    mocker.patch(
        "DIRAC.WorkloadManagementSystem.Agent.JobAgent.AgentModule.am_getOption",
        side_effect=mockAM)
    mocker.patch(
        "DIRAC.WorkloadManagementSystem.Agent.JobAgent.createJobWrapper",
        side_effect=mockJW)

    jobAgent = JobAgent('Test', 'Test1')
    jobAgent.log = gLogger
    jobAgent.log.setLevel('DEBUG')
    jobAgent.ceName = 'Test'

    result = jobAgent._submitJob(101, {}, {}, {}, {}, {})

    assert result['OK'] == expected['OK']

    if not result['OK']:
        assert result['Message'] == expected['Message']

示例#2

def test_submitJob(mocker, mockJWInput, expected):
    """Testing JobAgent()._submitJob()"""
    mocker.patch(
        "DIRAC.WorkloadManagementSystem.Agent.JobAgent.AgentModule.__init__")
    mocker.patch(
        "DIRAC.WorkloadManagementSystem.Agent.JobAgent.AgentModule._AgentModule__moduleProperties",
        side_effect=lambda x, y=None: y,
        create=True,
    )
    mocker.patch(
        "DIRAC.WorkloadManagementSystem.Agent.JobAgent.createJobWrapper",
        return_value=mockJWInput)

    jobAgent = JobAgent("Test", "Test1")
    jobAgent.log = gLogger
    jobAgent.log.setLevel("DEBUG")
    jobAgent.ceName = "Test"

    result = jobAgent._submitJob(101, {}, {}, {}, "", 1)

    assert result["OK"] == expected["OK"]

    if not result["OK"]:
        assert result["Message"] == expected["Message"]