fixing node creation mock

Author: giancarloromeo

packages/postgres-database/src/simcore_postgres_database/storage_models.py

@@ -7,6 +7,7 @@
 from .models.file_meta_data import file_meta_data
 from .models.groups import groups, user_to_groups
 from .models.projects import projects
+from .models.projects_nodes import projects_nodes
 from .models.tokens import tokens
 from .models.users import users
 
@@ -15,6 +16,7 @@
     "file_meta_data",
     "metadata",
     "projects",
+    "projects_nodes",
     "users",
     "groups",
     "user_to_groups",

services/storage/tests/fixtures/data_models.py

@@ -22,7 +22,7 @@
 from pytest_simcore.helpers.faker_factories import random_project, random_user
 from servicelib.utils import limited_gather
 from simcore_postgres_database.models.project_to_groups import project_to_groups
-from simcore_postgres_database.storage_models import projects, users
+from simcore_postgres_database.storage_models import projects, projects_nodes, users
 from sqlalchemy.dialects.postgresql import insert as pg_insert
 
 from ..helpers.utils import get_updated_project
@@ -228,34 +228,25 @@ async def _() -> None:
 async def create_project_node(
     user_id: UserID, aiopg_engine: Engine, faker: Faker
 ) -> Callable[..., Awaitable[NodeID]]:
-    async def _creator(
+    async def _(
         project_id: ProjectID, node_id: NodeID | None = None, **kwargs
     ) -> NodeID:
         async with aiopg_engine.acquire() as conn:
-            result = await conn.execute(
-                sa.select(projects.c.workbench).where(
-                    projects.c.uuid == f"{project_id}"
-                )
-            )
-            row = await result.fetchone()
-            assert row
-            project_workbench: dict[str, Any] = row[projects.c.workbench]
             new_node_id = node_id or NodeID(f"{faker.uuid4()}")
             node_data = {
                 "key": "simcore/services/frontend/file-picker",
                 "version": "1.0.0",
                 "label": "pytest_fake_node",
             }
             node_data.update(**kwargs)
-            project_workbench.update({f"{new_node_id}": node_data})
             await conn.execute(
-                projects.update()
-                .where(projects.c.uuid == f"{project_id}")
-                .values(workbench=project_workbench)
+                projects_nodes.insert().values(
+                    project_uuid=project_id, node_id=new_node_id, **node_data
+                )
             )
         return new_node_id
 
-    return _creator
+    return _
 
 
 @pytest.fixture