@assistant-ui/mcp-docs-server 0.1.39 → 0.2.1

package/src/tools/tests/resources.test.ts

@@ -1,102 +1,133 @@
-import { describe, it, expect } from "vitest";
+import { afterEach, beforeEach, describe, expect, it } from "vitest";
 import { server } from "../../index.js";
+import { createMcpTestClient, type McpTestClient } from "./mcp-test-client.js";
 
-function handler(method: string) {
-  const handlers = (server as any).server._requestHandlers;
-  const h = handlers.get(method);
-  if (!h) throw new Error(`No handler for ${method}`);
-  return h;
+type Resource = {
+  uri: string;
+};
+
+type ResourceListResult = {
+  resources: Resource[];
+};
+
+type ResourceReadResult = {
+  contents: Array<{
+    mimeType?: string;
+    text?: string;
+  }>;
+};
+
+function requireResource(resources: Resource[], prefix: string): Resource {
+  const resource = resources.find((candidate) =>
+    candidate.uri.startsWith(prefix),
+  );
+  expect(resource).toBeDefined();
+  if (!resource) throw new Error(`Missing resource with URI prefix: ${prefix}`);
+  return resource;
 }
 
 describe("MCP resources", () => {
+  let client: McpTestClient;
+
+  beforeEach(async () => {
+    client = await createMcpTestClient(server);
+    await client.initialize();
+  });
+
+  afterEach(async () => {
+    await client?.close();
+  });
+
   it("lists docs and example resources", async () => {
-    const result = await handler("resources/list")(
-      { method: "resources/list", params: {} },
-      {},
-    );
+    const result = await client.request<ResourceListResult>({
+      method: "resources/list",
+      params: {},
+    });
+
     expect(result.resources.length).toBeGreaterThan(0);
     expect(
-      result.resources.some((r: any) => r.uri.startsWith("aui-docs:///")),
+      result.resources.some((resource) =>
+        resource.uri.startsWith("aui-docs:///"),
+      ),
     ).toBe(true);
     expect(
-      result.resources.some((r: any) => r.uri.startsWith("aui-example:///")),
+      result.resources.some((resource) =>
+        resource.uri.startsWith("aui-example:///"),
+      ),
     ).toBe(true);
   });
 
   it("reads a documentation resource as markdown", async () => {
-    const list = await handler("resources/list")(
-      { method: "resources/list", params: {} },
-      {},
-    );
-    const doc = list.resources.find((r: any) =>
-      r.uri.startsWith("aui-docs:///"),
-    );
-    const read = await handler("resources/read")(
-      { method: "resources/read", params: { uri: doc.uri } },
-      {},
-    );
-    expect(read.contents[0].mimeType).toBe("text/markdown");
-    expect(typeof read.contents[0].text).toBe("string");
-    expect(read.contents[0].text.length).toBeGreaterThan(0);
+    const list = await client.request<ResourceListResult>({
+      method: "resources/list",
+      params: {},
+    });
+    const doc = requireResource(list.resources, "aui-docs:///");
+    const read = await client.request<ResourceReadResult>({
+      method: "resources/read",
+      params: { uri: doc.uri },
+    });
+
+    expect(read.contents[0]?.mimeType).toBe("text/markdown");
+    expect(typeof read.contents[0]?.text).toBe("string");
+    expect(read.contents[0]?.text.length).toBeGreaterThan(0);
   });
 
   it("reads a slash-containing documentation resource", async () => {
-    const list = await handler("resources/list")(
-      { method: "resources/list", params: {} },
-      {},
-    );
+    const list = await client.request<ResourceListResult>({
+      method: "resources/list",
+      params: {},
+    });
     const prefix = "aui-docs:///";
     const nested = list.resources.find(
-      (r: any) =>
-        r.uri.startsWith(prefix) && r.uri.slice(prefix.length).includes("/"),
+      (resource) =>
+        resource.uri.startsWith(prefix) &&
+        resource.uri.slice(prefix.length).includes("/"),
     );
+
     expect(nested).toBeDefined();
-    const read = await handler("resources/read")(
-      { method: "resources/read", params: { uri: nested.uri } },
-      {},
-    );
-    expect(read.contents[0].mimeType).toBe("text/markdown");
-    expect(read.contents[0].text.length).toBeGreaterThan(0);
+    if (!nested) throw new Error("Missing nested documentation resource");
+
+    const read = await client.request<ResourceReadResult>({
+      method: "resources/read",
+      params: { uri: nested.uri },
+    });
+
+    expect(read.contents[0]?.mimeType).toBe("text/markdown");
+    expect(read.contents[0]?.text.length).toBeGreaterThan(0);
   });
 
   it("errors on an unknown documentation resource", async () => {
     await expect(
-      handler("resources/read")(
-        {
-          method: "resources/read",
-          params: { uri: "aui-docs:///definitely-not-a-real-doc" },
-        },
-        {},
-      ),
+      client.request({
+        method: "resources/read",
+        params: { uri: "aui-docs:///definitely-not-a-real-doc" },
+      }),
     ).rejects.toThrow();
   });
 
   it("reads an example resource as markdown", async () => {
-    const list = await handler("resources/list")(
-      { method: "resources/list", params: {} },
-      {},
-    );
-    const example = list.resources.find((r: any) =>
-      r.uri.startsWith("aui-example:///"),
-    );
-    const read = await handler("resources/read")(
-      { method: "resources/read", params: { uri: example.uri } },
-      {},
-    );
-    expect(read.contents[0].mimeType).toBe("text/markdown");
-    expect(typeof read.contents[0].text).toBe("string");
-    expect(read.contents[0].text.length).toBeGreaterThan(0);
+    const list = await client.request<ResourceListResult>({
+      method: "resources/list",
+      params: {},
+    });
+    const example = requireResource(list.resources, "aui-example:///");
+    const read = await client.request<ResourceReadResult>({
+      method: "resources/read",
+      params: { uri: example.uri },
+    });
+
+    expect(read.contents[0]?.mimeType).toBe("text/markdown");
+    expect(typeof read.contents[0]?.text).toBe("string");
+    expect(read.contents[0]?.text.length).toBeGreaterThan(0);
   });
 
   it("errors on an unknown example resource", async () => {
     await expect(
-      handler("resources/read")(
-        {
-          method: "resources/read",
-          params: { uri: "aui-example:///definitely-not-a-real-example" },
-        },
-        {},
-      ),
+      client.request({
+        method: "resources/read",
+        params: { uri: "aui-example:///definitely-not-a-real-example" },
+      }),
     ).rejects.toThrow();
   });
 });

package/src/tools/xulux-templates.ts

@@ -1,4 +1,4 @@
-import { z } from "zod/v3";
+import { z } from "zod";
 import { logger } from "../utils/logger.js";
 import { formatMCPResponse } from "../utils/mcp-format.js";
 import { getXuluxCatalog } from "../xulux/catalog-client.js";
@@ -56,7 +56,7 @@ export const xuluxTemplatesListTool = {
     "Call this first for any assistant-ui app-building request. Use the features and customizable fields to decide which template fits. " +
     "If customizable is empty, the entry is a fixed demo that should be used as-is rather than configured. " +
     "Call assistantUITemplateDetails on the chosen template before requesting a preview.",
-  parameters: emptyInputSchema.shape,
+  parameters: emptyInputSchema,
   execute: async (_args: z.infer<typeof emptyInputSchema>) => {
     logger.info("Listing assistant-ui templates");
     try {
@@ -82,7 +82,7 @@ export const xuluxTemplateDetailsTool = {
     "Fixed demos return no configRoots; use those as-is without config. " +
     "Use this before calling assistantUITemplatePreview to understand exactly what config to write. " +
     "If assistantUITemplatePreview returns validationWarnings, call this again and cross-reference configRoots to correct the config.",
-  parameters: detailsInputSchema.shape,
+  parameters: detailsInputSchema,
   execute: async (args: z.infer<typeof detailsInputSchema>) => {
     logger.info(
       `Getting assistant-ui template details for: ${args.templateId}`,
@@ -115,7 +115,7 @@ export const xuluxTemplatePreviewTool = {
     "If you pass config, a preview session is created on the hosted template sandbox and the returned URLs reflect that configuration. " +
     "Do not pass config for fixed demos that have no configRoots in assistantUITemplateDetails. " +
     "This tool only returns URLs.",
-  parameters: previewInputSchema.shape,
+  parameters: previewInputSchema,
   execute: async (args: z.infer<typeof previewInputSchema>) => {
     logger.info(
       `Creating assistant-ui template preview for: ${args.templateId}`,