asynpg

Author: sanderegg

services/storage/tests/fixtures/data_models.py

@@ -82,9 +82,7 @@ async def _creator(**kwargs) -> dict[str, Any]:
                 .values(**random_project(**prj_config))
                 .returning(sa.literal_column("*"))
             )
-            await conn.commit()
-            row = result.fetchone()
-            assert row
+            row = result.one()
             created_project_uuids.append(row.uuid)
             return dict(row._asdict())
 
@@ -111,17 +109,15 @@ async def _creator(
                 .values(
                     project_uuid=f"{project_id}",
                     gid=sa.select(users.c.primary_gid)
-                    .where(users.c.id == f"{user_id}")
+                    .where(users.c.id == user_id)
                     .scalar_subquery(),
                     read=read,
                     write=write,
                     delete=delete,
                 )
                 .returning(sa.literal_column("*"))
             )
-            await conn.commit()
-            row = result.fetchone()
-            assert row
+            row = result.one()
             _created.append((row.project_uuid, row.gid))
 
     yield _creator