42 lines
1.5 KiB
TypeScript
42 lines
1.5 KiB
TypeScript
import type { Client } from "@modelcontextprotocol/client";
|
|
import { expect, type OnTestFinishedHandler } from "vitest";
|
|
|
|
import { createArtifactBinarySession } from "../../../src/composition/binary.js";
|
|
import { createServer } from "../../../src/server/createServer.js";
|
|
import type { TestMcp } from "./mcpFixture.js";
|
|
|
|
const openSessionTarget = async (
|
|
mcp: TestMcp,
|
|
onTestFinished: (handler: OnTestFinishedHandler) => void,
|
|
operation: "open_binary" | "open_bundle_resources",
|
|
path: string,
|
|
): Promise<Client> => {
|
|
const session = await createArtifactBinarySession(process.env);
|
|
onTestFinished(async () => {
|
|
await session.close();
|
|
});
|
|
const client = await mcp.connect(createServer({ kind: "session", session }));
|
|
const opened = await client.callTool({
|
|
name: operation,
|
|
arguments: { path },
|
|
});
|
|
expect(opened.isError, JSON.stringify(opened.content)).not.toBe(true);
|
|
return client;
|
|
};
|
|
|
|
/** Open a static artifact target in the production artifact-only MCP composition. */
|
|
export const openArtifactTarget = (
|
|
mcp: TestMcp,
|
|
onTestFinished: (handler: OnTestFinishedHandler) => void,
|
|
path: string,
|
|
): Promise<Client> =>
|
|
openSessionTarget(mcp, onTestFinished, "open_binary", path);
|
|
|
|
/** Open a caller-selected bundle resource root in the production MCP session. */
|
|
export const openBundleResourcesTarget = (
|
|
mcp: TestMcp,
|
|
onTestFinished: (handler: OnTestFinishedHandler) => void,
|
|
path: string,
|
|
): Promise<Client> =>
|
|
openSessionTarget(mcp, onTestFinished, "open_bundle_resources", path);
|