fix test

ITISFoundation · Jul 3, 2023 · e12bb89 · e12bb89
1 parent 11039c0
commit e12bb89
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/services/director-v2/tests/unit/with_dbs/test_modules_comp_scheduler_dask_scheduler.py b/services/director-v2/tests/unit/with_dbs/test_modules_comp_scheduler_dask_scheduler.py
@@ -435,11 +435,12 @@ async def _assert_schedule_pipeline_PENDING(
     mocked_dask_client.send_computation_tasks.assert_has_calls(
         calls=[
             mock.call(
-                published_project.project.prj_owner,
+                user_id=published_project.project.prj_owner,
                 project_id=published_project.project.uuid,
                 cluster_id=DEFAULT_CLUSTER_ID,
                 tasks={f"{p.node_id}": p.image},
                 callback=scheduler._wake_up_scheduler_now,
+                metadata=mock.ANY,
             )
             for p in expected_pending_tasks
         ],
@@ -596,6 +597,7 @@ async def _return_random_task_result(job_id) -> TaskOutputData:
             f"{next_pending_task.node_id}": next_pending_task.image,
         },
         callback=scheduler._wake_up_scheduler_now,
+        metadata=mock.ANY,
     )
     mocked_dask_client.send_computation_tasks.reset_mock()
     mocked_dask_client.get_tasks_status.assert_has_calls(