diff --git a/tests/unit/task/task_builder_test.py b/tests/unit/task/task_builder_test.py index 088e6faa..4175ffff 100644 --- a/tests/unit/task/task_builder_test.py +++ b/tests/unit/task/task_builder_test.py @@ -205,7 +205,7 @@ async def test_job_parameter_is_injected(self, task_config: TaskConfig, mocked_j job_handler = task_builder.build_job_handler(self.function_with_job_parameter, task_config) job = await job_handler(mocked_job_with_adapter) - assert job.variables["job"] == mocked_job_with_adapter + assert job.variables["received_job"] == mocked_job_with_adapter @pytest.mark.asyncio async def test_job_parameter_retains_variables(self, task_config: TaskConfig, mocked_job_with_adapter: Job): @@ -215,7 +215,7 @@ async def test_job_parameter_retains_variables(self, task_config: TaskConfig, mo job_handler = task_builder.build_job_handler(self.function_with_job_parameter, task_config) job = await job_handler(mocked_job_with_adapter) - assert job.variables["job"].variables == expected_variables + assert job.variables["received_job"].variables == expected_variables def function_with_job_parameter(x: int, job: Job): - return {"job": job} + return {"received_job": job}