Skip to content

Commit 556132c

Browse files
committed
fix tests in director-v2
1 parent fe07ac6 commit 556132c

File tree

1 file changed

+5
-4
lines changed

1 file changed

+5
-4
lines changed

services/director-v2/tests/unit/with_dbs/comp_scheduler/test_api_rpc_computations_tasks.py

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -28,19 +28,20 @@ async def test_get_computation_task_log_file_ids(
2828
fake_workbench_without_outputs: dict[str, Any],
2929
fake_workbench_adjacency: dict[str, Any],
3030
create_registered_user: Callable[..., dict[str, Any]],
31-
project: Callable[..., Awaitable[ProjectAtDB]],
31+
create_project: Callable[..., Awaitable[ProjectAtDB]],
3232
create_pipeline: Callable[..., Awaitable[CompPipelineAtDB]],
33-
create_tasks: Callable[..., Awaitable[list[CompTaskAtDB]]],
33+
create_tasks_from_project: Callable[..., Awaitable[list[CompTaskAtDB]]],
3434
create_comp_run: Callable[..., Awaitable[CompRunsAtDB]],
3535
rpc_client: RabbitMQRPCClient,
36+
with_product: dict[str, Any],
3637
):
3738
user = create_registered_user()
38-
proj = await project(user, workbench=fake_workbench_without_outputs)
39+
proj = await create_project(user, workbench=fake_workbench_without_outputs)
3940
await create_pipeline(
4041
project_id=f"{proj.uuid}",
4142
dag_adjacency_list=fake_workbench_adjacency,
4243
)
43-
comp_tasks = await create_tasks(
44+
comp_tasks = await create_tasks_from_project(
4445
user=user, project=proj, state=StateType.PUBLISHED, progress=None
4546
)
4647
comp_runs = await create_comp_run(

0 commit comments

Comments
 (0)