1211 lines
58 KiB
TypeScript
1211 lines
58 KiB
TypeScript
import { describe, it, expect, beforeEach, vi } from 'vitest';
|
|
import { MCPServer, type MCPServerDependencies, DEFAULT_PAGE_SIZE, encodeCursor, decodeCursor } from '../../src/main/engine/MCPServer';
|
|
|
|
|
|
|
|
function createMockPostEngine() {
|
|
return {
|
|
getAllPosts: vi.fn().mockResolvedValue({ items: [], hasMore: false, total: 0 }),
|
|
getPost: vi.fn().mockResolvedValue(null),
|
|
searchPosts: vi.fn().mockResolvedValue([]),
|
|
searchPostsFiltered: vi.fn().mockResolvedValue({ posts: [], total: 0 }),
|
|
createPost: vi.fn().mockResolvedValue({
|
|
id: 'post-1', title: 'Test', slug: 'test', content: '', status: 'draft',
|
|
tags: [], categories: [], createdAt: new Date(), updatedAt: new Date(),
|
|
}),
|
|
updatePost: vi.fn().mockResolvedValue(null),
|
|
publishPost: vi.fn().mockResolvedValue(null),
|
|
deletePost: vi.fn().mockResolvedValue(true),
|
|
getTagsWithCounts: vi.fn().mockResolvedValue([]),
|
|
getCategoriesWithCounts: vi.fn().mockResolvedValue([]),
|
|
getBlogStats: vi.fn().mockResolvedValue({
|
|
totalPosts: 0, draftCount: 0, publishedCount: 0, archivedCount: 0,
|
|
oldestPostDate: null, newestPostDate: null, postsPerYear: [], tagCount: 0, categoryCount: 0,
|
|
}),
|
|
getLinkedBy: vi.fn().mockResolvedValue([]),
|
|
getLinksTo: vi.fn().mockResolvedValue([]),
|
|
getPostsFiltered: vi.fn().mockResolvedValue([]),
|
|
getPostCounts: vi.fn().mockResolvedValue({ groups: [], totalPosts: 0 }),
|
|
};
|
|
}
|
|
|
|
function createMockMediaEngine() {
|
|
return {
|
|
getAllMedia: vi.fn().mockResolvedValue([]),
|
|
getMedia: vi.fn().mockResolvedValue(null),
|
|
updateMedia: vi.fn().mockResolvedValue(null),
|
|
getThumbnailDataUrl: vi.fn().mockResolvedValue(null),
|
|
};
|
|
}
|
|
|
|
function createMockScriptEngine() {
|
|
return {
|
|
createDraftScript: vi.fn().mockResolvedValue({
|
|
id: 'script-1', title: 'Test', slug: 'test', kind: 'macro',
|
|
entrypoint: 'main.py', content: '', enabled: true, version: 1,
|
|
filePath: '/test', createdAt: new Date(), updatedAt: new Date(),
|
|
}),
|
|
publishScript: vi.fn().mockResolvedValue({
|
|
id: 'script-1', title: 'Test', slug: 'test', kind: 'macro',
|
|
entrypoint: 'main.py', content: '', enabled: true, version: 1,
|
|
filePath: '/test', createdAt: new Date(), updatedAt: new Date(),
|
|
}),
|
|
deleteDraftScript: vi.fn().mockResolvedValue(true),
|
|
validateScript: vi.fn().mockResolvedValue({ valid: true, errors: [] }),
|
|
};
|
|
}
|
|
|
|
function createMockTemplateEngine() {
|
|
return {
|
|
createDraftTemplate: vi.fn().mockResolvedValue({
|
|
id: 'tpl-1', title: 'Test', slug: 'test', kind: 'post',
|
|
enabled: true, version: 1, filePath: '/test', content: '',
|
|
createdAt: new Date(), updatedAt: new Date(),
|
|
}),
|
|
publishTemplate: vi.fn().mockResolvedValue({
|
|
id: 'tpl-1', title: 'Test', slug: 'test', kind: 'post',
|
|
enabled: true, version: 1, filePath: '/test', content: '',
|
|
createdAt: new Date(), updatedAt: new Date(),
|
|
}),
|
|
deleteDraftTemplate: vi.fn().mockResolvedValue(true),
|
|
validateTemplate: vi.fn().mockResolvedValue({ valid: true, errors: [] }),
|
|
};
|
|
}
|
|
|
|
function createMockMetaEngine() {
|
|
return {
|
|
getProjectMetadata: vi.fn().mockResolvedValue(null),
|
|
};
|
|
}
|
|
|
|
function createMockPostMediaEngine() {
|
|
return {
|
|
getLinkedMediaDataForPost: vi.fn().mockResolvedValue([]),
|
|
getLinkedPostsForMedia: vi.fn().mockResolvedValue([]),
|
|
};
|
|
}
|
|
|
|
function createMockTagEngine() {
|
|
return {
|
|
getTagsWithCounts: vi.fn().mockResolvedValue([]),
|
|
};
|
|
}
|
|
|
|
/** Create stable mock instances that are returned by getter functions */
|
|
function createDependencies() {
|
|
const mockPostEngine = createMockPostEngine();
|
|
const mockMediaEngine = createMockMediaEngine();
|
|
const mockScriptEngine = createMockScriptEngine();
|
|
const mockTemplateEngine = createMockTemplateEngine();
|
|
const mockMetaEngine = createMockMetaEngine();
|
|
const mockPostMediaEngine = createMockPostMediaEngine();
|
|
const mockTagEngine = createMockTagEngine();
|
|
|
|
const deps: MCPServerDependencies = {
|
|
postEngine: mockPostEngine,
|
|
mediaEngine: mockMediaEngine,
|
|
scriptEngine: mockScriptEngine,
|
|
templateEngine: mockTemplateEngine,
|
|
metaEngine: mockMetaEngine,
|
|
postMediaEngine: mockPostMediaEngine,
|
|
tagEngine: mockTagEngine,
|
|
};
|
|
|
|
return { deps, mockPostEngine, mockMediaEngine, mockScriptEngine, mockTemplateEngine, mockMetaEngine, mockPostMediaEngine, mockTagEngine };
|
|
}
|
|
|
|
/** Helper: check if a key exists in an McpServer internal registry (plain object) */
|
|
function hasRegistered(mcpServer: unknown, registry: string, name: string): boolean {
|
|
const obj = (mcpServer as Record<string, Record<string, unknown>>)[registry];
|
|
return obj != null && name in obj;
|
|
}
|
|
|
|
describe('MCPServer', () => {
|
|
let server: MCPServer;
|
|
let deps: MCPServerDependencies;
|
|
let mockPostEngine: ReturnType<typeof createMockPostEngine>;
|
|
let mockMediaEngine: ReturnType<typeof createMockMediaEngine>;
|
|
let mockScriptEngine: ReturnType<typeof createMockScriptEngine>;
|
|
let mockTemplateEngine: ReturnType<typeof createMockTemplateEngine>;
|
|
let mockPostMediaEngine: ReturnType<typeof createMockPostMediaEngine>;
|
|
|
|
beforeEach(() => {
|
|
vi.clearAllMocks();
|
|
const mocks = createDependencies();
|
|
deps = mocks.deps;
|
|
mockPostEngine = mocks.mockPostEngine;
|
|
mockMediaEngine = mocks.mockMediaEngine;
|
|
mockScriptEngine = mocks.mockScriptEngine;
|
|
mockTemplateEngine = mocks.mockTemplateEngine;
|
|
mockPostMediaEngine = mocks.mockPostMediaEngine;
|
|
server = new MCPServer(deps);
|
|
});
|
|
|
|
describe('constructor', () => {
|
|
it('creates an MCPServer instance', () => {
|
|
expect(server).toBeInstanceOf(MCPServer);
|
|
});
|
|
});
|
|
|
|
describe('createMcpServer', () => {
|
|
it('creates an McpServer with registered tools', () => {
|
|
const mcpServer = server.createMcpServer();
|
|
expect(mcpServer).toBeDefined();
|
|
});
|
|
});
|
|
|
|
describe('proposal store', () => {
|
|
it('exposes the proposal store', () => {
|
|
expect(server.proposalStore).toBeDefined();
|
|
});
|
|
});
|
|
|
|
describe('registered tools', () => {
|
|
it('registers search_posts tool', () => {
|
|
const mcpServer = server.createMcpServer();
|
|
expect(hasRegistered(mcpServer, '_registeredTools', 'search_posts')).toBe(true);
|
|
});
|
|
|
|
it('registers draft_post tool', () => {
|
|
const mcpServer = server.createMcpServer();
|
|
expect(hasRegistered(mcpServer, '_registeredTools', 'draft_post')).toBe(true);
|
|
});
|
|
|
|
it('registers propose_script tool', () => {
|
|
const mcpServer = server.createMcpServer();
|
|
expect(hasRegistered(mcpServer, '_registeredTools', 'propose_script')).toBe(true);
|
|
});
|
|
|
|
it('registers propose_template tool', () => {
|
|
const mcpServer = server.createMcpServer();
|
|
expect(hasRegistered(mcpServer, '_registeredTools', 'propose_template')).toBe(true);
|
|
});
|
|
|
|
it('registers propose_media_metadata tool', () => {
|
|
const mcpServer = server.createMcpServer();
|
|
expect(hasRegistered(mcpServer, '_registeredTools', 'propose_media_metadata')).toBe(true);
|
|
});
|
|
|
|
it('registers propose_post_metadata tool', () => {
|
|
const mcpServer = server.createMcpServer();
|
|
expect(hasRegistered(mcpServer, '_registeredTools', 'propose_post_metadata')).toBe(true);
|
|
});
|
|
|
|
it('registers count_posts tool', () => {
|
|
const mcpServer = server.createMcpServer();
|
|
expect(hasRegistered(mcpServer, '_registeredTools', 'count_posts')).toBe(true);
|
|
});
|
|
|
|
it('registers accept_proposal tool', () => {
|
|
const mcpServer = server.createMcpServer();
|
|
expect(hasRegistered(mcpServer, '_registeredTools', 'accept_proposal')).toBe(true);
|
|
});
|
|
|
|
it('registers discard_proposal tool', () => {
|
|
const mcpServer = server.createMcpServer();
|
|
expect(hasRegistered(mcpServer, '_registeredTools', 'discard_proposal')).toBe(true);
|
|
});
|
|
});
|
|
|
|
describe('registered resources', () => {
|
|
it('registers bds://posts resource', () => {
|
|
const mcpServer = server.createMcpServer();
|
|
expect(hasRegistered(mcpServer, '_registeredResources', 'bds://posts')).toBe(true);
|
|
});
|
|
|
|
it('registers bds://media resource', () => {
|
|
const mcpServer = server.createMcpServer();
|
|
expect(hasRegistered(mcpServer, '_registeredResources', 'bds://media')).toBe(true);
|
|
});
|
|
|
|
it('registers bds://tags resource', () => {
|
|
const mcpServer = server.createMcpServer();
|
|
expect(hasRegistered(mcpServer, '_registeredResources', 'bds://tags')).toBe(true);
|
|
});
|
|
|
|
it('registers bds://categories resource', () => {
|
|
const mcpServer = server.createMcpServer();
|
|
expect(hasRegistered(mcpServer, '_registeredResources', 'bds://categories')).toBe(true);
|
|
});
|
|
|
|
it('registers bds://stats resource', () => {
|
|
const mcpServer = server.createMcpServer();
|
|
expect(hasRegistered(mcpServer, '_registeredResources', 'bds://stats')).toBe(true);
|
|
});
|
|
});
|
|
|
|
describe('registered resource templates', () => {
|
|
it('registers post resource template', () => {
|
|
const mcpServer = server.createMcpServer();
|
|
expect(hasRegistered(mcpServer, '_registeredResourceTemplates', 'post')).toBe(true);
|
|
});
|
|
|
|
it('registers media-item resource template', () => {
|
|
const mcpServer = server.createMcpServer();
|
|
expect(hasRegistered(mcpServer, '_registeredResourceTemplates', 'media-item')).toBe(true);
|
|
});
|
|
|
|
it('registers post-backlinks resource template', () => {
|
|
const mcpServer = server.createMcpServer();
|
|
expect(hasRegistered(mcpServer, '_registeredResourceTemplates', 'post-backlinks')).toBe(true);
|
|
});
|
|
|
|
it('registers post-outlinks resource template', () => {
|
|
const mcpServer = server.createMcpServer();
|
|
expect(hasRegistered(mcpServer, '_registeredResourceTemplates', 'post-outlinks')).toBe(true);
|
|
});
|
|
|
|
it('registers post-media resource template', () => {
|
|
const mcpServer = server.createMcpServer();
|
|
expect(hasRegistered(mcpServer, '_registeredResourceTemplates', 'post-media')).toBe(true);
|
|
});
|
|
|
|
it('registers media-posts resource template', () => {
|
|
const mcpServer = server.createMcpServer();
|
|
expect(hasRegistered(mcpServer, '_registeredResourceTemplates', 'media-posts')).toBe(true);
|
|
});
|
|
|
|
it('registers media-image resource template', () => {
|
|
const mcpServer = server.createMcpServer();
|
|
expect(hasRegistered(mcpServer, '_registeredResourceTemplates', 'media-image')).toBe(true);
|
|
});
|
|
|
|
it('registers posts-page resource template for pagination', () => {
|
|
const mcpServer = server.createMcpServer();
|
|
expect(hasRegistered(mcpServer, '_registeredResourceTemplates', 'posts-page')).toBe(true);
|
|
});
|
|
|
|
it('registers media-page resource template for pagination', () => {
|
|
const mcpServer = server.createMcpServer();
|
|
expect(hasRegistered(mcpServer, '_registeredResourceTemplates', 'media-page')).toBe(true);
|
|
});
|
|
});
|
|
|
|
describe('cursor encoding', () => {
|
|
it('round-trips offset through encode/decode', () => {
|
|
expect(decodeCursor(encodeCursor(0))).toBe(0);
|
|
expect(decodeCursor(encodeCursor(50))).toBe(50);
|
|
expect(decodeCursor(encodeCursor(999))).toBe(999);
|
|
});
|
|
|
|
it('returns 0 for invalid cursor', () => {
|
|
expect(decodeCursor('!!!invalid!!!')).toBe(0);
|
|
expect(decodeCursor('')).toBe(0);
|
|
});
|
|
|
|
it('returns 0 for negative offset', () => {
|
|
// Encoding a negative offset then decoding should clamp to 0
|
|
expect(decodeCursor(encodeCursor(-5))).toBe(0);
|
|
});
|
|
});
|
|
|
|
describe('registered prompts', () => {
|
|
it('registers draft-blog-post prompt', () => {
|
|
const mcpServer = server.createMcpServer();
|
|
expect(hasRegistered(mcpServer, '_registeredPrompts', 'draft-blog-post')).toBe(true);
|
|
});
|
|
|
|
it('registers improve-media-metadata prompt', () => {
|
|
const mcpServer = server.createMcpServer();
|
|
expect(hasRegistered(mcpServer, '_registeredPrompts', 'improve-media-metadata')).toBe(true);
|
|
});
|
|
|
|
it('registers content-audit prompt', () => {
|
|
const mcpServer = server.createMcpServer();
|
|
expect(hasRegistered(mcpServer, '_registeredPrompts', 'content-audit')).toBe(true);
|
|
});
|
|
});
|
|
|
|
describe('start and stop', () => {
|
|
it('starts the server and returns a port', async () => {
|
|
const port = await server.start(0);
|
|
expect(port).toBeGreaterThan(0);
|
|
await server.stop();
|
|
});
|
|
|
|
it('stop is idempotent', async () => {
|
|
await server.start(0);
|
|
await server.stop();
|
|
await expect(server.stop()).resolves.toBeUndefined();
|
|
});
|
|
});
|
|
|
|
describe('cleanup', () => {
|
|
it('cleans up proposals and stops the server', async () => {
|
|
server.proposalStore.create('draftPost', { postId: 'test' });
|
|
await server.start(0);
|
|
await server.cleanup();
|
|
expect(server.proposalStore.getAll()).toHaveLength(0);
|
|
});
|
|
});
|
|
|
|
describe('accept_proposal', () => {
|
|
it('accepts a draftPost proposal by publishing', async () => {
|
|
mockPostEngine.publishPost.mockResolvedValue({
|
|
id: 'post-1', title: 'Test', slug: 'test', content: '', status: 'published',
|
|
tags: [], categories: [], createdAt: new Date(), updatedAt: new Date(), publishedAt: new Date(),
|
|
});
|
|
|
|
const proposalId = server.proposalStore.create('draftPost', { postId: 'post-1' });
|
|
const result = await server.acceptProposal(proposalId);
|
|
expect(result.success).toBe(true);
|
|
expect(mockPostEngine.publishPost).toHaveBeenCalledWith('post-1');
|
|
expect(server.proposalStore.get(proposalId)).toBeUndefined();
|
|
});
|
|
|
|
it('accepts a proposeScript proposal by creating script', async () => {
|
|
const proposalId = server.proposalStore.create('proposeScript', {
|
|
scriptId: 'script-1',
|
|
});
|
|
const result = await server.acceptProposal(proposalId);
|
|
expect(result.success).toBe(true);
|
|
expect(mockScriptEngine.publishScript).toHaveBeenCalledWith('script-1');
|
|
expect(server.proposalStore.get(proposalId)).toBeUndefined();
|
|
});
|
|
|
|
it('accepts a proposeTemplate proposal by creating template', async () => {
|
|
const proposalId = server.proposalStore.create('proposeTemplate', {
|
|
templateId: 'tpl-1',
|
|
});
|
|
const result = await server.acceptProposal(proposalId);
|
|
expect(result.success).toBe(true);
|
|
expect(mockTemplateEngine.publishTemplate).toHaveBeenCalledWith('tpl-1');
|
|
});
|
|
|
|
it('accepts a proposeMediaMetadata proposal by updating media', async () => {
|
|
mockMediaEngine.updateMedia.mockResolvedValue({ id: 'media-1' });
|
|
const proposalId = server.proposalStore.create('proposeMediaMetadata', {
|
|
mediaId: 'media-1', changes: { alt: 'New alt text' },
|
|
});
|
|
const result = await server.acceptProposal(proposalId);
|
|
expect(result.success).toBe(true);
|
|
expect(mockMediaEngine.updateMedia).toHaveBeenCalledWith('media-1', { alt: 'New alt text' });
|
|
});
|
|
|
|
it('accepts a proposePostMetadata proposal by updating post', async () => {
|
|
mockPostEngine.updatePost.mockResolvedValue({ id: 'post-1' });
|
|
const proposalId = server.proposalStore.create('proposePostMetadata', {
|
|
postId: 'post-1', changes: { title: 'Updated Title' },
|
|
});
|
|
const result = await server.acceptProposal(proposalId);
|
|
expect(result.success).toBe(true);
|
|
expect(mockPostEngine.updatePost).toHaveBeenCalledWith('post-1', { title: 'Updated Title' });
|
|
});
|
|
|
|
it('returns failure for non-existent proposal', async () => {
|
|
const result = await server.acceptProposal('non-existent');
|
|
expect(result.success).toBe(false);
|
|
});
|
|
});
|
|
|
|
describe('discard_proposal', () => {
|
|
it('discards a draftPost proposal by deleting the post', async () => {
|
|
const proposalId = server.proposalStore.create('draftPost', { postId: 'post-1' });
|
|
const result = await server.discardProposal(proposalId);
|
|
expect(result.success).toBe(true);
|
|
expect(mockPostEngine.deletePost).toHaveBeenCalledWith('post-1');
|
|
expect(server.proposalStore.get(proposalId)).toBeUndefined();
|
|
});
|
|
|
|
it('discards a proposeScript proposal by removing from store', async () => {
|
|
const proposalId = server.proposalStore.create('proposeScript', { scriptId: 'script-1' });
|
|
const result = await server.discardProposal(proposalId);
|
|
expect(result.success).toBe(true);
|
|
expect(mockScriptEngine.deleteDraftScript).toHaveBeenCalledWith('script-1');
|
|
expect(server.proposalStore.get(proposalId)).toBeUndefined();
|
|
});
|
|
|
|
it('returns failure for non-existent proposal', async () => {
|
|
const result = await server.discardProposal('non-existent');
|
|
expect(result.success).toBe(false);
|
|
});
|
|
});
|
|
|
|
// ── Tool annotations ────────────────────────────────────────────────
|
|
|
|
describe('tool annotations', () => {
|
|
function getToolAnnotations(toolName: string): Record<string, unknown> | undefined {
|
|
const mcpServer = server.createMcpServer();
|
|
const tool = (mcpServer as Record<string, Record<string, { annotations?: Record<string, unknown> }>>)._registeredTools[toolName];
|
|
return tool?.annotations;
|
|
}
|
|
|
|
it('search_posts has readOnlyHint true', () => {
|
|
const annotations = getToolAnnotations('search_posts');
|
|
expect(annotations).toEqual({ readOnlyHint: true, openWorldHint: false });
|
|
});
|
|
|
|
it('draft_post has readOnlyHint false, destructiveHint false', () => {
|
|
const annotations = getToolAnnotations('draft_post');
|
|
expect(annotations).toMatchObject({ readOnlyHint: false, destructiveHint: false });
|
|
});
|
|
|
|
it('propose_script has readOnlyHint false, destructiveHint false', () => {
|
|
const annotations = getToolAnnotations('propose_script');
|
|
expect(annotations).toMatchObject({ readOnlyHint: false, destructiveHint: false });
|
|
});
|
|
|
|
it('propose_template has readOnlyHint false, destructiveHint false', () => {
|
|
const annotations = getToolAnnotations('propose_template');
|
|
expect(annotations).toMatchObject({ readOnlyHint: false, destructiveHint: false });
|
|
});
|
|
|
|
it('propose_media_metadata has readOnlyHint false, destructiveHint false', () => {
|
|
const annotations = getToolAnnotations('propose_media_metadata');
|
|
expect(annotations).toMatchObject({ readOnlyHint: false, destructiveHint: false });
|
|
});
|
|
|
|
it('propose_post_metadata has readOnlyHint false, destructiveHint false', () => {
|
|
const annotations = getToolAnnotations('propose_post_metadata');
|
|
expect(annotations).toMatchObject({ readOnlyHint: false, destructiveHint: false });
|
|
});
|
|
|
|
it('accept_proposal has readOnlyHint false, destructiveHint false, idempotentHint true', () => {
|
|
const annotations = getToolAnnotations('accept_proposal');
|
|
expect(annotations).toEqual({ readOnlyHint: false, destructiveHint: false, idempotentHint: true });
|
|
});
|
|
|
|
it('discard_proposal has readOnlyHint false, destructiveHint true, idempotentHint true', () => {
|
|
const annotations = getToolAnnotations('discard_proposal');
|
|
expect(annotations).toEqual({ readOnlyHint: false, destructiveHint: true, idempotentHint: true });
|
|
});
|
|
|
|
it('check_term has readOnlyHint true', () => {
|
|
const annotations = getToolAnnotations('check_term');
|
|
expect(annotations).toEqual({ readOnlyHint: true, openWorldHint: false });
|
|
});
|
|
});
|
|
|
|
// ── Tool visibility ─────────────────────────────────────────────────
|
|
|
|
describe('tool visibility', () => {
|
|
function getToolMeta(toolName: string): Record<string, unknown> | undefined {
|
|
const mcpServer = server.createMcpServer();
|
|
const tool = (mcpServer as Record<string, Record<string, { _meta?: Record<string, unknown> }>>)._registeredTools[toolName];
|
|
return tool?._meta;
|
|
}
|
|
|
|
it('accept_proposal has app-only visibility', () => {
|
|
const meta = getToolMeta('accept_proposal');
|
|
expect(meta).toBeDefined();
|
|
const ui = (meta as Record<string, unknown>).ui as Record<string, unknown>;
|
|
expect(ui?.visibility).toEqual(['app']);
|
|
});
|
|
|
|
it('discard_proposal has app-only visibility', () => {
|
|
const meta = getToolMeta('discard_proposal');
|
|
expect(meta).toBeDefined();
|
|
const ui = (meta as Record<string, unknown>).ui as Record<string, unknown>;
|
|
expect(ui?.visibility).toEqual(['app']);
|
|
});
|
|
|
|
it('draft_post has model+app visibility (default)', () => {
|
|
const meta = getToolMeta('draft_post');
|
|
expect(meta).toBeDefined();
|
|
const ui = (meta as Record<string, unknown>).ui as Record<string, unknown>;
|
|
// no explicit visibility = default ["model", "app"]
|
|
expect(ui?.visibility).toBeUndefined();
|
|
});
|
|
});
|
|
|
|
// ── Resource handler behavior ───────────────────────────────────────
|
|
|
|
describe('resource handlers', () => {
|
|
function getResource(mcpServer: unknown, uri: string) {
|
|
return (mcpServer as Record<string, Record<string, { readCallback: (...args: unknown[]) => Promise<unknown> }>>)._registeredResources[uri];
|
|
}
|
|
|
|
function getResourceTemplate(mcpServer: unknown, name: string) {
|
|
return (mcpServer as Record<string, Record<string, { readCallback: (...args: unknown[]) => Promise<unknown> }>>)._registeredResourceTemplates[name];
|
|
}
|
|
|
|
it('bds://posts calls getAllPosts with pagination limit and returns JSON', async () => {
|
|
const postsData = { items: [{ id: 'p1', title: 'Hello' }], hasMore: false, total: 1 };
|
|
mockPostEngine.getAllPosts.mockResolvedValue(postsData);
|
|
const mcpServer = server.createMcpServer();
|
|
const resource = getResource(mcpServer, 'bds://posts');
|
|
const result = await resource.readCallback(new URL('bds://posts'), {}) as { contents: Array<{ text: string }> };
|
|
expect(mockPostEngine.getAllPosts).toHaveBeenCalledWith({ limit: DEFAULT_PAGE_SIZE });
|
|
expect(JSON.parse(result.contents[0].text)).toEqual(postsData);
|
|
});
|
|
|
|
it('bds://posts includes nextCursor when hasMore is true', async () => {
|
|
const postsData = {
|
|
items: Array.from({ length: DEFAULT_PAGE_SIZE }, (_, i) => ({ id: `p${i}` })),
|
|
hasMore: true,
|
|
total: 120,
|
|
};
|
|
mockPostEngine.getAllPosts.mockResolvedValue(postsData);
|
|
const mcpServer = server.createMcpServer();
|
|
const resource = getResource(mcpServer, 'bds://posts');
|
|
const result = await resource.readCallback(new URL('bds://posts'), {}) as { contents: Array<{ text: string }> };
|
|
const parsed = JSON.parse(result.contents[0].text);
|
|
expect(parsed.nextCursor).toBeTruthy();
|
|
expect(parsed.hasMore).toBe(true);
|
|
expect(parsed.total).toBe(120);
|
|
// Cursor should decode to the next offset
|
|
expect(decodeCursor(parsed.nextCursor)).toBe(DEFAULT_PAGE_SIZE);
|
|
});
|
|
|
|
it('bds://posts omits nextCursor when hasMore is false', async () => {
|
|
const postsData = { items: [{ id: 'p1' }], hasMore: false, total: 1 };
|
|
mockPostEngine.getAllPosts.mockResolvedValue(postsData);
|
|
const mcpServer = server.createMcpServer();
|
|
const resource = getResource(mcpServer, 'bds://posts');
|
|
const result = await resource.readCallback(new URL('bds://posts'), {}) as { contents: Array<{ text: string }> };
|
|
const parsed = JSON.parse(result.contents[0].text);
|
|
expect(parsed.nextCursor).toBeUndefined();
|
|
});
|
|
|
|
it('bds://media returns paginated response with items, total, hasMore', async () => {
|
|
const allMedia = Array.from({ length: 3 }, (_, i) => ({ id: `m${i}` }));
|
|
mockMediaEngine.getAllMedia.mockResolvedValue(allMedia);
|
|
const mcpServer = server.createMcpServer();
|
|
const resource = getResource(mcpServer, 'bds://media');
|
|
const result = await resource.readCallback(new URL('bds://media'), {}) as { contents: Array<{ text: string }> };
|
|
const parsed = JSON.parse(result.contents[0].text);
|
|
expect(parsed.items).toHaveLength(3);
|
|
expect(parsed.total).toBe(3);
|
|
expect(parsed.hasMore).toBe(false);
|
|
expect(parsed.nextCursor).toBeUndefined();
|
|
});
|
|
|
|
it('bds://media includes nextCursor when more items than page size', async () => {
|
|
const allMedia = Array.from({ length: DEFAULT_PAGE_SIZE + 10 }, (_, i) => ({ id: `m${i}` }));
|
|
mockMediaEngine.getAllMedia.mockResolvedValue(allMedia);
|
|
const mcpServer = server.createMcpServer();
|
|
const resource = getResource(mcpServer, 'bds://media');
|
|
const result = await resource.readCallback(new URL('bds://media'), {}) as { contents: Array<{ text: string }> };
|
|
const parsed = JSON.parse(result.contents[0].text);
|
|
expect(parsed.items).toHaveLength(DEFAULT_PAGE_SIZE);
|
|
expect(parsed.total).toBe(DEFAULT_PAGE_SIZE + 10);
|
|
expect(parsed.hasMore).toBe(true);
|
|
expect(parsed.nextCursor).toBeTruthy();
|
|
expect(decodeCursor(parsed.nextCursor)).toBe(DEFAULT_PAGE_SIZE);
|
|
});
|
|
|
|
it('bds://stats calls getBlogStats and returns JSON', async () => {
|
|
const stats = { totalPosts: 42 };
|
|
mockPostEngine.getBlogStats.mockResolvedValue(stats);
|
|
const mcpServer = server.createMcpServer();
|
|
const resource = getResource(mcpServer, 'bds://stats');
|
|
const result = await resource.readCallback(new URL('bds://stats'), {}) as { contents: Array<{ text: string }> };
|
|
expect(JSON.parse(result.contents[0].text)).toEqual(stats);
|
|
});
|
|
|
|
it('bds://posts/{id} calls getPost with correct id and includes backlinks', async () => {
|
|
const post = { id: 'post-1', title: 'Test' };
|
|
mockPostEngine.getPost.mockResolvedValue(post);
|
|
mockPostEngine.getLinkedBy.mockResolvedValue([
|
|
{ id: 'p2', title: 'Referring Post', slug: 'referring-post' },
|
|
]);
|
|
const mcpServer = server.createMcpServer();
|
|
const tpl = getResourceTemplate(mcpServer, 'post');
|
|
const result = await tpl.readCallback(new URL('bds://posts/post-1'), { id: 'post-1' }, {}) as { contents: Array<{ text: string }> };
|
|
expect(mockPostEngine.getPost).toHaveBeenCalledWith('post-1');
|
|
expect(mockPostEngine.getLinkedBy).toHaveBeenCalledWith('post-1');
|
|
const parsed = JSON.parse(result.contents[0].text);
|
|
expect(parsed.id).toBe('post-1');
|
|
expect(parsed.backlinks).toEqual([
|
|
{ id: 'p2', title: 'Referring Post', slug: 'referring-post' },
|
|
]);
|
|
});
|
|
|
|
it('bds://posts/{id} returns empty backlinks when none exist', async () => {
|
|
const post = { id: 'post-1', title: 'Test' };
|
|
mockPostEngine.getPost.mockResolvedValue(post);
|
|
mockPostEngine.getLinkedBy.mockResolvedValue([]);
|
|
const mcpServer = server.createMcpServer();
|
|
const tpl = getResourceTemplate(mcpServer, 'post');
|
|
const result = await tpl.readCallback(new URL('bds://posts/post-1'), { id: 'post-1' }, {}) as { contents: Array<{ text: string }> };
|
|
const parsed = JSON.parse(result.contents[0].text);
|
|
expect(parsed.backlinks).toEqual([]);
|
|
});
|
|
|
|
it('posts-page template decodes cursor and passes offset to getAllPosts', async () => {
|
|
const cursor = encodeCursor(50);
|
|
mockPostEngine.getAllPosts.mockResolvedValue({ items: [{ id: 'p50' }], hasMore: false, total: 60 });
|
|
const mcpServer = server.createMcpServer();
|
|
const tpl = getResourceTemplate(mcpServer, 'posts-page');
|
|
const result = await tpl.readCallback(new URL(`bds://posts?cursor=${cursor}`), { cursor }, {}) as { contents: Array<{ text: string }> };
|
|
expect(mockPostEngine.getAllPosts).toHaveBeenCalledWith({ limit: DEFAULT_PAGE_SIZE, offset: 50 });
|
|
const parsed = JSON.parse(result.contents[0].text);
|
|
expect(parsed.items).toEqual([{ id: 'p50' }]);
|
|
expect(parsed.nextCursor).toBeUndefined();
|
|
});
|
|
|
|
it('posts-page template includes nextCursor for subsequent pages', async () => {
|
|
const cursor = encodeCursor(50);
|
|
mockPostEngine.getAllPosts.mockResolvedValue({
|
|
items: Array.from({ length: DEFAULT_PAGE_SIZE }, (_, i) => ({ id: `p${50 + i}` })),
|
|
hasMore: true,
|
|
total: 200,
|
|
});
|
|
const mcpServer = server.createMcpServer();
|
|
const tpl = getResourceTemplate(mcpServer, 'posts-page');
|
|
const result = await tpl.readCallback(new URL(`bds://posts?cursor=${cursor}`), { cursor }, {}) as { contents: Array<{ text: string }> };
|
|
const parsed = JSON.parse(result.contents[0].text);
|
|
expect(parsed.nextCursor).toBeTruthy();
|
|
expect(decodeCursor(parsed.nextCursor)).toBe(100);
|
|
});
|
|
|
|
it('media-page template decodes cursor and returns correct slice', async () => {
|
|
const allMedia = Array.from({ length: 80 }, (_, i) => ({ id: `m${i}` }));
|
|
mockMediaEngine.getAllMedia.mockResolvedValue(allMedia);
|
|
const cursor = encodeCursor(50);
|
|
const mcpServer = server.createMcpServer();
|
|
const tpl = getResourceTemplate(mcpServer, 'media-page');
|
|
const result = await tpl.readCallback(new URL(`bds://media?cursor=${cursor}`), { cursor }, {}) as { contents: Array<{ text: string }> };
|
|
const parsed = JSON.parse(result.contents[0].text);
|
|
expect(parsed.items).toHaveLength(30);
|
|
expect(parsed.items[0].id).toBe('m50');
|
|
expect(parsed.total).toBe(80);
|
|
expect(parsed.hasMore).toBe(false);
|
|
expect(parsed.nextCursor).toBeUndefined();
|
|
});
|
|
|
|
it('media-page template includes nextCursor when more pages remain', async () => {
|
|
const allMedia = Array.from({ length: 120 }, (_, i) => ({ id: `m${i}` }));
|
|
mockMediaEngine.getAllMedia.mockResolvedValue(allMedia);
|
|
const cursor = encodeCursor(0);
|
|
const mcpServer = server.createMcpServer();
|
|
const tpl = getResourceTemplate(mcpServer, 'media-page');
|
|
const result = await tpl.readCallback(new URL(`bds://media?cursor=${cursor}`), { cursor }, {}) as { contents: Array<{ text: string }> };
|
|
const parsed = JSON.parse(result.contents[0].text);
|
|
expect(parsed.items).toHaveLength(DEFAULT_PAGE_SIZE);
|
|
expect(parsed.hasMore).toBe(true);
|
|
expect(decodeCursor(parsed.nextCursor)).toBe(DEFAULT_PAGE_SIZE);
|
|
});
|
|
|
|
it('bds://posts/{id}/media calls getLinkedMediaDataForPost', async () => {
|
|
const linkedMedia = [{ id: 'm1', filename: 'photo.jpg' }];
|
|
mockPostMediaEngine.getLinkedMediaDataForPost.mockResolvedValue(linkedMedia);
|
|
const mcpServer = server.createMcpServer();
|
|
const tpl = getResourceTemplate(mcpServer, 'post-media');
|
|
const result = await tpl.readCallback(new URL('bds://posts/p1/media'), { id: 'p1' }, {}) as { contents: Array<{ text: string }> };
|
|
expect(mockPostMediaEngine.getLinkedMediaDataForPost).toHaveBeenCalledWith('p1');
|
|
expect(JSON.parse(result.contents[0].text)).toEqual(linkedMedia);
|
|
});
|
|
|
|
it('bds://media/{id}/image returns thumbnail blob for images', async () => {
|
|
mockMediaEngine.getMedia.mockResolvedValue({ id: 'img-1', mimeType: 'image/jpeg', filename: 'photo.jpg' });
|
|
mockMediaEngine.getThumbnailDataUrl.mockResolvedValue('data:image/webp;base64,AAAA');
|
|
const mcpServer = server.createMcpServer();
|
|
const tpl = getResourceTemplate(mcpServer, 'media-image');
|
|
const result = await tpl.readCallback(new URL('bds://media/img-1/image'), { id: 'img-1' }, {}) as { contents: Array<{ mimeType: string; blob: string }> };
|
|
expect(mockMediaEngine.getThumbnailDataUrl).toHaveBeenCalledWith('img-1', 'medium');
|
|
expect(result.contents[0].mimeType).toBe('image/webp');
|
|
expect(result.contents[0].blob).toBe('AAAA');
|
|
});
|
|
|
|
it('bds://media/{id}/image returns text error for non-images', async () => {
|
|
mockMediaEngine.getMedia.mockResolvedValue({ id: 'doc-1', mimeType: 'application/pdf', filename: 'doc.pdf' });
|
|
const mcpServer = server.createMcpServer();
|
|
const tpl = getResourceTemplate(mcpServer, 'media-image');
|
|
const result = await tpl.readCallback(new URL('bds://media/doc-1/image'), { id: 'doc-1' }, {}) as { contents: Array<{ mimeType: string; text: string }> };
|
|
expect(result.contents[0].mimeType).toBe('text/plain');
|
|
expect(result.contents[0].text).toContain('Not an image');
|
|
});
|
|
|
|
it('bds://media/{id}/image returns text error when thumbnail unavailable', async () => {
|
|
mockMediaEngine.getMedia.mockResolvedValue({ id: 'img-2', mimeType: 'image/png', filename: 'pic.png' });
|
|
mockMediaEngine.getThumbnailDataUrl.mockResolvedValue(null);
|
|
const mcpServer = server.createMcpServer();
|
|
const tpl = getResourceTemplate(mcpServer, 'media-image');
|
|
const result = await tpl.readCallback(new URL('bds://media/img-2/image'), { id: 'img-2' }, {}) as { contents: Array<{ mimeType: string; text: string }> };
|
|
expect(result.contents[0].text).toContain('Thumbnail not available');
|
|
});
|
|
});
|
|
|
|
// ── Tool handler behavior ──────────────────────────────────────────
|
|
|
|
describe('tool handlers', () => {
|
|
function getTool(mcpServer: unknown, name: string) {
|
|
return (mcpServer as Record<string, Record<string, { handler: (...args: unknown[]) => Promise<unknown> }>>)._registeredTools[name];
|
|
}
|
|
|
|
it('search_posts with query only calls searchPosts and includes backlinks', async () => {
|
|
const searchResults = [{ id: 'p1', title: 'Found', slug: 'found' }];
|
|
mockPostEngine.searchPosts.mockResolvedValue(searchResults);
|
|
mockPostEngine.getLinkedBy.mockResolvedValue([{ id: 'px', title: 'Ref', slug: 'ref' }]);
|
|
const mcpServer = server.createMcpServer();
|
|
const tool = getTool(mcpServer, 'search_posts');
|
|
const result = await tool.handler({ query: 'test' }, {}) as { content: Array<{ text: string }> };
|
|
expect(mockPostEngine.searchPosts).toHaveBeenCalledWith('test');
|
|
const parsed = JSON.parse(result.content[0].text);
|
|
expect(parsed.posts[0].backlinks).toEqual([{ id: 'px', title: 'Ref', slug: 'ref' }]);
|
|
expect(parsed.total).toBe(1);
|
|
expect(parsed.offset).toBe(0);
|
|
expect(parsed.limit).toBe(50);
|
|
expect(parsed.hasMore).toBe(false);
|
|
});
|
|
|
|
it('search_posts with query applies offset and limit', async () => {
|
|
const searchResults = Array.from({ length: 10 }, (_, i) => ({ id: `p${i}`, title: `Post ${i}`, slug: `post-${i}` }));
|
|
mockPostEngine.searchPosts.mockResolvedValue(searchResults);
|
|
const mcpServer = server.createMcpServer();
|
|
const tool = getTool(mcpServer, 'search_posts');
|
|
const result = await tool.handler({ query: 'test', offset: 2, limit: 3 }, {}) as { content: Array<{ text: string }> };
|
|
const parsed = JSON.parse(result.content[0].text);
|
|
expect(parsed.posts).toHaveLength(3);
|
|
expect(parsed.posts[0].id).toBe('p2');
|
|
expect(parsed.posts[2].id).toBe('p4');
|
|
expect(parsed.total).toBe(10);
|
|
expect(parsed.hasMore).toBe(true);
|
|
});
|
|
|
|
it('search_posts defaults to limit 50 when not specified', async () => {
|
|
const searchResults = Array.from({ length: 60 }, (_, i) => ({ id: `p${i}`, title: `Post ${i}`, slug: `post-${i}` }));
|
|
mockPostEngine.searchPosts.mockResolvedValue(searchResults);
|
|
const mcpServer = server.createMcpServer();
|
|
const tool = getTool(mcpServer, 'search_posts');
|
|
const result = await tool.handler({ query: 'test' }, {}) as { content: Array<{ text: string }> };
|
|
const parsed = JSON.parse(result.content[0].text);
|
|
expect(parsed.posts).toHaveLength(50);
|
|
expect(parsed.total).toBe(60);
|
|
expect(parsed.hasMore).toBe(true);
|
|
});
|
|
|
|
it('search_posts with filters only calls getPostsFiltered and includes backlinks', async () => {
|
|
const filtered = [{ id: 'p2', title: 'Filtered' }];
|
|
mockPostEngine.getPostsFiltered.mockResolvedValue(filtered);
|
|
mockPostEngine.getLinkedBy.mockResolvedValue([]);
|
|
const mcpServer = server.createMcpServer();
|
|
const tool = getTool(mcpServer, 'search_posts');
|
|
const result = await tool.handler({ category: 'tech', status: 'published' }, {}) as { content: Array<{ text: string }> };
|
|
expect(mockPostEngine.getPostsFiltered).toHaveBeenCalledWith({ categories: ['tech'], status: 'published' });
|
|
const parsed = JSON.parse(result.content[0].text);
|
|
expect(parsed.posts[0].backlinks).toEqual([]);
|
|
expect(parsed.total).toBe(1);
|
|
expect(parsed.hasMore).toBe(false);
|
|
expect(mockPostEngine.getLinkedBy).toHaveBeenCalledWith('p2');
|
|
});
|
|
|
|
it('search_posts with filters applies offset and limit', async () => {
|
|
const filtered = Array.from({ length: 10 }, (_, i) => ({ id: `p${i}`, title: `Post ${i}` }));
|
|
mockPostEngine.getPostsFiltered.mockResolvedValue(filtered);
|
|
const mcpServer = server.createMcpServer();
|
|
const tool = getTool(mcpServer, 'search_posts');
|
|
const result = await tool.handler({ category: 'tech', offset: 3, limit: 2 }, {}) as { content: Array<{ text: string }> };
|
|
const parsed = JSON.parse(result.content[0].text);
|
|
expect(parsed.posts).toHaveLength(2);
|
|
expect(parsed.posts[0].id).toBe('p3');
|
|
expect(parsed.total).toBe(10);
|
|
expect(parsed.hasMore).toBe(true);
|
|
});
|
|
|
|
it('search_posts with query + filters calls searchPostsFiltered and includes backlinks', async () => {
|
|
const combined = [
|
|
{ id: 'p1', title: 'TypeScript Guide', categories: ['tech'] },
|
|
];
|
|
mockPostEngine.searchPostsFiltered.mockResolvedValue({ posts: combined, total: 1 });
|
|
mockPostEngine.getLinkedBy.mockResolvedValue([{ id: 'p9', title: 'See Also', slug: 'see-also' }]);
|
|
const mcpServer = server.createMcpServer();
|
|
const tool = getTool(mcpServer, 'search_posts');
|
|
const result = await tool.handler({ query: 'typescript', category: 'tech' }, {}) as { content: Array<{ text: string }> };
|
|
// Should call searchPostsFiltered, not searchPosts + getPostsFiltered separately
|
|
expect(mockPostEngine.searchPostsFiltered).toHaveBeenCalledWith(
|
|
'typescript',
|
|
{ categories: ['tech'] },
|
|
{ offset: 0, limit: 50 },
|
|
);
|
|
expect(mockPostEngine.searchPosts).not.toHaveBeenCalled();
|
|
expect(mockPostEngine.getPostsFiltered).not.toHaveBeenCalled();
|
|
const parsed = JSON.parse(result.content[0].text);
|
|
expect(parsed.posts).toHaveLength(1);
|
|
expect(parsed.posts[0].id).toBe('p1');
|
|
expect(parsed.posts[0].backlinks).toEqual([{ id: 'p9', title: 'See Also', slug: 'see-also' }]);
|
|
expect(parsed.total).toBe(1);
|
|
expect(parsed.hasMore).toBe(false);
|
|
});
|
|
|
|
it('search_posts with query + filters passes pagination to searchPostsFiltered', async () => {
|
|
mockPostEngine.searchPostsFiltered.mockResolvedValue({ posts: [{ id: 'p3', title: 'Result' }], total: 20 });
|
|
const mcpServer = server.createMcpServer();
|
|
const tool = getTool(mcpServer, 'search_posts');
|
|
const result = await tool.handler({ query: 'keyword', status: 'published', offset: 5, limit: 10 }, {}) as { content: Array<{ text: string }> };
|
|
expect(mockPostEngine.searchPostsFiltered).toHaveBeenCalledWith(
|
|
'keyword',
|
|
{ status: 'published' },
|
|
{ offset: 5, limit: 10 },
|
|
);
|
|
const parsed = JSON.parse(result.content[0].text);
|
|
expect(parsed.total).toBe(20);
|
|
expect(parsed.hasMore).toBe(true);
|
|
});
|
|
|
|
it('search_posts with query + multiple filters builds correct filter', async () => {
|
|
mockPostEngine.searchPostsFiltered.mockResolvedValue({ posts: [], total: 0 });
|
|
const mcpServer = server.createMcpServer();
|
|
const tool = getTool(mcpServer, 'search_posts');
|
|
await tool.handler({ query: 'test', category: 'tech', tags: ['js'], year: 2025, status: 'published' }, {});
|
|
expect(mockPostEngine.searchPostsFiltered).toHaveBeenCalledWith(
|
|
'test',
|
|
{ categories: ['tech'], tags: ['js'], year: 2025, status: 'published' },
|
|
{ offset: 0, limit: 50 },
|
|
);
|
|
});
|
|
|
|
// ── count_posts ──────────────────────────────────────────────────
|
|
|
|
it('count_posts calls getPostCounts with correct args', async () => {
|
|
mockPostEngine.getPostCounts.mockResolvedValue({
|
|
groups: [
|
|
{ month: 1, tag: 'Politik', count: 12 },
|
|
{ month: 2, tag: 'Politik', count: 5 },
|
|
],
|
|
totalPosts: 300,
|
|
});
|
|
const mcpServer = server.createMcpServer();
|
|
const tool = getTool(mcpServer, 'count_posts');
|
|
const result = await tool.handler({ groupBy: ['month', 'tag'], year: 2004 }, {}) as { content: Array<{ text: string }> };
|
|
expect(mockPostEngine.getPostCounts).toHaveBeenCalledWith(
|
|
['month', 'tag'],
|
|
{ year: 2004 },
|
|
);
|
|
const parsed = JSON.parse(result.content[0].text);
|
|
expect(parsed.totalPosts).toBe(300);
|
|
expect(parsed.groups).toHaveLength(2);
|
|
});
|
|
|
|
it('count_posts returns error when month without year', async () => {
|
|
const mcpServer = server.createMcpServer();
|
|
const tool = getTool(mcpServer, 'count_posts');
|
|
const result = await tool.handler({ groupBy: ['tag'], month: 6 }, {}) as { content: Array<{ text: string }>; isError?: boolean };
|
|
expect(result.isError).toBe(true);
|
|
});
|
|
|
|
it('draft_post creates a draft and stores proposal', async () => {
|
|
const createdPost = { id: 'new-post', title: 'Draft Title', status: 'draft' };
|
|
mockPostEngine.createPost.mockResolvedValue(createdPost);
|
|
const mcpServer = server.createMcpServer();
|
|
const tool = getTool(mcpServer, 'draft_post');
|
|
const result = await tool.handler({ title: 'Draft Title', content: '# Hello' }, {}) as { content: Array<{ text: string }> };
|
|
expect(mockPostEngine.createPost).toHaveBeenCalledWith(expect.objectContaining({ title: 'Draft Title', content: '# Hello', status: 'draft' }));
|
|
const parsed = JSON.parse(result.content[0].text);
|
|
expect(parsed.proposalId).toBeTruthy();
|
|
expect(parsed.post).toEqual(createdPost);
|
|
// Verify proposal is in the store
|
|
const proposal = server.proposalStore.get(parsed.proposalId);
|
|
expect(proposal).toBeDefined();
|
|
expect(proposal!.type).toBe('draftPost');
|
|
expect(proposal!.data.postId).toBe('new-post');
|
|
});
|
|
|
|
it('propose_script stores proposal in ProposalStore', async () => {
|
|
const mcpServer = server.createMcpServer();
|
|
const tool = getTool(mcpServer, 'propose_script');
|
|
const result = await tool.handler({ title: 'My Script', kind: 'macro', content: 'print("hi")' }, {}) as { content: Array<{ text: string }> };
|
|
const parsed = JSON.parse(result.content[0].text);
|
|
expect(parsed.proposalId).toBeTruthy();
|
|
const proposal = server.proposalStore.get(parsed.proposalId);
|
|
expect(proposal).toBeDefined();
|
|
expect(proposal!.type).toBe('proposeScript');
|
|
expect(mockScriptEngine.createDraftScript).toHaveBeenCalledWith({
|
|
title: 'My Script', kind: 'macro', content: 'print("hi")', entrypoint: undefined,
|
|
});
|
|
expect(proposal!.data.scriptId).toBe('script-1');
|
|
});
|
|
|
|
it('propose_script calls validateScript and includes validation result in preview', async () => {
|
|
mockScriptEngine.validateScript.mockResolvedValue({ valid: true, errors: [] });
|
|
const mcpServer = server.createMcpServer();
|
|
const tool = getTool(mcpServer, 'propose_script');
|
|
const result = await tool.handler({ title: 'Valid Script', kind: 'macro', content: 'print("hello")' }, {}) as { content: Array<{ text: string }> };
|
|
expect(mockScriptEngine.validateScript).toHaveBeenCalledWith('print("hello")');
|
|
const parsed = JSON.parse(result.content[0].text);
|
|
expect(parsed.preview.syntaxValid).toBe(true);
|
|
expect(parsed.preview.syntaxErrors).toEqual([]);
|
|
});
|
|
|
|
it('propose_script includes syntax errors in preview when validation fails', async () => {
|
|
mockScriptEngine.validateScript.mockResolvedValue({ valid: false, errors: ['invalid syntax (line 1, col 6)'] });
|
|
const mcpServer = server.createMcpServer();
|
|
const tool = getTool(mcpServer, 'propose_script');
|
|
const result = await tool.handler({ title: 'Bad Script', kind: 'macro', content: 'def (' }, {}) as { content: Array<{ text: string }> };
|
|
const parsed = JSON.parse(result.content[0].text);
|
|
expect(parsed.proposalId).toBeTruthy();
|
|
expect(parsed.preview.syntaxValid).toBe(false);
|
|
expect(parsed.preview.syntaxErrors).toEqual(['invalid syntax (line 1, col 6)']);
|
|
});
|
|
|
|
it('propose_template calls validateTemplate and includes validation result in preview', async () => {
|
|
mockTemplateEngine.validateTemplate.mockResolvedValue({ valid: true, errors: [] });
|
|
const mcpServer = server.createMcpServer();
|
|
const tool = getTool(mcpServer, 'propose_template');
|
|
const result = await tool.handler({ title: 'Valid Template', kind: 'post', content: '<h1>{{ title }}</h1>' }, {}) as { content: Array<{ text: string }> };
|
|
expect(mockTemplateEngine.validateTemplate).toHaveBeenCalledWith('<h1>{{ title }}</h1>');
|
|
const parsed = JSON.parse(result.content[0].text);
|
|
expect(parsed.preview.syntaxValid).toBe(true);
|
|
expect(parsed.preview.syntaxErrors).toEqual([]);
|
|
});
|
|
|
|
it('propose_template includes syntax errors in preview when validation fails', async () => {
|
|
mockTemplateEngine.validateTemplate.mockResolvedValue({ valid: false, errors: ['tag "{% invalid" not closed'] });
|
|
const mcpServer = server.createMcpServer();
|
|
const tool = getTool(mcpServer, 'propose_template');
|
|
const result = await tool.handler({ title: 'Bad Template', kind: 'post', content: '{% invalid' }, {}) as { content: Array<{ text: string }> };
|
|
const parsed = JSON.parse(result.content[0].text);
|
|
expect(parsed.proposalId).toBeTruthy();
|
|
expect(parsed.preview.syntaxValid).toBe(false);
|
|
expect(parsed.preview.syntaxErrors).toEqual(['tag "{% invalid" not closed']);
|
|
});
|
|
|
|
it('propose_media_metadata loads current media and stores proposal', async () => {
|
|
const currentMedia = { id: 'img-1', alt: 'Old alt', title: 'Old title' };
|
|
mockMediaEngine.getMedia.mockResolvedValue(currentMedia);
|
|
const mcpServer = server.createMcpServer();
|
|
const tool = getTool(mcpServer, 'propose_media_metadata');
|
|
const result = await tool.handler({ mediaId: 'img-1', alt: 'New alt' }, {}) as { content: Array<{ text: string }> };
|
|
const parsed = JSON.parse(result.content[0].text);
|
|
expect(parsed.current).toEqual(currentMedia);
|
|
expect(parsed.proposed).toEqual({ alt: 'New alt' });
|
|
const proposal = server.proposalStore.get(parsed.proposalId);
|
|
expect(proposal!.type).toBe('proposeMediaMetadata');
|
|
expect(proposal!.data.mediaId).toBe('img-1');
|
|
});
|
|
|
|
it('propose_media_metadata returns error for non-existent media', async () => {
|
|
mockMediaEngine.getMedia.mockResolvedValue(null);
|
|
const mcpServer = server.createMcpServer();
|
|
const tool = getTool(mcpServer, 'propose_media_metadata');
|
|
const result = await tool.handler({ mediaId: 'no-such', alt: 'New alt' }, {}) as { content: Array<{ text: string }>; isError?: boolean };
|
|
expect(result.isError).toBe(true);
|
|
const parsed = JSON.parse(result.content[0].text);
|
|
expect(parsed.error).toContain('not found');
|
|
// No proposal should be created
|
|
expect(server.proposalStore.getAll()).toHaveLength(0);
|
|
});
|
|
|
|
it('propose_post_metadata loads current post and stores proposal', async () => {
|
|
const currentPost = { id: 'post-1', title: 'Old Title', excerpt: 'Old excerpt' };
|
|
mockPostEngine.getPost.mockResolvedValue(currentPost);
|
|
const mcpServer = server.createMcpServer();
|
|
const tool = getTool(mcpServer, 'propose_post_metadata');
|
|
const result = await tool.handler({ postId: 'post-1', title: 'New Title' }, {}) as { content: Array<{ text: string }> };
|
|
const parsed = JSON.parse(result.content[0].text);
|
|
expect(parsed.current).toEqual(currentPost);
|
|
expect(parsed.proposed).toEqual({ title: 'New Title' });
|
|
const proposal = server.proposalStore.get(parsed.proposalId);
|
|
expect(proposal!.type).toBe('proposePostMetadata');
|
|
});
|
|
|
|
it('propose_post_metadata returns error for non-existent post', async () => {
|
|
mockPostEngine.getPost.mockResolvedValue(null);
|
|
const mcpServer = server.createMcpServer();
|
|
const tool = getTool(mcpServer, 'propose_post_metadata');
|
|
const result = await tool.handler({ postId: 'no-such', title: 'New Title' }, {}) as { content: Array<{ text: string }>; isError?: boolean };
|
|
expect(result.isError).toBe(true);
|
|
const parsed = JSON.parse(result.content[0].text);
|
|
expect(parsed.error).toContain('not found');
|
|
expect(server.proposalStore.getAll()).toHaveLength(0);
|
|
});
|
|
|
|
it('draft_post returns error when createPost fails', async () => {
|
|
mockPostEngine.createPost.mockRejectedValue(new Error('No active project'));
|
|
const mcpServer = server.createMcpServer();
|
|
const tool = getTool(mcpServer, 'draft_post');
|
|
const result = await tool.handler({ title: 'Test', content: '# Hello' }, {}) as { content: Array<{ text: string }>; isError?: boolean };
|
|
expect(result.isError).toBe(true);
|
|
const parsed = JSON.parse(result.content[0].text);
|
|
expect(parsed.error).toContain('No active project');
|
|
expect(server.proposalStore.getAll()).toHaveLength(0);
|
|
});
|
|
|
|
it('propose_media_metadata returns error when getMedia throws', async () => {
|
|
mockMediaEngine.getMedia.mockRejectedValue(new Error('DB error'));
|
|
const mcpServer = server.createMcpServer();
|
|
const tool = getTool(mcpServer, 'propose_media_metadata');
|
|
const result = await tool.handler({ mediaId: 'img-1', alt: 'New' }, {}) as { content: Array<{ text: string }>; isError?: boolean };
|
|
expect(result.isError).toBe(true);
|
|
const parsed = JSON.parse(result.content[0].text);
|
|
expect(parsed.error).toContain('DB error');
|
|
});
|
|
|
|
it('propose_post_metadata returns error when getPost throws', async () => {
|
|
mockPostEngine.getPost.mockRejectedValue(new Error('DB error'));
|
|
const mcpServer = server.createMcpServer();
|
|
const tool = getTool(mcpServer, 'propose_post_metadata');
|
|
const result = await tool.handler({ postId: 'p1', title: 'New' }, {}) as { content: Array<{ text: string }>; isError?: boolean };
|
|
expect(result.isError).toBe(true);
|
|
const parsed = JSON.parse(result.content[0].text);
|
|
expect(parsed.error).toContain('DB error');
|
|
});
|
|
|
|
// ── check_term tool ──────────────────────────────────────────────
|
|
|
|
it('registers check_term tool', () => {
|
|
const mcpServer = server.createMcpServer();
|
|
expect(hasRegistered(mcpServer, '_registeredTools', 'check_term')).toBe(true);
|
|
});
|
|
|
|
it('check_term returns category and tag info for a term that exists as both', async () => {
|
|
mockPostEngine.getCategoriesWithCounts.mockResolvedValue([
|
|
{ category: 'wiki', count: 3 },
|
|
{ category: 'tech', count: 5 },
|
|
]);
|
|
mockPostEngine.getTagsWithCounts.mockResolvedValue([
|
|
{ tag: 'wiki', count: 1 },
|
|
{ tag: 'python', count: 4 },
|
|
]);
|
|
const mcpServer = server.createMcpServer();
|
|
const tool = getTool(mcpServer, 'check_term');
|
|
const result = await tool.handler({ term: 'wiki' }, {}) as { content: Array<{ text: string }> };
|
|
const parsed = JSON.parse(result.content[0].text);
|
|
expect(parsed.term).toBe('wiki');
|
|
expect(parsed.asCategory).toBe(true);
|
|
expect(parsed.categoryPostCount).toBe(3);
|
|
expect(parsed.asTag).toBe(true);
|
|
expect(parsed.tagPostCount).toBe(1);
|
|
});
|
|
|
|
it('check_term returns false for a term that does not exist', async () => {
|
|
mockPostEngine.getCategoriesWithCounts.mockResolvedValue([
|
|
{ category: 'tech', count: 5 },
|
|
]);
|
|
mockPostEngine.getTagsWithCounts.mockResolvedValue([
|
|
{ tag: 'python', count: 4 },
|
|
]);
|
|
const mcpServer = server.createMcpServer();
|
|
const tool = getTool(mcpServer, 'check_term');
|
|
const result = await tool.handler({ term: 'nonexistent' }, {}) as { content: Array<{ text: string }> };
|
|
const parsed = JSON.parse(result.content[0].text);
|
|
expect(parsed.term).toBe('nonexistent');
|
|
expect(parsed.asCategory).toBe(false);
|
|
expect(parsed.categoryPostCount).toBe(0);
|
|
expect(parsed.asTag).toBe(false);
|
|
expect(parsed.tagPostCount).toBe(0);
|
|
});
|
|
|
|
it('check_term is case-insensitive', async () => {
|
|
mockPostEngine.getCategoriesWithCounts.mockResolvedValue([
|
|
{ category: 'Wiki', count: 3 },
|
|
]);
|
|
mockPostEngine.getTagsWithCounts.mockResolvedValue([]);
|
|
const mcpServer = server.createMcpServer();
|
|
const tool = getTool(mcpServer, 'check_term');
|
|
const result = await tool.handler({ term: 'wiki' }, {}) as { content: Array<{ text: string }> };
|
|
const parsed = JSON.parse(result.content[0].text);
|
|
expect(parsed.asCategory).toBe(true);
|
|
expect(parsed.categoryPostCount).toBe(3);
|
|
});
|
|
|
|
// ── search_posts month validation ────────────────────────────────
|
|
|
|
it('search_posts returns error when month is given without year', async () => {
|
|
const mcpServer = server.createMcpServer();
|
|
const tool = getTool(mcpServer, 'search_posts');
|
|
const result = await tool.handler({ month: 3 }, {}) as { content: Array<{ text: string }>; isError?: boolean };
|
|
expect(result.isError).toBe(true);
|
|
const parsed = JSON.parse(result.content[0].text);
|
|
expect(parsed.error).toContain('month');
|
|
expect(parsed.error).toContain('year');
|
|
});
|
|
|
|
it('search_posts accepts month when year is also given', async () => {
|
|
mockPostEngine.getPostsFiltered.mockResolvedValue([]);
|
|
const mcpServer = server.createMcpServer();
|
|
const tool = getTool(mcpServer, 'search_posts');
|
|
const result = await tool.handler({ year: 2025, month: 3 }, {}) as { content: Array<{ text: string }>; isError?: boolean };
|
|
expect(result.isError).toBeUndefined();
|
|
expect(mockPostEngine.getPostsFiltered).toHaveBeenCalledWith(
|
|
expect.objectContaining({ year: 2025, month: 3 }),
|
|
);
|
|
});
|
|
|
|
// ── search_posts ambiguity hints ─────────────────────────────────
|
|
|
|
it('search_posts includes hint when category term also exists as tag', async () => {
|
|
mockPostEngine.getPostsFiltered.mockResolvedValue([
|
|
{ id: 'p1', title: 'Post', categories: ['wiki'], tags: [] },
|
|
]);
|
|
mockPostEngine.getTagsWithCounts.mockResolvedValue([
|
|
{ tag: 'wiki', count: 2 },
|
|
]);
|
|
mockPostEngine.getLinkedBy.mockResolvedValue([]);
|
|
mockPostEngine.getLinksTo.mockResolvedValue([]);
|
|
const mcpServer = server.createMcpServer();
|
|
const tool = getTool(mcpServer, 'search_posts');
|
|
const result = await tool.handler({ category: 'wiki' }, {}) as { content: Array<{ text: string }> };
|
|
// Should have a second content item with the hint
|
|
expect(result.content.length).toBeGreaterThan(1);
|
|
const hintText = result.content.find(c => c.text.includes('wiki'))?.text ?? '';
|
|
expect(hintText).toContain('tag');
|
|
});
|
|
|
|
it('search_posts includes hint when tag terms also exist as categories', async () => {
|
|
mockPostEngine.getPostsFiltered.mockResolvedValue([
|
|
{ id: 'p1', title: 'Post', categories: [], tags: ['wiki'] },
|
|
]);
|
|
mockPostEngine.getCategoriesWithCounts.mockResolvedValue([
|
|
{ category: 'wiki', count: 3 },
|
|
]);
|
|
mockPostEngine.getLinkedBy.mockResolvedValue([]);
|
|
mockPostEngine.getLinksTo.mockResolvedValue([]);
|
|
const mcpServer = server.createMcpServer();
|
|
const tool = getTool(mcpServer, 'search_posts');
|
|
const result = await tool.handler({ tags: ['wiki'] }, {}) as { content: Array<{ text: string }> };
|
|
expect(result.content.length).toBeGreaterThan(1);
|
|
const hintText = result.content[1].text;
|
|
expect(hintText).toContain('wiki');
|
|
expect(hintText).toContain('category');
|
|
});
|
|
|
|
it('search_posts does not include hint when no ambiguity exists', async () => {
|
|
mockPostEngine.getPostsFiltered.mockResolvedValue([
|
|
{ id: 'p1', title: 'Post', categories: ['tech'], tags: [] },
|
|
]);
|
|
mockPostEngine.getTagsWithCounts.mockResolvedValue([
|
|
{ tag: 'python', count: 4 },
|
|
]);
|
|
mockPostEngine.getLinkedBy.mockResolvedValue([]);
|
|
mockPostEngine.getLinksTo.mockResolvedValue([]);
|
|
const mcpServer = server.createMcpServer();
|
|
const tool = getTool(mcpServer, 'search_posts');
|
|
const result = await tool.handler({ category: 'tech' }, {}) as { content: Array<{ text: string }> };
|
|
expect(result.content).toHaveLength(1);
|
|
});
|
|
});
|
|
|
|
// ── Prompt handler behavior ────────────────────────────────────────
|
|
|
|
describe('prompt handlers', () => {
|
|
function getPrompt(mcpServer: unknown, name: string) {
|
|
return (mcpServer as Record<string, Record<string, { callback: (...args: unknown[]) => Promise<unknown> }>>)._registeredPrompts[name];
|
|
}
|
|
|
|
it('draft-blog-post returns messages with topic', async () => {
|
|
const mcpServer = server.createMcpServer();
|
|
const prompt = getPrompt(mcpServer, 'draft-blog-post');
|
|
const result = await prompt.callback({ topic: 'AI Safety' }, {}) as { messages: Array<{ role: string; content: { type: string; text: string } }> };
|
|
expect(result.messages).toHaveLength(1);
|
|
expect(result.messages[0].role).toBe('user');
|
|
expect(result.messages[0].content.text).toContain('AI Safety');
|
|
expect(result.messages[0].content.text).toContain('draft_post');
|
|
});
|
|
|
|
it('improve-media-metadata returns messages with scope', async () => {
|
|
const mcpServer = server.createMcpServer();
|
|
const prompt = getPrompt(mcpServer, 'improve-media-metadata');
|
|
const result = await prompt.callback({ scope: 'missing-alt' }, {}) as { messages: Array<{ role: string; content: { type: string; text: string } }> };
|
|
expect(result.messages).toHaveLength(1);
|
|
expect(result.messages[0].content.text).toContain('missing alt text');
|
|
});
|
|
|
|
it('content-audit returns messages with category', async () => {
|
|
const mcpServer = server.createMcpServer();
|
|
const prompt = getPrompt(mcpServer, 'content-audit');
|
|
const result = await prompt.callback({ category: 'tech' }, {}) as { messages: Array<{ role: string; content: { type: string; text: string } }> };
|
|
expect(result.messages).toHaveLength(1);
|
|
expect(result.messages[0].content.text).toContain('tech');
|
|
});
|
|
|
|
it('content-audit without category reviews all posts', async () => {
|
|
const mcpServer = server.createMcpServer();
|
|
const prompt = getPrompt(mcpServer, 'content-audit');
|
|
const result = await prompt.callback({}, {}) as { messages: Array<{ role: string; content: { type: string; text: string } }> };
|
|
expect(result.messages[0].content.text).toContain('all posts');
|
|
});
|
|
});
|
|
});
|