minor

Author: pcrespov

services/director-v2/tests/integration/01/test_computation_api.py

@@ -416,12 +416,12 @@ async def test_run_partial_computation(
 
     def _convert_to_pipeline_details(
         project: ProjectAtDB,
-        exp_pipeline_adj_list: dict[int, list[int]],
-        exp_node_states: dict[int, dict[str, Any]],
+        expected_pipeline_adj_list: dict[int, list[int]],
+        expected_node_states: dict[int, dict[str, Any]],
     ) -> PipelineDetails:
         workbench_node_uuids = list(project.workbench.keys())
         converted_adj_list: dict[NodeID, list[NodeID]] = {}
-        for node_key, next_nodes in exp_pipeline_adj_list.items():
+        for node_key, next_nodes in expected_pipeline_adj_list.items():
             converted_adj_list[NodeID(workbench_node_uuids[node_key])] = [
                 NodeID(workbench_node_uuids[n]) for n in next_nodes
             ]
@@ -434,7 +434,7 @@ def _convert_to_pipeline_details(
                 currentStatus=s.get("currentStatus", RunningState.NOT_STARTED),
                 progress=s.get("progress"),
             )
-            for n, s in exp_node_states.items()
+            for n, s in expected_node_states.items()
         }
         pipeline_progress = 0
         for node_id in converted_adj_list:

services/director-v2/tests/integration/conftest.py

@@ -37,7 +37,7 @@ def update_project_workbench_with_comp_tasks(
     postgres_db: sa.engine.Engine,
 ) -> Callable:
     def _updator(project_uuid: str):
-        with postgres_db.connect() as con:
+        with postgres_db.connect() as con, con.begin():
 
             # select all projects_nodes for this project
             result = con.execute(