Add PatchEntityDetailsError

Author: kmcginnes

packages/graph-explorer/src/connector/queries/patchEntityDetails.test.ts

@@ -3,7 +3,10 @@ import {
   createTestableVertex,
   FakeExplorer,
 } from "@/utils/testing";
-import { patchEntityDetails } from "./patchEntityDetails";
+import {
+  patchEntityDetails,
+  PatchEntityDetailsError,
+} from "./patchEntityDetails";
 import { createQueryClient } from "@/core/queryClient";
 import {
   createResultScalar,
@@ -222,4 +225,55 @@ describe("patchEntityDetails", () => {
 
     expect(result).toStrictEqual([expectedBundle]);
   });
+
+  it("should throw when source vertex not found", async () => {
+    const explorer = new FakeExplorer();
+    const client = createQueryClient({ explorer });
+
+    const edge = createTestableEdge();
+    explorer.addTestableEdge(edge);
+    explorer.vertexMap.delete(edge.source.id);
+
+    await expect(() =>
+      patchEntityDetails(client, [edge.asFragmentResult()])
+    ).rejects.toThrow(
+      new PatchEntityDetailsError(
+        "Failed to fetch the details of some of the entity results (vertices: 1, edges: 0)."
+      )
+    );
+  });
+
+  it("should throw when target vertex not found", async () => {
+    const explorer = new FakeExplorer();
+    const client = createQueryClient({ explorer });
+
+    const edge = createTestableEdge();
+    explorer.addTestableEdge(edge);
+    explorer.vertexMap.delete(edge.target.id);
+
+    await expect(() =>
+      patchEntityDetails(client, [edge.asFragmentResult()])
+    ).rejects.toThrow(
+      new PatchEntityDetailsError(
+        "Failed to fetch the details of some of the entity results (vertices: 1, edges: 0)."
+      )
+    );
+  });
+
+  it("should throw when edge full details not found", async () => {
+    const explorer = new FakeExplorer();
+    const client = createQueryClient({ explorer });
+
+    const edge = createTestableEdge();
+    explorer.addTestableEdge(edge);
+    explorer.edgeMap.delete(edge.id);
+
+    await expect(() =>
+      patchEntityDetails(client, [edge.asFragmentResult()])
+    ).rejects.toThrow(
+      new PatchEntityDetailsError(
+        "Failed to fetch the details of some of the entity results (vertices: 0, edges: 1)."
+      )
+    );
+  });
 });

packages/graph-explorer/src/connector/queries/patchEntityDetails.ts

@@ -39,7 +39,9 @@ export async function patchEntityDetails(
       missingVertices,
       missingEdges,
     });
-    throw new Error("Failed to fetch entity details");
+    throw new PatchEntityDetailsError(
+      `Failed to fetch the details of some of the entity results (vertices: ${missingVertices.length}, edges: ${missingEdges.length}).`
+    );
   }
 
   // Create a mapping from vertex ID to full vertex details
@@ -76,7 +78,7 @@ function patchVertex(
   const fullVertex = vertexDetailsMap.get(vertex.id);
 
   if (!fullVertex) {
-    throw new Error("Failed to fetch vertex details");
+    throw new PatchEntityDetailsError("Failed to fetch vertex details");
   }
 
   return createPatchedResultVertex({
@@ -94,8 +96,21 @@ function patchEdge(
   const fullSource = vertexDetailsMap.get(edge.sourceId);
   const fullTarget = vertexDetailsMap.get(edge.targetId);
 
-  if (!fullEdge || !fullSource || !fullTarget) {
-    throw new Error("Failed to fetch edge details");
+  if (!fullEdge) {
+    throw new PatchEntityDetailsError(
+      "Could not find the full details of the edge"
+    );
+  }
+
+  if (!fullSource) {
+    throw new PatchEntityDetailsError(
+      "Could not find the full details of the source vertex"
+    );
+  }
+  if (!fullTarget) {
+    throw new PatchEntityDetailsError(
+      "Could not find the full details of the target vertex"
+    );
   }
 
   return createPatchedResultEdge({
@@ -118,3 +133,11 @@ function patchBundle(
     ),
   };
 }
+
+export class PatchEntityDetailsError extends Error {
+  constructor(message: string) {
+    super(message);
+    this.name = "PatchEntityDetailsError";
+    Object.setPrototypeOf(this, PatchEntityDetailsError.prototype);
+  }
+}

packages/graph-explorer/src/utils/createDisplayError.ts

@@ -1,6 +1,7 @@
 import { ZodError } from "zod";
 import { NetworkError } from "./NetworkError";
 import { isCancellationError } from "./isCancellationError";
+import { PatchEntityDetailsError } from "@/connector/queries/patchEntityDetails";
 
 export type DisplayError = {
   title: string;
@@ -118,6 +119,13 @@ export function createDisplayError(error: any): DisplayError {
     };
   }
 
+  if (error instanceof PatchEntityDetailsError) {
+    return {
+      title: "Failed to update entity details",
+      message: error.message,
+    };
+  }
+
   if (error instanceof ZodError) {
     return {
       title: "Unrecognized Result Format",