def test__getInputs(mocker, mockReplyInput, manifestOptions, expected): mockReply.return_value = mockReplyInput module_str = "DIRAC.WorkloadManagementSystem.Client.JobState.JobState.JobDB.getInputData" mocker.patch(module_str, side_effect=mockReply) mocker.patch( "DIRAC.WorkloadManagementSystem.Client.JobState.JobState.JobDB.__init__", side_effect=mockNone) mocker.patch( "DIRAC.WorkloadManagementSystem.Client.JobState.JobState.JobLoggingDB.__init__", side_effect=mockNone) mocker.patch( "DIRAC.WorkloadManagementSystem.Client.JobState.JobState.TaskQueueDB.__init__", side_effect=mockNone) inputData = InputData() js = CachedJobState(1) js.setAttribute('JobType', 'User') manifest = JobManifest() for varName, varValue in manifestOptions.iteritems(): manifest.setOption(varName, varValue) js.setManifest(manifest) res = inputData._getInputs(js) assert res['OK'] is True assert res['Value'] == expected
def test__getInputSandbox(mocker, manifestOptions, expected): mocker.patch("DIRAC.WorkloadManagementSystem.Client.JobState.JobState.JobDB.__init__", side_effect=mockNone) mocker.patch("DIRAC.WorkloadManagementSystem.Client.JobState.JobState.JobLoggingDB.__init__", side_effect=mockNone) mocker.patch("DIRAC.WorkloadManagementSystem.Client.JobState.JobState.TaskQueueDB.__init__", side_effect=mockNone) inputData = InputData() js = CachedJobState(1) js.setAttribute("JobType", "User") manifest = JobManifest() for varName, varValue in manifestOptions.items(): manifest.setOption(varName, varValue) js.setManifest(manifest) res = inputData._getInputSandbox(js) assert res["OK"] is True assert res["Value"] == expected