Memory MCP Server

/** * 유닛 테스트를 위한 공통 헬퍼 함수들 */ import * as fs from 'fs'; import * as path from 'path'; import * as os from 'os'; import type { ToolExecutionContext } from '../../src/tools/types'; import type { CreateNoteInput } from '../../src/tools/schemas'; /** * 테스트용 임시 디렉토리 생성 */ export function createTempDir(prefix: string = 'memory-mcp-test-'): string { return fs.mkdtempSync(path.join(os.tmpdir(), prefix)); } /** * 테스트용 ToolExecutionContext 생성 */ export function createTestContext( overrides: Partial<ToolExecutionContext> = {} ): ToolExecutionContext { const tempDir = createTempDir(); const vaultPath = overrides.vaultPath || path.join(tempDir, 'vault'); const indexPath = overrides.indexPath || path.join(tempDir, 'index.db'); // Ensure vault directory exists if (!fs.existsSync(vaultPath)) { fs.mkdirSync(vaultPath, { recursive: true }); } return { vaultPath, indexPath, logger: { debug: jest.fn(), info: jest.fn(), warn: jest.fn(), error: jest.fn(), }, policy: { maxRetries: 1, timeoutMs: 5000, }, mode: 'dev', ...overrides, }; } /** * 테스트 정리 헬퍼 */ export function cleanupTestContext(context: ToolExecutionContext): void { try { // IndexRecoveryQueue 정리 if (context._recoveryQueue) { context._recoveryQueue.cleanup(); delete context._recoveryQueue; } // SearchEngine 인스턴스 정리 (SQLite 연결 종료) if (context._searchEngineInstance) { context._searchEngineInstance.close(); delete context._searchEngineInstance; } // 임시 디렉토리 정리 const tempDir = path.dirname(context.vaultPath); if (fs.existsSync(tempDir)) { fs.rmSync(tempDir, { recursive: true, force: true }); } } catch (error) { // Ignore cleanup errors } } /** * Mock 노트 데이터 생성 (타입 안전) * @param overrides - 기본값을 덮어쓸 필드들 * @returns CreateNoteInput 타입에 맞는 노트 데이터 */ export function createMockNote( overrides: Partial<CreateNoteInput> = {} ): CreateNoteInput { return { title: 'Test Note', content: 'This is test content', category: 'Resources' as const, tags: ['test'], ...overrides, }; }