fixed broken tests

ITISFoundation · Dec 13, 2024 · 4204990 · 4204990
1 parent d6b89a9
commit 4204990
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/services/director-v2/tests/unit/with_dbs/comp_scheduler/test_scheduler_dask.py b/services/director-v2/tests/unit/with_dbs/comp_scheduler/test_scheduler_dask.py
@@ -255,6 +255,7 @@ async def _return_tasks_pending(job_ids: list[str]) -> list[DaskClientTaskState]
                 callback=mock.ANY,
                 metadata=mock.ANY,
                 hardware_info=mock.ANY,
+                run_id=mock.ANY,
             )
             for p in expected_pending_tasks
         ],
@@ -654,6 +655,7 @@ async def _return_random_task_result(job_id) -> TaskOutputData:
         callback=mock.ANY,
         metadata=mock.ANY,
         hardware_info=mock.ANY,
+        run_id=mock.ANY,
     )
     mocked_dask_client.send_computation_tasks.reset_mock()
     mocked_dask_client.get_tasks_status.assert_has_calls(