Trends Hub

import { Client } from "@modelcontextprotocol/sdk/client/index.js";
import { StdioClientTransport } from "@modelcontextprotocol/sdk/client/stdio.js";
import { afterAll, describe, expect, test } from "@rstest/core";

const transport = new StdioClientTransport({
  command: "node",
  args: ["."],
});

const client = new Client({
  name: "test-client",
  version: "0.0.0",
});

describe("MCP 工具测试", async () => {
  await client.connect(transport);

  const toolsResponse = await client.listTools();

  afterAll(async () => {
    await client.close();
  });

  test("应能列出可用工具", async () => {
    expect(toolsResponse).toBeDefined();
    expect(toolsResponse).toHaveProperty("tools");
    expect(toolsResponse.tools).toBeInstanceOf(Array);
    expect(toolsResponse.tools).not.toHaveLength(0);
  });

  describe.each(toolsResponse.tools.map((tool) => [tool.name]))("%s", async (toolName) => {
    test("应能调用工具", async () => {
      const result = await client.callTool({
        name: toolName as unknown as string,
      });
      expect(result).toBeDefined();
      expect(result).toHaveProperty("content");
      expect(result.content).toBeInstanceOf(Array);
      expect(result.content).not.toHaveLength(0);
    });
  });
});