|
1 | | -""" unit test (pytest) of JobManager service |
| 1 | +""" unit test (pytest) of JobAdministration module |
2 | 2 | """ |
3 | 3 |
|
4 | 4 | from unittest.mock import MagicMock |
|
22 | 22 | ([1, 2], "Kill", [1], True, [1]), |
23 | 23 | ([1, 2], "Kill", [2], True, [2]), |
24 | 24 | ([1, 2], "Kill", [], True, []), |
25 | | - ([1, 2], "Kill", [1,2], True, [1, 2]), |
| 25 | + ([1, 2], "Kill", [1, 2], True, [1, 2]), |
26 | 26 | ], |
27 | 27 | ) |
28 | 28 | def test___kill_delete_jobs(mocker, jobIDs_list, right, filtered_jobs, expected_res, expected_value): |
29 | 29 | mocker.patch("DIRAC.WorkloadManagementSystem.Utilities.jobAdministration.JobDB", MagicMock()) |
30 | 30 | mocker.patch("DIRAC.WorkloadManagementSystem.Utilities.jobAdministration.TaskQueueDB", MagicMock()) |
31 | 31 | mocker.patch("DIRAC.WorkloadManagementSystem.Utilities.jobAdministration.PilotAgentsDB", MagicMock()) |
32 | 32 | mocker.patch("DIRAC.WorkloadManagementSystem.Utilities.jobAdministration.StorageManagementDB", MagicMock()) |
33 | | - mocker.patch("DIRAC.WorkloadManagementSystem.Utilities.jobAdministration.filterJobStateTransition", return_value={"OK": True, "Value": filtered_jobs}) |
| 33 | + mocker.patch( |
| 34 | + "DIRAC.WorkloadManagementSystem.Utilities.jobAdministration.filterJobStateTransition", |
| 35 | + return_value={"OK": True, "Value": filtered_jobs}, |
| 36 | + ) |
34 | 37 |
|
35 | 38 | res = kill_delete_jobs(right, jobIDs_list) |
36 | 39 | assert res["OK"] == expected_res |
|
0 commit comments