diff --git a/services/director-v2/tests/unit/with_dbs/comp_scheduler/test_worker.py b/services/director-v2/tests/unit/with_dbs/comp_scheduler/test_worker.py index e9bc9e831a80..9eb301e09101 100644 --- a/services/director-v2/tests/unit/with_dbs/comp_scheduler/test_worker.py +++ b/services/director-v2/tests/unit/with_dbs/comp_scheduler/test_worker.py @@ -71,7 +71,7 @@ async def test_worker_properly_autocalls_scheduler_api( use_on_demand_clusters=False, ) mocked_get_scheduler_worker.assert_called_once_with(initialized_app) - mocked_get_scheduler_worker.return_value.schedule_pipeline.assert_called_once_with( + mocked_get_scheduler_worker.return_value.apply.assert_called_once_with( user_id=published_project.project.prj_owner, project_id=published_project.project.uuid, iteration=1,