Relocate tests under /test (#1220)

This commit is contained in:
Konstantin Konstantinov 2025-12-06 01:56:12 +02:00 • committed by GitHub
commit 5a9de1c638
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

71
test/helpers/mcp.ts Normal file
View file

@ -0,0 +1,71 @@
import { InMemoryTransport } from '../../src/inMemory.js';
import { Client } from '../../src/client/index.js';
import { Server } from '../../src/server/index.js';
import { InMemoryTaskStore, InMemoryTaskMessageQueue } from '../../src/experimental/tasks/stores/in-memory.js';
import type { ClientCapabilities, ServerCapabilities } from '../../src/types.js';
export interface InMemoryTaskEnvironment {
client: Client;
server: Server;
taskStore: InMemoryTaskStore;
clientTransport: InMemoryTransport;
serverTransport: InMemoryTransport;
}
export async function createInMemoryTaskEnvironment(options?: {
clientCapabilities?: ClientCapabilities;
serverCapabilities?: ServerCapabilities;
}): Promise<InMemoryTaskEnvironment> {
const taskStore = new InMemoryTaskStore();
const [clientTransport, serverTransport] = InMemoryTransport.createLinkedPair();
const client = new Client(
{
name: 'test-client',
version: '1.0.0'
},
{
capabilities: options?.clientCapabilities ?? {
tasks: {
list: {},
requests: {
tools: {
call: {}
}
}
}
}
}
);
const server = new Server(
{
name: 'test-server',
version: '1.0.0'
},
{
capabilities: options?.serverCapabilities ?? {
tasks: {
list: {},
requests: {
tools: {
call: {}
}
}
}
},
taskStore,
taskMessageQueue: new InMemoryTaskMessageQueue()
}
);
await Promise.all([client.connect(clientTransport), server.connect(serverTransport)]);
return {
client,
server,
taskStore,
clientTransport,
serverTransport
};
}