@apollo/client-ai-apps 0.3.2 → 0.4.0

package/src/vite/__tests__/application_manifest_plugin.test.ts

@@ -1,8 +1,10 @@
-import { expect, test, vi, describe, beforeEach, Mock } from "vitest";
-import { ApplicationManifestPlugin } from "../application_manifest_plugin";
+import type { Mock } from "vitest";
+import { expect, test, vi, describe, beforeEach } from "vitest";
+import { ApplicationManifestPlugin } from "../application_manifest_plugin.js";
 import fs from "fs";
 import * as glob from "glob";
 import path from "path";
+import type { ManifestWidgetSettings } from "../../types/application-manifest.js";
 
 const root = process.cwd();
 
@@ -31,12 +33,7 @@ vi.mock(import("path"), async (importOriginal) => {
   };
 });
 
-vi.mock(import("glob"), async (importOriginal) => {
-  const actual = await importOriginal();
-  return {
-    glob: vi.fn(),
-  };
-});
+vi.mock(import("glob"));
 
 beforeEach(() => {
   vi.clearAllMocks();
@@ -44,19 +41,39 @@ beforeEach(() => {
 
 describe("buildStart", () => {
   test("Should write to dev application manifest file when using a serve command", async () => {
-    vi.spyOn(fs, "readFileSync").mockImplementation((path) => {
-      if (path === "package.json") {
-        return JSON.stringify({});
-      } else if (path === root + "/my-component.tsx") {
-        return `
-            const MY_QUERY = gql\`query HelloWorldQuery($name: string!) @tool(name: "hello-world", description: "This is an awesome tool!", extraInputs: [{
-              name: "doStuff",
-              type: "boolean",
-              description: "Should we do stuff?"
-            }]) { helloWorld(name: $name) }\`;
-        `;
-      }
+    mockReadFile({
+      "package.json": JSON.stringify({
+        labels: {
+          toolInvocation: {
+            invoking: "Testing global...",
+            invoked: "Tested global!",
+          },
+        },
+        widgetSettings: {
+          description: "Test",
+          domain: "https://example.com",
+          prefersBorder: true,
+        } satisfies ManifestWidgetSettings,
+      }),
+      [`${root}/my-component.tsx`]: `
+const MY_QUERY = gql\`query HelloWorldQuery($name: string!) @tool(
+  name: "hello-world",
+  description: "This is an awesome tool!",
+  extraInputs: [{
+    name: "doStuff",
+    type: "boolean",
+    description: "Should we do stuff?"
+  }],
+  labels: {
+    toolInvocation: {
+      invoking: "Testing tool...",
+      invoked: "Tested tool!"
+    }
+  }
+) { helloWorld(name: $name) }\`;
+      `,
     });
+
     vi.spyOn(glob, "glob").mockImplementation(() =>
       Promise.resolve(["my-component.tsx"])
     );
@@ -81,10 +98,16 @@ describe("buildStart", () => {
       {
         "csp": {
           "connectDomains": [],
+          "frameDomains": [],
+          "redirectDomains": [],
           "resourceDomains": [],
         },
         "format": "apollo-ai-app-manifest",
         "hash": "abc",
+        "labels": {
+          "toolInvocation/invoked": "Tested global!",
+          "toolInvocation/invoking": "Testing global...",
+        },
         "operations": [
           {
             "body": "query HelloWorldQuery($name: string!) {
@@ -103,6 +126,10 @@ describe("buildStart", () => {
                     "type": "boolean",
                   },
                 ],
+                "labels": {
+                  "toolInvocation/invoked": "Tested tool!",
+                  "toolInvocation/invoking": "Testing tool...",
+                },
                 "name": "hello-world",
               },
             ],
@@ -114,19 +141,21 @@ describe("buildStart", () => {
         ],
         "resource": "http://localhost:undefined",
         "version": "1",
+        "widgetSettings": {
+          "description": "Test",
+          "domain": "https://example.com",
+          "prefersBorder": true,
+        },
       }
     `);
   });
 
   test("Should NOT write to dev application manifest file when using a build command", async () => {
-    vi.spyOn(fs, "readFileSync").mockImplementation((path) => {
-      if (path === "package.json") {
-        return JSON.stringify({});
-      } else if (path === "my-component.tsx") {
-        return `
-            const MY_QUERY = gql\`query HelloWorldQuery @tool(name: "hello-world", description: "This is an awesome tool!") { helloWorld }\`;
-        `;
-      }
+    mockReadFile({
+      "package.json": JSON.stringify({}),
+      "my-component.tsx": `
+        const MY_QUERY = gql\`query HelloWorldQuery @tool(name: "hello-world", description: "This is an awesome tool!") { helloWorld }\`;
+      `,
     });
     vi.spyOn(glob, "glob").mockImplementation(() =>
       Promise.resolve(["my-component.tsx"])
@@ -142,14 +171,11 @@ describe("buildStart", () => {
   });
 
   test("Should not process files that do not contain gql tags", async () => {
-    vi.spyOn(fs, "readFileSync").mockImplementation((path) => {
-      if (path === "package.json") {
-        return JSON.stringify({});
-      } else if (path === root + "/my-component.tsx") {
-        return `
-            const MY_QUERY = \`query HelloWorldQuery @tool(name: "hello-world", description: "This is an awesome tool!") { helloWorld }\`;
-        `;
-      }
+    mockReadFile({
+      "package.json": JSON.stringify({}),
+      [`${root}/my-component.tsx`]: `
+        const MY_QUERY = \`query HelloWorldQuery @tool(name: "hello-world", description: "This is an awesome tool!") { helloWorld }\`;
+      `,
     });
     vi.spyOn(glob, "glob").mockImplementation(() =>
       Promise.resolve(["my-component.tsx"])
@@ -175,6 +201,8 @@ describe("buildStart", () => {
       {
         "csp": {
           "connectDomains": [],
+          "frameDomains": [],
+          "redirectDomains": [],
           "resourceDomains": [],
         },
         "format": "apollo-ai-app-manifest",
@@ -187,14 +215,11 @@ describe("buildStart", () => {
   });
 
   test("Should capture queries when writing to manifest file", async () => {
-    vi.spyOn(fs, "readFileSync").mockImplementation((path) => {
-      if (path === "package.json") {
-        return JSON.stringify({});
-      } else if (path === root + "/my-component.tsx") {
-        return `
-            const MY_QUERY = gql\`query HelloWorldQuery { helloWorld }\`;
-        `;
-      }
+    mockReadFile({
+      "package.json": JSON.stringify({}),
+      [`${root}/my-component.tsx`]: `
+        const MY_QUERY = gql\`query HelloWorldQuery { helloWorld }\`;
+      `,
     });
     vi.spyOn(glob, "glob").mockImplementation(() =>
       Promise.resolve(["my-component.tsx"])
@@ -220,6 +245,8 @@ describe("buildStart", () => {
       {
         "csp": {
           "connectDomains": [],
+          "frameDomains": [],
+          "redirectDomains": [],
           "resourceDomains": [],
         },
         "format": "apollo-ai-app-manifest",
@@ -244,14 +271,11 @@ describe("buildStart", () => {
   });
 
   test("Should capture queries as prefetch when query is marked with @prefetch directive", async () => {
-    vi.spyOn(fs, "readFileSync").mockImplementation((path) => {
-      if (path === "package.json") {
-        return JSON.stringify({});
-      } else if (path === root + "/my-component.tsx") {
-        return `
-            const MY_QUERY = gql\`query HelloWorldQuery @prefetch { helloWorld }\`;
-        `;
-      }
+    mockReadFile({
+      "package.json": JSON.stringify({}),
+      [`${root}/my-component.tsx`]: `
+        const MY_QUERY = gql\`query HelloWorldQuery @prefetch { helloWorld }\`;
+      `,
     });
     vi.spyOn(glob, "glob").mockImplementation(() =>
       Promise.resolve(["my-component.tsx"])
@@ -277,6 +301,8 @@ describe("buildStart", () => {
       {
         "csp": {
           "connectDomains": [],
+          "frameDomains": [],
+          "redirectDomains": [],
           "resourceDomains": [],
         },
         "format": "apollo-ai-app-manifest",
@@ -302,15 +328,12 @@ describe("buildStart", () => {
   });
 
   test("Should error when multiple operations are marked with @prefetch", async () => {
-    vi.spyOn(fs, "readFileSync").mockImplementation((path) => {
-      if (path === "package.json") {
-        return JSON.stringify({});
-      } else if (path === "my-component.tsx") {
-        return `
-            const MY_QUERY = gql\`query HelloWorldQuery @prefetch { helloWorld }\`;
-            const MY_QUERY2 = gql\`query HelloWorldQuery2 @prefetch { helloWorld }\`;
-        `;
-      }
+    mockReadFile({
+      "package.json": JSON.stringify({}),
+      "my-component.tsx": `
+        const MY_QUERY = gql\`query HelloWorldQuery @prefetch { helloWorld }\`;
+        const MY_QUERY2 = gql\`query HelloWorldQuery2 @prefetch { helloWorld }\`;
+      `,
     });
     vi.spyOn(glob, "glob").mockImplementation(() =>
       Promise.resolve(["my-component.tsx"])
@@ -328,14 +351,11 @@ describe("buildStart", () => {
   });
 
   test("Should capture mutations when writing to manifest file", async () => {
-    vi.spyOn(fs, "readFileSync").mockImplementation((path) => {
-      if (path === "package.json") {
-        return JSON.stringify({});
-      } else if (path === root + "/my-component.tsx") {
-        return `
-            const MY_QUERY = gql\`mutation HelloWorldQuery @tool(name: "hello-world", description: "This is an awesome tool!") { helloWorld }\`;
-        `;
-      }
+    mockReadFile({
+      "package.json": JSON.stringify({}),
+      [`${root}/my-component.tsx`]: `
+        const MY_QUERY = gql\`mutation HelloWorldQuery @tool(name: "hello-world", description: "This is an awesome tool!") { helloWorld }\`;
+      `,
     });
     vi.spyOn(glob, "glob").mockImplementation(() =>
       Promise.resolve(["my-component.tsx"])
@@ -361,6 +381,8 @@ describe("buildStart", () => {
       {
         "csp": {
           "connectDomains": [],
+          "frameDomains": [],
+          "redirectDomains": [],
           "resourceDomains": [],
         },
         "format": "apollo-ai-app-manifest",
@@ -390,14 +412,11 @@ describe("buildStart", () => {
   });
 
   test("Should throw error when a subscription operation type is discovered", async () => {
-    vi.spyOn(fs, "readFileSync").mockImplementation((path) => {
-      if (path === "package.json") {
-        return JSON.stringify({});
-      } else if (path === "my-component.tsx") {
-        return `
-            const MY_QUERY = gql\`subscription HelloWorldQuery @tool(name: "hello-world", description: "This is an awesome tool!") { helloWorld }\`;
-        `;
-      }
+    mockReadFile({
+      "package.json": JSON.stringify({}),
+      "my-component.tsx": `
+        const MY_QUERY = gql\`subscription HelloWorldQuery @tool(name: "hello-world", description: "This is an awesome tool!") { helloWorld }\`;
+      `,
     });
     vi.spyOn(glob, "glob").mockImplementation(() =>
       Promise.resolve(["my-component.tsx"])
@@ -416,18 +435,15 @@ describe("buildStart", () => {
   });
 
   test("Should use custom entry point when in serve mode and provided in package.json", async () => {
-    vi.spyOn(fs, "readFileSync").mockImplementation((path) => {
-      if (path === "package.json") {
-        return JSON.stringify({
-          entry: {
-            staging: "http://staging.awesome.com",
-          },
-        });
-      } else if (path === "my-component.tsx") {
-        return `
-            const MY_QUERY = gql\`query HelloWorldQuery @tool(name: "hello-world", description: "This is an awesome tool!") { helloWorld }\`;
-        `;
-      }
+    mockReadFile({
+      "package.json": JSON.stringify({
+        entry: {
+          staging: "http://staging.awesome.com",
+        },
+      }),
+      "my-component.tsx": `
+        const MY_QUERY = gql\`query HelloWorldQuery @tool(name: "hello-world", description: "This is an awesome tool!") { helloWorld }\`;
+      `,
     });
     vi.spyOn(glob, "glob").mockImplementation(() =>
       Promise.resolve(["my-component.tsx"])
@@ -444,21 +460,18 @@ describe("buildStart", () => {
     });
     await plugin.buildStart();
 
-    let [file, content] = (fs.writeFileSync as unknown as Mock).mock.calls[0];
+    let [, content] = (fs.writeFileSync as unknown as Mock).mock.calls[0];
     let contentObj = JSON.parse(content);
 
     expect(contentObj.resource).toBe("http://staging.awesome.com");
   });
 
   test("Should use https when enabled in server config", async () => {
-    vi.spyOn(fs, "readFileSync").mockImplementation((path) => {
-      if (path === "package.json") {
-        return JSON.stringify({});
-      } else if (path === "my-component.tsx") {
-        return `
-            const MY_QUERY = gql\`query HelloWorldQuery @tool(name: "hello-world", description: "This is an awesome tool!") { helloWorld }\`;
-        `;
-      }
+    mockReadFile({
+      "package.json": JSON.stringify({}),
+      "my-component.tsx": `
+        const MY_QUERY = gql\`query HelloWorldQuery @tool(name: "hello-world", description: "This is an awesome tool!") { helloWorld }\`;
+      `,
     });
     vi.spyOn(glob, "glob").mockImplementation(() =>
       Promise.resolve(["my-component.tsx"])
@@ -474,21 +487,18 @@ describe("buildStart", () => {
     });
     await plugin.buildStart();
 
-    let [file, content] = (fs.writeFileSync as unknown as Mock).mock.calls[0];
+    let [, content] = (fs.writeFileSync as unknown as Mock).mock.calls[0];
     let contentObj = JSON.parse(content);
 
     expect(contentObj.resource).toBe("https://localhost:5678");
   });
 
   test("Should use custom host when specified in server config", async () => {
-    vi.spyOn(fs, "readFileSync").mockImplementation((path) => {
-      if (path === "package.json") {
-        return JSON.stringify({});
-      } else if (path === "my-component.tsx") {
-        return `
-            const MY_QUERY = gql\`query HelloWorldQuery @tool(name: "hello-world", description: "This is an awesome tool!") { helloWorld }\`;
-        `;
-      }
+    mockReadFile({
+      "package.json": JSON.stringify({}),
+      "my-component.tsx": `
+        const MY_QUERY = gql\`query HelloWorldQuery @tool(name: "hello-world", description: "This is an awesome tool!") { helloWorld }\`;
+      `,
     });
     vi.spyOn(glob, "glob").mockImplementation(() =>
       Promise.resolve(["my-component.tsx"])
@@ -504,21 +514,18 @@ describe("buildStart", () => {
     });
     await plugin.buildStart();
 
-    let [file, content] = (fs.writeFileSync as unknown as Mock).mock.calls[0];
+    let [, content] = (fs.writeFileSync as unknown as Mock).mock.calls[0];
     let contentObj = JSON.parse(content);
 
     expect(contentObj.resource).toBe("http://awesome.com:5678");
   });
 
   test("Should error when tool name is not provided", async () => {
-    vi.spyOn(fs, "readFileSync").mockImplementation((path) => {
-      if (path === "package.json") {
-        return JSON.stringify({});
-      } else if (path === "my-component.tsx") {
-        return `
-            const MY_QUERY = gql\`query HelloWorldQuery @tool { helloWorld }\`;
-        `;
-      }
+    mockReadFile({
+      "package.json": JSON.stringify({}),
+      "my-component.tsx": `
+        const MY_QUERY = gql\`query HelloWorldQuery @tool { helloWorld }\`;
+      `,
     });
     vi.spyOn(glob, "glob").mockImplementation(() =>
       Promise.resolve(["my-component.tsx"])
@@ -537,14 +544,11 @@ describe("buildStart", () => {
   });
 
   test("Should error when tool description is not provided", async () => {
-    vi.spyOn(fs, "readFileSync").mockImplementation((path) => {
-      if (path === "package.json") {
-        return JSON.stringify({});
-      } else if (path === "my-component.tsx") {
-        return `
-            const MY_QUERY = gql\`query HelloWorldQuery @tool(name: "hello-world") { helloWorld }\`;
-        `;
-      }
+    mockReadFile({
+      "package.json": JSON.stringify({}),
+      "my-component.tsx": `
+        const MY_QUERY = gql\`query HelloWorldQuery @tool(name: "hello-world") { helloWorld }\`;
+      `,
     });
     vi.spyOn(glob, "glob").mockImplementation(() =>
       Promise.resolve(["my-component.tsx"])
@@ -563,14 +567,11 @@ describe("buildStart", () => {
   });
 
   test("Should error when tool name contains spaces", async () => {
-    vi.spyOn(fs, "readFileSync").mockImplementation((path) => {
-      if (path === "package.json") {
-        return JSON.stringify({});
-      } else if (path === "my-component.tsx") {
-        return `
-            const MY_QUERY = gql\`query HelloWorldQuery @tool(name: "hello world", description: "A tool") { helloWorld }\`;
-        `;
-      }
+    mockReadFile({
+      "package.json": JSON.stringify({}),
+      "my-component.tsx": `
+        const MY_QUERY = gql\`query HelloWorldQuery @tool(name: "hello world", description: "A tool") { helloWorld }\`;
+      `,
     });
     vi.spyOn(glob, "glob").mockImplementation(() =>
       Promise.resolve(["my-component.tsx"])
@@ -589,14 +590,11 @@ describe("buildStart", () => {
   });
 
   test("Should error when tool name is not a string", async () => {
-    vi.spyOn(fs, "readFileSync").mockImplementation((path) => {
-      if (path === "package.json") {
-        return JSON.stringify({});
-      } else if (path === "my-component.tsx") {
-        return `
-            const MY_QUERY = gql\`query HelloWorldQuery @tool(name: true) { helloWorld }\`;
-        `;
-      }
+    mockReadFile({
+      "package.json": JSON.stringify({}),
+      "my-component.tsx": `
+        const MY_QUERY = gql\`query HelloWorldQuery @tool(name: true) { helloWorld }\`;
+      `,
     });
     vi.spyOn(glob, "glob").mockImplementation(() =>
       Promise.resolve(["my-component.tsx"])
@@ -615,14 +613,11 @@ describe("buildStart", () => {
   });
 
   test("Should error when tool description is not a string", async () => {
-    vi.spyOn(fs, "readFileSync").mockImplementation((path) => {
-      if (path === "package.json") {
-        return JSON.stringify({});
-      } else if (path === "my-component.tsx") {
-        return `
-            const MY_QUERY = gql\`query HelloWorldQuery @tool(name: "hello-world", description: false) { helloWorld }\`;
-        `;
-      }
+    mockReadFile({
+      "package.json": JSON.stringify({}),
+      "my-component.tsx": `
+        const MY_QUERY = gql\`query HelloWorldQuery @tool(name: "hello-world", description: false) { helloWorld }\`;
+      `,
     });
     vi.spyOn(glob, "glob").mockImplementation(() =>
       Promise.resolve(["my-component.tsx"])
@@ -641,14 +636,11 @@ describe("buildStart", () => {
   });
 
   test("Should error when extraInputs is not an array", async () => {
-    vi.spyOn(fs, "readFileSync").mockImplementation((path) => {
-      if (path === "package.json") {
-        return JSON.stringify({});
-      } else if (path === "my-component.tsx") {
-        return `
-            const MY_QUERY = gql\`query HelloWorldQuery @tool(name: "hello-world", description: "hello", extraInputs: false ) { helloWorld }\`;
-        `;
-      }
+    mockReadFile({
+      "package.json": JSON.stringify({}),
+      "my-component.tsx": `
+        const MY_QUERY = gql\`query HelloWorldQuery @tool(name: "hello-world", description: "hello", extraInputs: false ) { helloWorld }\`;
+      `,
     });
     vi.spyOn(glob, "glob").mockImplementation(() =>
       Promise.resolve(["my-component.tsx"])
@@ -666,17 +658,241 @@ describe("buildStart", () => {
     );
   });
 
+  test("Should error when widgetSettings.prefersBorder is not a boolean", async () => {
+    mockReadFile({
+      "package.json": JSON.stringify({
+        widgetSettings: {
+          prefersBorder: "test",
+        },
+      }),
+      "my-component.tsx": `
+        const MY_QUERY = gql\`query HelloWorldQuery @tool(name: "test", description: "Test") { helloWorld }\`;
+      `,
+    });
+    vi.spyOn(glob, "glob").mockImplementation(() =>
+      Promise.resolve(["my-component.tsx"])
+    );
+    vi.spyOn(path, "resolve").mockImplementation((_, file) => file);
+    vi.spyOn(fs, "writeFileSync");
+
+    const plugin = ApplicationManifestPlugin();
+    plugin.configResolved({ command: "serve", server: {} });
+
+    await expect(
+      async () => await plugin.buildStart()
+    ).rejects.toThrowErrorMatchingInlineSnapshot(
+      `[Error: Expected 'widgetSettings.prefersBorder' to be of type 'boolean' but found 'string' instead.]`
+    );
+  });
+
+  test("Should error when widgetSettings.description is not a string", async () => {
+    mockReadFile({
+      "package.json": JSON.stringify({
+        widgetSettings: {
+          description: true,
+        },
+      }),
+      "my-component.tsx": `
+        const MY_QUERY = gql\`query HelloWorldQuery @tool(name: "test", description: "Test") { helloWorld }\`;
+      `,
+    });
+    vi.spyOn(glob, "glob").mockImplementation(() =>
+      Promise.resolve(["my-component.tsx"])
+    );
+    vi.spyOn(path, "resolve").mockImplementation((_, file) => file);
+    vi.spyOn(fs, "writeFileSync");
+
+    const plugin = ApplicationManifestPlugin();
+    plugin.configResolved({ command: "serve", server: {} });
+
+    await expect(
+      async () => await plugin.buildStart()
+    ).rejects.toThrowErrorMatchingInlineSnapshot(
+      `[Error: Expected 'widgetSettings.description' to be of type 'string' but found 'boolean' instead.]`
+    );
+  });
+
+  test("Should error when widgetSettings.domain is not a string", async () => {
+    mockReadFile({
+      "package.json": JSON.stringify({
+        widgetSettings: {
+          domain: true,
+        },
+      }),
+      "my-component.tsx": `
+        const MY_QUERY = gql\`query HelloWorldQuery @tool(name: "test", description: "Test") { helloWorld }\`;
+      `,
+    });
+    vi.spyOn(glob, "glob").mockImplementation(() =>
+      Promise.resolve(["my-component.tsx"])
+    );
+    vi.spyOn(path, "resolve").mockImplementation((_, file) => file);
+    vi.spyOn(fs, "writeFileSync");
+
+    const plugin = ApplicationManifestPlugin();
+    plugin.configResolved({ command: "serve", server: {} });
+
+    await expect(
+      async () => await plugin.buildStart()
+    ).rejects.toThrowErrorMatchingInlineSnapshot(
+      `[Error: Expected 'widgetSettings.domain' to be of type 'string' but found 'boolean' instead.]`
+    );
+  });
+
+  test("Should allow empty widgetSettings value", async () => {
+    mockReadFile({
+      "package.json": JSON.stringify({
+        widgetSettings: {},
+      }),
+      "my-component.tsx": `
+        const MY_QUERY = gql\`query HelloWorldQuery @tool(name: "test", description: "Test") { helloWorld }\`;
+      `,
+    });
+    vi.spyOn(glob, "glob").mockImplementation(() =>
+      Promise.resolve(["my-component.tsx"])
+    );
+    vi.spyOn(path, "resolve").mockImplementation((_, file) => file);
+    vi.spyOn(fs, "writeFileSync");
+
+    const plugin = ApplicationManifestPlugin();
+    plugin.configResolved({ command: "serve", server: {}, build: {} });
+
+    await expect(plugin.buildStart()).resolves.toBeUndefined();
+  });
+
+  test("Should error when labels.toolInvocation.invoking in package.json is not a string", async () => {
+    mockReadFile({
+      "package.json": JSON.stringify({
+        labels: {
+          toolInvocation: {
+            invoking: true,
+          },
+        },
+      }),
+      "my-component.tsx": `
+        const MY_QUERY = gql\`query HelloWorldQuery @tool(name: "test", description: "Test") { helloWorld }\`;
+      `,
+    });
+    vi.spyOn(glob, "glob").mockImplementation(() =>
+      Promise.resolve(["my-component.tsx"])
+    );
+    vi.spyOn(path, "resolve").mockImplementation((_, file) => file);
+    vi.spyOn(fs, "writeFileSync");
+
+    const plugin = ApplicationManifestPlugin();
+    plugin.configResolved({ command: "serve", server: {} });
+
+    await expect(
+      async () => await plugin.buildStart()
+    ).rejects.toThrowErrorMatchingInlineSnapshot(
+      `[Error: Expected 'labels.toolInvocation.invoking' to be of type 'string' but found 'boolean' instead.]`
+    );
+  });
+
+  test("Should error when labels.toolInvocation.invoking in @tool is not a string", async () => {
+    mockReadFile({
+      "package.json": JSON.stringify({}),
+      "my-component.tsx": `
+        const MY_QUERY = gql\`query HelloWorldQuery @tool(name: "test", description: "Test", labels: { toolInvocation: { invoking: true } }) { helloWorld }\`;
+      `,
+    });
+    vi.spyOn(glob, "glob").mockImplementation(() =>
+      Promise.resolve(["my-component.tsx"])
+    );
+    vi.spyOn(path, "resolve").mockImplementation((_, file) => file);
+    vi.spyOn(fs, "writeFileSync");
+
+    const plugin = ApplicationManifestPlugin();
+    plugin.configResolved({ command: "serve", server: {} });
+
+    await expect(
+      async () => await plugin.buildStart()
+    ).rejects.toThrowErrorMatchingInlineSnapshot(
+      `[Error: Expected 'labels.toolInvocation.invoking' to be of type 'string' but found 'boolean' instead.]`
+    );
+  });
+
+  test("Should error when labels.toolInvocation.invoked in package.json is not a string", async () => {
+    mockReadFile({
+      "package.json": JSON.stringify({
+        labels: {
+          toolInvocation: {
+            invoked: true,
+          },
+        },
+      }),
+      "my-component.tsx": `
+        const MY_QUERY = gql\`query HelloWorldQuery @tool(name: "test", description: "Test") { helloWorld }\`;
+      `,
+    });
+    vi.spyOn(glob, "glob").mockImplementation(() =>
+      Promise.resolve(["my-component.tsx"])
+    );
+    vi.spyOn(path, "resolve").mockImplementation((_, file) => file);
+    vi.spyOn(fs, "writeFileSync");
+
+    const plugin = ApplicationManifestPlugin();
+    plugin.configResolved({ command: "serve", server: {} });
+
+    await expect(
+      async () => await plugin.buildStart()
+    ).rejects.toThrowErrorMatchingInlineSnapshot(
+      `[Error: Expected 'labels.toolInvocation.invoked' to be of type 'string' but found 'boolean' instead.]`
+    );
+  });
+
+  test("Should error when labels.toolInvocation.invoked in @tool is not a string", async () => {
+    mockReadFile({
+      "package.json": JSON.stringify({}),
+      "my-component.tsx": `
+        const MY_QUERY = gql\`query HelloWorldQuery @tool(name: "test", description: "Test", labels: { toolInvocation: { invoked: true } }) { helloWorld }\`;
+      `,
+    });
+    vi.spyOn(glob, "glob").mockImplementation(() =>
+      Promise.resolve(["my-component.tsx"])
+    );
+    vi.spyOn(path, "resolve").mockImplementation((_, file) => file);
+    vi.spyOn(fs, "writeFileSync");
+
+    const plugin = ApplicationManifestPlugin();
+    plugin.configResolved({ command: "serve", server: {} });
+
+    await expect(
+      async () => await plugin.buildStart()
+    ).rejects.toThrowErrorMatchingInlineSnapshot(
+      `[Error: Expected 'labels.toolInvocation.invoked' to be of type 'string' but found 'boolean' instead.]`
+    );
+  });
+
+  test("Should allow empty labels value", async () => {
+    mockReadFile({
+      "package.json": JSON.stringify({
+        labels: {},
+      }),
+      "my-component.tsx": `
+        const MY_QUERY = gql\`query HelloWorldQuery @tool(name: "test", description: "Test", labels: {}) { helloWorld }\`;
+      `,
+    });
+    vi.spyOn(glob, "glob").mockImplementation(() =>
+      Promise.resolve(["my-component.tsx"])
+    );
+    vi.spyOn(path, "resolve").mockImplementation((_, file) => file);
+    vi.spyOn(fs, "writeFileSync");
+
+    const plugin = ApplicationManifestPlugin();
+    plugin.configResolved({ command: "serve", server: {}, build: {} });
+
+    await expect(plugin.buildStart()).resolves.toBeUndefined();
+  });
+
   test("Should error when an unknown type is discovered", async () => {
-    vi.spyOn(fs, "readFileSync").mockImplementation((path) => {
-      if (path === "package.json") {
-        return JSON.stringify({});
-      } else if (path === "my-component.tsx") {
-        return `
-            const MY_QUERY = gql\`query HelloWorldQuery @tool(name: "hello-world", description: "hello", extraInputs: [{
-              name: 3.1
-            }] ) { helloWorld }\`;
-        `;
-      }
+    mockReadFile({
+      "package.json": JSON.stringify({}),
+      "my-component.tsx": `
+        const MY_QUERY = gql\`query HelloWorldQuery @tool(name: "hello-world", description: "hello", extraInputs: [{
+          name: 3.1
+        }] ) { helloWorld }\`;
+      `,
     });
     vi.spyOn(glob, "glob").mockImplementation(() =>
       Promise.resolve(["my-component.tsx"])
@@ -695,25 +911,22 @@ describe("buildStart", () => {
   });
 
   test("Should order operations and fragments when generating normalized operation", async () => {
-    vi.spyOn(fs, "readFileSync").mockImplementation((path) => {
-      if (path === "package.json") {
-        return JSON.stringify({});
-      } else if (path === root + "/my-component.tsx") {
-        return `
-            const MY_QUERY = gql\`
-              fragment A on User { firstName }
-              fragment B on User { lastName }
-              query HelloWorldQuery @tool(name: "hello-world", description: "This is an awesome tool!") {
-                helloWorld {
-                  ...B
-                  ...A
-                  ...C
-                }
-
-              fragment C on User { middleName }
-              }\`;
-        `;
-      }
+    mockReadFile({
+      "package.json": JSON.stringify({}),
+      [`${root}/my-component.tsx`]: `
+        const MY_QUERY = gql\`
+          fragment A on User { firstName }
+          fragment B on User { lastName }
+          query HelloWorldQuery @tool(name: "hello-world", description: "This is an awesome tool!") {
+            helloWorld {
+              ...B
+              ...A
+              ...C
+            }
+
+            fragment C on User { middleName }
+          }\`;
+      `,
     });
     vi.spyOn(glob, "glob").mockImplementation(() =>
       Promise.resolve(["my-component.tsx"])
@@ -739,6 +952,8 @@ describe("buildStart", () => {
       {
         "csp": {
           "connectDomains": [],
+          "frameDomains": [],
+          "redirectDomains": [],
           "resourceDomains": [],
         },
         "format": "apollo-ai-app-manifest",
@@ -792,18 +1007,15 @@ describe("buildStart", () => {
 
 describe("writeBundle", () => {
   test("Should use custom entry point when in build mode and provided in package.json", async () => {
-    vi.spyOn(fs, "readFileSync").mockImplementation((path) => {
-      if (path === "package.json") {
-        return JSON.stringify({
-          entry: {
-            staging: "http://staging.awesome.com",
-          },
-        });
-      } else if (path === "my-component.tsx") {
-        return `
-            const MY_QUERY = gql\`query HelloWorldQuery @tool(name: "hello-world", description: "This is an awesome tool!") { helloWorld }\`;
-        `;
-      }
+    mockReadFile({
+      "package.json": JSON.stringify({
+        entry: {
+          staging: "http://staging.awesome.com",
+        },
+      }),
+      "my-component.tsx": `
+        const MY_QUERY = gql\`query HelloWorldQuery @tool(name: "hello-world", description: "This is an awesome tool!") { helloWorld }\`;
+      `,
     });
     vi.spyOn(glob, "glob").mockImplementation(() =>
       Promise.resolve(["my-component.tsx"])
@@ -822,21 +1034,18 @@ describe("writeBundle", () => {
     await plugin.buildStart();
     await plugin.writeBundle();
 
-    let [file, content] = (fs.writeFileSync as unknown as Mock).mock.calls[0];
+    let [, content] = (fs.writeFileSync as unknown as Mock).mock.calls[0];
     let contentObj = JSON.parse(content);
 
     expect(contentObj.resource).toBe("http://staging.awesome.com");
   });
 
   test("Should use index.html when in build production and not provided in package.json", async () => {
-    vi.spyOn(fs, "readFileSync").mockImplementation((path) => {
-      if (path === "package.json") {
-        return JSON.stringify({});
-      } else if (path === "my-component.tsx") {
-        return `
-            const MY_QUERY = gql\`query HelloWorldQuery @tool(name: "hello-world", description: "This is an awesome tool!") { helloWorld }\`;
-        `;
-      }
+    mockReadFile({
+      "package.json": JSON.stringify({}),
+      "my-component.tsx": `
+        const MY_QUERY = gql\`query HelloWorldQuery @tool(name: "hello-world", description: "This is an awesome tool!") { helloWorld }\`;
+      `,
     });
     vi.spyOn(glob, "glob").mockImplementation(() =>
       Promise.resolve(["my-component.tsx"])
@@ -855,21 +1064,18 @@ describe("writeBundle", () => {
     await plugin.buildStart();
     await plugin.writeBundle();
 
-    let [file, content] = (fs.writeFileSync as unknown as Mock).mock.calls[0];
+    let [, content] = (fs.writeFileSync as unknown as Mock).mock.calls[0];
     let contentObj = JSON.parse(content);
 
     expect(contentObj.resource).toBe("index.html");
   });
 
   test("Should throw an error when in build mode and using a mode that is not production and not provided in package.json", async () => {
-    vi.spyOn(fs, "readFileSync").mockImplementation((path) => {
-      if (path === "package.json") {
-        return JSON.stringify({});
-      } else if (path === "my-component.tsx") {
-        return `
-            const MY_QUERY = gql\`query HelloWorldQuery @tool(name: "hello-world", description: "This is an awesome tool!") { helloWorld }\`;
-        `;
-      }
+    mockReadFile({
+      "package.json": JSON.stringify({}),
+      "my-component.tsx": `
+        const MY_QUERY = gql\`query HelloWorldQuery @tool(name: "hello-world", description: "This is an awesome tool!") { helloWorld }\`;
+      `,
     });
     vi.spyOn(glob, "glob").mockImplementation(() =>
       Promise.resolve(["my-component.tsx"])
@@ -895,14 +1101,11 @@ describe("writeBundle", () => {
   });
 
   test("Should always write to both locations when running in build mode", async () => {
-    vi.spyOn(fs, "readFileSync").mockImplementation((path) => {
-      if (path === "package.json") {
-        return JSON.stringify({});
-      } else if (path === "my-component.tsx") {
-        return `
-            const MY_QUERY = gql\`query HelloWorldQuery @tool(name: "hello-world", description: "This is an awesome tool!") { helloWorld }\`;
-        `;
-      }
+    mockReadFile({
+      "package.json": JSON.stringify({}),
+      "my-component.tsx": `
+        const MY_QUERY = gql\`query HelloWorldQuery @tool(name: "hello-world", description: "This is an awesome tool!") { helloWorld }\`;
+      `,
     });
     vi.spyOn(glob, "glob").mockImplementation(() =>
       Promise.resolve(["my-component.tsx"])
@@ -927,18 +1130,15 @@ describe("writeBundle", () => {
 
 describe("configureServer", () => {
   test("Should write to manifest file when package.json or file is updated", async () => {
-    vi.spyOn(fs, "readFileSync").mockImplementation((path) => {
-      if (path === "package.json") {
-        return JSON.stringify({});
-      } else if (path === "my-component.tsx") {
-        return `
-            const MY_QUERY = gql\`query HelloWorldQuery($name: string!) @tool(name: "hello-world", description: "This is an awesome tool!", extraInputs: [{
-              name: "doStuff",
-              type: "boolean",
-              description: "Should we do stuff?"
-            }]) { helloWorld(name: $name) }\`;
-        `;
-      }
+    mockReadFile({
+      "package.json": JSON.stringify({}),
+      "my-component.tsx": `
+        const MY_QUERY = gql\`query HelloWorldQuery($name: string!) @tool(name: "hello-world", description: "This is an awesome tool!", extraInputs: [{
+          name: "doStuff",
+          type: "boolean",
+          description: "Should we do stuff?"
+        }]) { helloWorld(name: $name) }\`;
+      `,
     });
     vi.spyOn(glob, "glob").mockImplementation(() =>
       Promise.resolve(["my-component.tsx"])
@@ -946,16 +1146,18 @@ describe("configureServer", () => {
     vi.spyOn(path, "resolve").mockImplementation((_, file) => file);
     vi.spyOn(fs, "writeFileSync");
 
+    let _callbacks: Function[] = [];
+
     const server = {
       watcher: {
         init: () => {
-          this._callbacks = [];
+          _callbacks = [];
         },
         on: (_event: string, callback: Function) => {
-          this._callbacks.push(callback);
+          _callbacks.push(callback);
         },
-        trigger: async (file) => {
-          for (const callback of this._callbacks) {
+        trigger: async (file: string) => {
+          for (const callback of _callbacks) {
             await callback(file);
           }
         },
@@ -970,10 +1172,28 @@ describe("configureServer", () => {
       build: { outDir: "/dist" },
     });
     await plugin.buildStart();
-    await plugin.configureServer(server);
+    plugin.configureServer(server);
     await server.watcher.trigger("package.json");
     await server.watcher.trigger("my-component.tsx");
 
     expect(fs.writeFileSync).toBeCalledTimes(6);
   });
 });
+
+type FilePath = string;
+
+function mockReadFile(mocks: Record<FilePath, string | (() => string)>) {
+  vi.spyOn(fs, "readFileSync").mockImplementation((path) => {
+    const mock = mocks[path.toString()];
+
+    if (!mock) {
+      throw new Error(`No matched mock for path '${path}'`);
+    }
+
+    if (typeof mock === "function") {
+      return mock();
+    }
+
+    return mock;
+  });
+}