fix: update environment and operations for performance tests

Author: hubert-marek

backend/src/platform/isolationEngine/environment.py

@@ -65,17 +65,30 @@ def _copy_custom_indexes(self, src_schema: str, dst_schema: str) -> None:
             ).fetchall()
             for idx_name, idx_def in rows:
                 # Rewrite the CREATE INDEX to target the new schema
-                new_def = idx_def.replace(
-                    f" ON {src_schema}.", f" ON {dst_schema}."
-                )
+                new_def = idx_def.replace(f" ON {src_schema}.", f" ON {dst_schema}.")
                 # Avoid name collisions by prefixing with target schema
                 new_idx_name = f"{dst_schema}_{idx_name}"
+                # Handle both CREATE INDEX and CREATE UNIQUE INDEX
+                new_def = new_def.replace(
+                    f"CREATE UNIQUE INDEX {idx_name}",
+                    f"CREATE UNIQUE INDEX IF NOT EXISTS {new_idx_name}",
+                )
                 new_def = new_def.replace(
                     f"CREATE INDEX {idx_name}",
                     f"CREATE INDEX IF NOT EXISTS {new_idx_name}",
                 )
                 try:
-                    conn.execute(text(new_def))
+                    # Use a savepoint so a single index failure doesn't abort
+                    # the entire transaction and block subsequent indexes.
+                    nested = conn.begin_nested()
+                    try:
+                        conn.execute(text(new_def))
+                        nested.commit()
+                    except Exception as exc:
+                        nested.rollback()
+                        logger.warning(
+                            f"Could not copy index {idx_name} to {dst_schema}: {exc}"
+                        )
                 except Exception as exc:
                     logger.warning(
                         f"Could not copy index {idx_name} to {dst_schema}: {exc}"

backend/src/services/box/database/operations.py

@@ -547,6 +547,7 @@ def update_folder(
             )
         old_path = folder.path or "/"
         folder.parent_id = parent_id
+        folder.parent = new_parent  # Keep ORM relationship in sync with FK
         folder.path = _compute_folder_path(new_parent)
         # Cascade path changes to all descendants
         _cascade_path_update(session, folder, old_path)
@@ -1019,6 +1020,7 @@ def update_file(
                 conflicts=[{"type": "file", "id": existing.id, "name": existing.name}],
             )
         file.parent_id = parent_id
+        file.parent = new_parent  # Keep ORM relationship in sync with FK
         file.path = _compute_folder_path(new_parent)  # new_parent.path + new_parent.id
     if tags is not None:
         file.tags = tags