Skip to content

Commit 828be6e

Browse files
continue fixing
1 parent 25c2d5d commit 828be6e

File tree

1 file changed

+6
-2
lines changed

1 file changed

+6
-2
lines changed

services/director-v2/tests/unit/with_dbs/test_utils_rabbitmq.py

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -96,6 +96,7 @@ async def project(
9696
async def tasks(
9797
user: dict[str, Any],
9898
project: ProjectAtDB,
99+
fake_workbench_without_outputs: dict[str, Any],
99100
fake_workbench_adjacency: dict[str, Any],
100101
create_pipeline: Callable[..., Awaitable[CompPipelineAtDB]],
101102
create_tasks: Callable[..., Awaitable[list[CompTaskAtDB]]],
@@ -104,7 +105,7 @@ async def tasks(
104105
project_id=f"{project.uuid}",
105106
dag_adjacency_list=fake_workbench_adjacency,
106107
)
107-
comp_tasks = await create_tasks(user, project)
108+
comp_tasks = await create_tasks(user, project, fake_workbench_without_outputs)
108109
assert len(comp_tasks) > 0
109110
return comp_tasks
110111

@@ -162,6 +163,7 @@ async def test_publish_service_resource_tracking_started(
162163
create_rabbitmq_client: Callable[[str], RabbitMQClient],
163164
user: dict[str, Any],
164165
project: ProjectAtDB,
166+
fake_workbench_without_outputs: dict[str, Any],
165167
simcore_user_agent: str,
166168
tasks: list[CompTaskAtDB],
167169
mocked_message_parser: mock.AsyncMock,
@@ -193,7 +195,9 @@ async def test_publish_service_resource_tracking_started(
193195
project_id=project.uuid,
194196
project_name=project.name,
195197
node_id=random_task.node_id,
196-
node_name=project.workbench[NodeIDStr(f"{random_task.node_id}")].label,
198+
node_name=fake_workbench_without_outputs[
199+
NodeIDStr(f"{random_task.node_id}")
200+
].label,
197201
parent_project_id=None,
198202
parent_node_id=None,
199203
root_parent_project_id=None,

0 commit comments

Comments
 (0)