continue fixing

Author: giancarloromeo

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

@@ -281,6 +281,7 @@ async def running_project(
         tasks=await create_tasks(
             user=user,
             project=created_project,
+            workbench=fake_workbench_without_outputs,
             state=StateType.RUNNING,
             progress=0.0,
             start=now_time,
@@ -318,6 +319,7 @@ async def running_project_mark_for_cancellation(
         tasks=await create_tasks(
             user=user,
             project=created_project,
+            workbench=fake_workbench_without_outputs,
             state=StateType.RUNNING,
             progress=0.0,
             start=now_time,

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

@@ -193,6 +193,7 @@ def test_project_save_state_ok(
     mock_save_service_state: None,
     cli_runner: CliRunner,
     project_at_db: ProjectAtDB,
+    fake_workbench_without_outputs: dict[str, Any],
     capsys: pytest.CaptureFixture,
 ):
     with capsys.disabled() as _disabled:
@@ -204,8 +205,8 @@ def test_project_save_state_ok(
     print(result.stdout)
     assert result.exit_code == os.EX_OK, _format_cli_error(result)
     assert result.stdout.endswith(f"Save complete for project {project_at_db.uuid}\n")
-    for node_uuid, node_content in project_at_db.workbench.items():
-        assert f"Saving state for {node_uuid} {node_content.label}" in result.stdout
+    for node_uuid, node_content in fake_workbench_without_outputs.items():
+        assert f"Saving state for {node_uuid} {node_content['label']}" in result.stdout
 
     assert (
         f"Saving project '{project_at_db.uuid}' - '{project_at_db.name}'"