diff --git a/diracx-routers/src/diracx/routers/jobs/status.py b/diracx-routers/src/diracx/routers/jobs/status.py index 4d53a449..2c5e2f74 100644 --- a/diracx-routers/src/diracx/routers/jobs/status.py +++ b/diracx-routers/src/diracx/routers/jobs/status.py @@ -12,8 +12,10 @@ ) from diracx.logic.jobs.status import remove_jobs as remove_jobs_bl from diracx.logic.jobs.status import reschedule_jobs as reschedule_jobs_bl +from diracx.logic.jobs.status import ( + set_job_parameters_or_attributes as set_job_parameters_or_attributes_bl, +) from diracx.logic.jobs.status import set_job_statuses as set_job_statuses_bl -from diracx.logic.jobs.status import set_job_parameters_or_attributes as set_job_parameters_or_attributes_bl from ..dependencies import ( Config, @@ -136,6 +138,4 @@ async def patch_metadata( check_permissions: CheckWMSPolicyCallable, ): await check_permissions(action=ActionType.MANAGE, job_db=job_db, job_ids=updates) - await set_job_parameters_or_attributes_bl( - updates, job_db, job_parameters_db - ) + await set_job_parameters_or_attributes_bl(updates, job_db, job_parameters_db) diff --git a/diracx-routers/tests/jobs/test_status.py b/diracx-routers/tests/jobs/test_status.py index acc9aa09..be5fabe2 100644 --- a/diracx-routers/tests/jobs/test_status.py +++ b/diracx-routers/tests/jobs/test_status.py @@ -942,6 +942,7 @@ def test_remove_jobs_valid_job_ids( # r = normal_user_client.get(f"/api/jobs/{job_id}/status") # assert r.status_code == HTTPStatus.NOT_FOUND, r.json() + def test_patch_metadata(normal_user_client: TestClient, valid_job_id: int): # Arrange r = normal_user_client.post( @@ -1003,4 +1004,4 @@ def test_patch_metadata(normal_user_client: TestClient, valid_job_id: int): assert datetime.fromisoformat( r.json()[0]["HeartBeatTime"] ) == datetime.fromisoformat(hbt) - assert r.json()[0]["UserPriority"] == 2 \ No newline at end of file + assert r.json()[0]["UserPriority"] == 2