def test_finalize(mocker, failedFlag, expectedRes, finalStates): mocker.patch( "DIRAC.WorkloadManagementSystem.JobWrapper.JobWrapper.getSystemSection", side_effect=getSystemSectionMock) mocker.patch( "DIRAC.WorkloadManagementSystem.JobWrapper.JobWrapper.ModuleFactory", side_effect=MagicMock()) jw = JobWrapper() jw.jobArgs = {"Executable": "/bin/ls"} jw.failedFlag = failedFlag res = jw.finalize() assert res == expectedRes assert jw.jobReport.jobStatusInfo[0][0] == finalStates[0] assert jw.jobReport.jobStatusInfo[0][1] == finalStates[1]