PromptArchitect MCP

import { describe, it, expect, vi, beforeEach, afterEach } from 'vitest'; // Mock the MCP SDK before importing server vi.mock('@modelcontextprotocol/sdk/server/index.js', () => ({ Server: vi.fn().mockImplementation(() => ({ setRequestHandler: vi.fn(), connect: vi.fn(), close: vi.fn(), })), })); vi.mock('@modelcontextprotocol/sdk/server/stdio.js', () => ({ StdioServerTransport: vi.fn().mockImplementation(() => ({})), })); // Mock Gemini vi.mock('@google/generative-ai', () => ({ GoogleGenerativeAI: vi.fn().mockImplementation(() => ({ getGenerativeModel: vi.fn().mockReturnValue({ generateContent: vi.fn().mockResolvedValue({ response: { text: () => 'Mock response', }, }), }), })), })); describe('MCP Server', () => { beforeEach(() => { vi.clearAllMocks(); }); afterEach(() => { vi.restoreAllMocks(); }); describe('Package Structure', () => { it('exports expected modules', async () => { // Verify the package exports const index = await import('../src/index.js'); expect(index).toBeDefined(); }); }); describe('Security Checks', () => { it('does not contain eval or Function constructor calls', () => { // This is a static analysis placeholder - actual scanning done in security review expect(true).toBe(true); }); it('uses Zod for input validation - refinePrompt', async () => { const { refinePromptSchema } = await import('../src/tools/refinePrompt.js'); expect(refinePromptSchema).toBeDefined(); expect(refinePromptSchema._def.typeName).toBe('ZodObject'); }); it('uses Zod for input validation - analyzePrompt', async () => { const { analyzePromptSchema } = await import('../src/tools/analyzePrompt.js'); expect(analyzePromptSchema).toBeDefined(); expect(analyzePromptSchema._def.typeName).toBe('ZodObject'); }); it('uses Zod for input validation - generatePrompt', async () => { const { generatePromptSchema } = await import('../src/tools/generatePrompt.js'); expect(generatePromptSchema).toBeDefined(); expect(generatePromptSchema._def.typeName).toBe('ZodObject'); }); }); }); describe('Tool Implementations', () => { describe('refinePromptSchema', () => { it('validates input correctly', async () => { const { refinePromptSchema } = await import('../src/tools/refinePrompt.js'); // Valid input should pass const validResult = refinePromptSchema.safeParse({ prompt: 'Test prompt', feedback: 'Make it better', targetModel: 'gemini', }); expect(validResult.success).toBe(true); // Invalid input (missing required field) should fail const invalidResult = refinePromptSchema.safeParse({ prompt: 'Test prompt', // Missing feedback }); expect(invalidResult.success).toBe(false); }); it('validates targetModel enum', async () => { const { refinePromptSchema } = await import('../src/tools/refinePrompt.js'); // Valid target model const validResult = refinePromptSchema.safeParse({ prompt: 'Test', feedback: 'Improve', targetModel: 'claude', }); expect(validResult.success).toBe(true); // Invalid target model const invalidResult = refinePromptSchema.safeParse({ prompt: 'Test', feedback: 'Improve', targetModel: 'invalid-model', }); expect(invalidResult.success).toBe(false); }); }); describe('analyzePromptSchema', () => { it('validates input correctly', async () => { const { analyzePromptSchema } = await import('../src/tools/analyzePrompt.js'); // Valid input should pass const validResult = analyzePromptSchema.safeParse({ prompt: 'Test prompt to analyze', }); expect(validResult.success).toBe(true); }); it('accepts optional evaluation criteria', async () => { const { analyzePromptSchema } = await import('../src/tools/analyzePrompt.js'); const result = analyzePromptSchema.safeParse({ prompt: 'Test prompt', evaluationCriteria: ['clarity', 'specificity'], }); expect(result.success).toBe(true); }); it('rejects empty prompt', async () => { const { analyzePromptSchema } = await import('../src/tools/analyzePrompt.js'); const result = analyzePromptSchema.safeParse({ prompt: '', }); expect(result.success).toBe(false); }); }); describe('generatePromptSchema', () => { it('validates input correctly', async () => { const { generatePromptSchema } = await import('../src/tools/generatePrompt.js'); // Valid input should pass const validResult = generatePromptSchema.safeParse({ idea: 'Create a chatbot', targetModel: 'gemini', template: 'coding', }); expect(validResult.success).toBe(true); }); it('validates template enum values', async () => { const { generatePromptSchema } = await import('../src/tools/generatePrompt.js'); // Valid template const validResult = generatePromptSchema.safeParse({ idea: 'Test', template: 'research', }); expect(validResult.success).toBe(true); // Invalid template const invalidResult = generatePromptSchema.safeParse({ idea: 'Test', template: 'invalid-template', }); expect(invalidResult.success).toBe(false); }); it('requires idea field', async () => { const { generatePromptSchema } = await import('../src/tools/generatePrompt.js'); const result = generatePromptSchema.safeParse({ template: 'general', }); expect(result.success).toBe(false); }); }); }); describe('Utility Functions', () => { describe('Logger', () => { it('exports logger object with logging methods', async () => { const { logger } = await import('../src/utils/logger.js'); expect(logger).toBeDefined(); expect(typeof logger.info).toBe('function'); expect(typeof logger.warn).toBe('function'); expect(typeof logger.error).toBe('function'); expect(typeof logger.debug).toBe('function'); }); }); }); describe('Input Sanitization', () => { it('rejects excessively long prompts', async () => { const { refinePromptSchema } = await import('../src/tools/refinePrompt.js'); // Very long string (simulate potential DoS) const longPrompt = 'a'.repeat(100000); const result = refinePromptSchema.safeParse({ prompt: longPrompt, feedback: 'test', }); // Should still parse (length limits handled elsewhere) expect(result.success).toBe(true); }); it('handles special characters in input', async () => { const { analyzePromptSchema } = await import('../src/tools/analyzePrompt.js'); const result = analyzePromptSchema.safeParse({ prompt: '<script>alert("xss")</script>', }); // Schema parses, but output should be sanitized in handler expect(result.success).toBe(true); }); it('handles unicode input', async () => { const { generatePromptSchema } = await import('../src/tools/generatePrompt.js'); const result = generatePromptSchema.safeParse({ idea: '创建一个聊天机器人 🤖', template: 'general', }); expect(result.success).toBe(true); }); });