mirror of
https://github.com/mauriceboe/TREK.git
synced 2026-10-20 05:56:53 +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.
53 lines
1.9 KiB
TypeScript
53 lines
1.9 KiB
TypeScript
import { createTestRegistry, McpController, McpRegistry, Tool, type McpAccessPolicy } from '../../../src/nest-mcp';
|
|
import type { TestCtx } from './harness';
|
|
|
|
import { describe, expect, it } from 'vitest';
|
|
|
|
@McpController()
|
|
class One {
|
|
@Tool({ name: 'one_tool' })
|
|
one() {}
|
|
}
|
|
|
|
@McpController()
|
|
class Two {
|
|
@Tool({ name: 'two_tool', access: { group: 'trips', mode: 'write' } })
|
|
two() {}
|
|
}
|
|
|
|
describe('createTestRegistry', () => {
|
|
it('builds a registry from hand-constructed instances without a Nest app', () => {
|
|
const registry = createTestRegistry([new One(), new Two()], { accessPolicy: () => true });
|
|
expect(registry).toBeInstanceOf(McpRegistry);
|
|
expect(registry.list().map((e) => e.name)).toEqual(['one_tool', 'two_tool']);
|
|
});
|
|
|
|
it('validates on construction (declarative access needs a policy)', () => {
|
|
expect(() => createTestRegistry([new One(), new Two()])).toThrow(/no accessPolicy was configured/);
|
|
});
|
|
|
|
it('surfaces a validateAccess rejection on construction', () => {
|
|
expect(() =>
|
|
createTestRegistry([new Two()], {
|
|
accessPolicy: () => true,
|
|
validateAccess: ({ group, mode }) => `unknown group '${group}' (mode '${mode}')`,
|
|
}),
|
|
).toThrow(/invalid access declarations: tool "two_tool" \(Two\.two\): unknown group 'trips' \(mode 'write'\)/);
|
|
});
|
|
|
|
it('forwards the accessPolicy option', async () => {
|
|
const calls: string[] = [];
|
|
const policy: McpAccessPolicy = (access, ctx) => {
|
|
calls.push(`${access.group}:${access.mode}:${(ctx as TestCtx).userId}`);
|
|
return false;
|
|
};
|
|
const registry = createTestRegistry([new Two()], { accessPolicy: policy });
|
|
// A denied entry is filtered before any server interaction, so a stub suffices.
|
|
await registry.attach(
|
|
{} as Parameters<McpRegistry['attach']>[0],
|
|
{ userId: 1 } as Parameters<McpRegistry['attach']>[1],
|
|
);
|
|
expect(calls).toEqual(['trips:write:1']);
|
|
});
|
|
});
|