server-in-memory.test.ts•1.61 kB
import { describe, expect, test } from 'vitest';
import { Client } from '@modelcontextprotocol/sdk/client/index.js';
import { InMemoryTransport } from '@modelcontextprotocol/sdk/inMemory.js';
import { createServer } from '../src/server.js';
describe('MCP Server in memory', () => {
test('listTools should return a list of tools', async () => {
const client = new Client(
{
name: 'test client',
version: '1.0',
},
{
capabilities: {
roots: {
listChanged: true,
},
},
},
);
const server = createServer();
const [clientTransport, serverTransport] =
InMemoryTransport.createLinkedPair();
await Promise.all([
client.connect(clientTransport),
server.connect(serverTransport),
]);
const result = await client.listTools();
expect(result.tools.length).toBeGreaterThan(0);
});
test('callTool test_tool should return a result', async () => {
const client = new Client({
name: 'test client',
version: '1.0',
});
const server = createServer();
const [clientTransport, serverTransport] =
InMemoryTransport.createLinkedPair();
await Promise.all([
client.connect(clientTransport),
server.connect(serverTransport),
]);
const result = await client.callTool({
name: 'test_tool',
arguments: {
hello: 'world',
},
});
expect(result).toEqual({
content: [
{
type: 'text',
text: 'Hello, world',
},
],
isError: false,
});
});
});