mirror of
https://github.com/mauriceboe/TREK.git
synced 2026-10-19 13:36:48 +00:00
Maurice
724c88c201
A mechanical prettier --write over server/src, server/tests, server/scripts and shared/src with the workspaces' .prettierrc; nothing but layout and import order changes, and every file keeps the same set of imports. shared is formatted throughout and its format baseline is empty. The server baseline keeps the 35 files that Prettier would grow past their lint:size limit (they need a split first, which is not a formatting change), and the generated entity files stay as their generator writes them.
87 lines
2.8 KiB
TypeScript
87 lines
2.8 KiB
TypeScript
/**
|
|
* The process-level plugin MCP tool source — MCPSINK-001 to MCPSINK-005.
|
|
*
|
|
* The unset case is the one that matters and the one none of the sibling sinks
|
|
* covers: every app built without the plugins runtime, and every test that
|
|
* never registers a source, goes through it on each session.
|
|
*/
|
|
import { registerTools } from '../../../src/mcp/tools';
|
|
import type { McpContext, McpDynamicTool } from '../../../src/nest-mcp';
|
|
import { getPluginMcpToolSource, setPluginMcpToolSource } from '../../../src/plugin-mcp-tools';
|
|
import { McpServer } from '@modelcontextprotocol/sdk/server/mcp';
|
|
|
|
import { afterEach, describe, expect, it, vi } from 'vitest';
|
|
|
|
const echo: McpDynamicTool = {
|
|
options: {
|
|
name: 'plugin_demo_echo',
|
|
description: 'Echoes.',
|
|
access: { group: 'plugins', mode: 'use' } as never,
|
|
},
|
|
handler: () => ({ content: [{ type: 'text', text: 'dynamic' }] }),
|
|
};
|
|
|
|
/** Records what attach() was handed, without needing a real registry. */
|
|
function spyRegistry() {
|
|
const attach = vi.fn();
|
|
return { attach, registry: { attach } as unknown as Parameters<typeof registerTools>[0] };
|
|
}
|
|
|
|
afterEach(() => {
|
|
setPluginMcpToolSource(null);
|
|
});
|
|
|
|
describe('setPluginMcpToolSource', () => {
|
|
it('MCPSINK-001: is unset by default', () => {
|
|
expect(getPluginMcpToolSource()).toBeNull();
|
|
});
|
|
|
|
it('MCPSINK-002: round-trips a source and clears back to null', () => {
|
|
const source = () => [echo];
|
|
setPluginMcpToolSource(source);
|
|
expect(getPluginMcpToolSource()).toBe(source);
|
|
|
|
setPluginMcpToolSource(null);
|
|
expect(getPluginMcpToolSource()).toBeNull();
|
|
});
|
|
});
|
|
|
|
describe('registerTools wiring', () => {
|
|
it('MCPSINK-003: passes no dynamic source when none is registered', async () => {
|
|
const { attach, registry } = spyRegistry();
|
|
|
|
await registerTools(registry, new McpServer({ name: 't', version: '1' }), 1, null);
|
|
|
|
expect(attach.mock.calls[0][2]).toEqual({ onInvoke: undefined, dynamicTools: undefined });
|
|
});
|
|
|
|
it('MCPSINK-004: passes the registered source through to attach', async () => {
|
|
const source = (_ctx: McpContext) => [echo];
|
|
setPluginMcpToolSource(source);
|
|
const { attach, registry } = spyRegistry();
|
|
|
|
await registerTools(registry, new McpServer({ name: 't', version: '1' }), 1, null);
|
|
|
|
expect(attach.mock.calls[0][2].dynamicTools).toBe(source);
|
|
});
|
|
|
|
it('MCPSINK-005: an explicit argument wins over the process-level source', async () => {
|
|
setPluginMcpToolSource(() => [echo]);
|
|
const explicit = () => [];
|
|
const { attach, registry } = spyRegistry();
|
|
|
|
await registerTools(
|
|
registry,
|
|
new McpServer({ name: 't', version: '1' }),
|
|
1,
|
|
null,
|
|
false,
|
|
undefined,
|
|
undefined,
|
|
explicit,
|
|
);
|
|
|
|
expect(attach.mock.calls[0][2].dynamicTools).toBe(explicit);
|
|
});
|
|
});
|