Function withClient

packages/hosts/mcp/src/tool-server-protocol.test.ts:61–114

(
  config: TestServerConfig,
  run: (client: Client) => Promise<void>,
  options?: { readonly manualInputRequired?: boolean },
)

});

const withClient = async (
  config: TestServerConfig,
  run: (client: Client) => Promise<void>,
  options?: { readonly manualInputRequired?: boolean },
) => {
  const requestBodies = new WeakMap<Request, unknown>();
  const handler = createMcpHandler(
    (context) =>
      Effect.runPromise(
        Effect.gen(function* () {
          const requestStatePrincipal = config.requestStatePrincipal?.() ?? "principal-test";
          const requestStateResource = config.requestStateResource?.() ?? defaultMcpResource;
          const requestStateBinding = yield* Effect.promise(() =>
            mcpRequestStateBindingFromBody({
              body: context.requestInfo ? requestBodies.get(context.requestInfo) : undefined,
              principal: requestStatePrincipal,
              resource: requestStateResource,
            }),
          );
          return yield* buildMcpServer({
            ...config,
            requestStateSigningKey: REQUEST_STATE_KEY,
            requestStatePrincipal,
            ...(requestStateBinding === null ? {} : { requestStateBinding }),
          });
        }),
      ),
    { legacy: "reject" },
  );
  const transport = new StreamableHTTPClientTransport(new URL("http://executor.test/mcp"), {
    fetch: async (input, init) => {
      const request =
        input instanceof Request ? new Request(input, init) : new Request(input.toString(), init);
      requestBodies.set(request, await request.clone().json());
      return handler.fetch(request, { parsedBody: requestBodies.get(request) });
    },
  });
  const client = new Client(
    { name: "executor-protocol-test", version: "1.0.0" },
    {
      capabilities: { elicitation: { form: {} } },
      versionNegotiation: { mode: { pin: "2026-07-28" } },
      ...(options?.manualInputRequired ? { inputRequired: { autoFulfill: false } } : {}),
    },
  );
  await client.connect(transport);
  // oxlint-disable-next-line executor/no-try-catch-or-throw -- boundary: test helper owns the client transport and in-process HTTP handler
  try {
    await run(client);
  } finally {
    await client.close();
    await handler.close();
  }
};

const manualToolCall = (
  client: Client,