57 lines
2.7 KiB
TypeScript
57 lines
2.7 KiB
TypeScript
import { Client } from '@modelcontextprotocol/sdk/client/index.js';
|
|
import { InMemoryTransport } from '@modelcontextprotocol/sdk/inMemory.js';
|
|
import { afterEach, describe, expect, it, vi } from 'vitest';
|
|
import type { RedditBotApiClient } from '../src/mcp/api-client.js';
|
|
import { createRedditBotMcpServer } from '../src/mcp/server.js';
|
|
|
|
describe('RedditBot MCP server', () => {
|
|
const closeables: Array<{ close(): Promise<void> }> = [];
|
|
|
|
afterEach(async () => {
|
|
await Promise.all(closeables.splice(0).map((item) => item.close()));
|
|
});
|
|
|
|
async function connectedClient() {
|
|
const get = vi.fn(async () => ({ status: 'operational' }));
|
|
const post = vi.fn(async () => ({ success: true, running: true }));
|
|
const server = createRedditBotMcpServer({ get, post } as unknown as RedditBotApiClient);
|
|
const client = new Client({ name: 'test-client', version: '1.0.0' });
|
|
const [clientTransport, serverTransport] = InMemoryTransport.createLinkedPair();
|
|
await Promise.all([server.connect(serverTransport), client.connect(clientTransport)]);
|
|
closeables.push(client, server);
|
|
return { client, get, post };
|
|
}
|
|
|
|
it('advertises read, control, and explicit external-send tools', async () => {
|
|
const { client } = await connectedClient();
|
|
const tools = await client.listTools();
|
|
const names = tools.tools.map((tool) => tool.name);
|
|
|
|
expect(names).toEqual(expect.arrayContaining([
|
|
'get_overview', 'list_accounts', 'list_conversations', 'start_bot', 'stop_bot',
|
|
'set_bot_running', 'trigger_sync', 'send_reply', 'send_message_to_user', 'start_conversation',
|
|
]));
|
|
expect(tools.tools.find((tool) => tool.name === 'get_overview')?.annotations?.readOnlyHint).toBe(true);
|
|
expect(tools.tools.find((tool) => tool.name === 'send_reply')?.annotations?.destructiveHint).toBe(true);
|
|
expect(tools.tools.find((tool) => tool.name === 'send_message_to_user')?.annotations?.destructiveHint).toBe(true);
|
|
});
|
|
|
|
it('routes start and stop tools to the versioned control API', async () => {
|
|
const { client, post } = await connectedClient();
|
|
|
|
await client.callTool({ name: 'start_bot', arguments: {} });
|
|
await client.callTool({ name: 'stop_bot', arguments: {} });
|
|
|
|
expect(post).toHaveBeenNthCalledWith(1, '/api/v1/control/start');
|
|
expect(post).toHaveBeenNthCalledWith(2, '/api/v1/control/stop');
|
|
});
|
|
|
|
it('routes automatic user messages to the versioned messaging API', async () => {
|
|
const { client, post } = await connectedClient();
|
|
|
|
await client.callTool({ name: 'send_message_to_user', arguments: { username: 'alice', message: 'hello' } });
|
|
|
|
expect(post).toHaveBeenCalledWith('/api/v1/messages', { username: 'alice', message: 'hello' });
|
|
});
|
|
});
|