memento-mcp-server 0.1.0

package/src/test/test-batch-scheduler.ts

@@ -0,0 +1,182 @@
+import { describe, it, expect, beforeEach, afterEach } from 'vitest';
+import { getBatchScheduler, BatchScheduler } from '../services/batch-scheduler.js';
+import { getPerformanceMonitor } from '../services/performance-monitor.js';
+import { ForgettingPolicyService } from '../services/forgetting-policy-service.js';
+import Database from 'better-sqlite3';
+import { DatabaseUtils } from '../utils/database.js';
+
+// Mock dependencies
+import { vi } from 'vitest';
+vi.mock('../services/performance-monitor.js');
+vi.mock('../services/forgetting-policy-service.js');
+
+describe('BatchScheduler', () => {
+  let db: Database.Database;
+  let batchScheduler: BatchScheduler;
+  let mockPerformanceMonitor: any;
+  let mockForgettingService: any;
+
+  beforeEach(() => {
+    // Create in-memory database for testing
+    db = new Database(':memory:');
+    DatabaseUtils.initializeDatabase(db);
+
+    // Mock performance monitor
+    mockPerformanceMonitor = {
+      initialize: vi.fn(),
+      collectMetrics: vi.fn().mockResolvedValue({
+        timestamp: new Date().toISOString(),
+        cpu: { user: 0, system: 0, idle: 0 },
+        memory: { rss: 0, heapTotal: 0, heapUsed: 0, external: 0, arrayBuffers: 0 },
+        database: { totalMemories: 0, size: 0, queryTime: 0 },
+        uptime: 0
+      }),
+      getActiveAlerts: vi.fn().mockReturnValue([])
+    };
+
+    // Mock forgetting service
+    mockForgettingService = {
+      executeMemoryCleanup: vi.fn().mockResolvedValue({
+        softDeleted: [],
+        hardDeleted: [],
+        reviewed: []
+      })
+    };
+
+    // Setup mocks
+    vi.mocked(getPerformanceMonitor).mockReturnValue(mockPerformanceMonitor);
+    vi.mocked(ForgettingPolicyService).mockImplementation(() => mockForgettingService);
+
+    batchScheduler = getBatchScheduler({
+      cleanupInterval: 1000, // 1초 for testing
+      monitoringInterval: 500, // 0.5초 for testing
+      enableLogging: false
+    });
+  });
+
+  afterEach(() => {
+    if (db) {
+      db.close();
+    }
+    batchScheduler.stop();
+  });
+
+  describe('초기화', () => {
+    it('should initialize successfully', async () => {
+      await expect(batchScheduler.start(db)).resolves.not.toThrow();
+      expect(mockPerformanceMonitor.initialize).toHaveBeenCalledWith(db);
+    });
+
+    it('should not start if already running', async () => {
+      await batchScheduler.start(db);
+      await expect(batchScheduler.start(db)).rejects.toThrow('BatchScheduler is already running');
+    });
+  });
+
+  describe('상태 관리', () => {
+    it('should return correct status when not running', () => {
+      const status = batchScheduler.getStatus();
+      expect(status.isRunning).toBe(false);
+      expect(status.activeJobs).toEqual([]);
+    });
+
+    it('should return correct status when running', async () => {
+      await batchScheduler.start(db);
+      const status = batchScheduler.getStatus();
+      expect(status.isRunning).toBe(true);
+      expect(status.activeJobs).toContain('cleanup');
+      expect(status.activeJobs).toContain('monitoring');
+    });
+  });
+
+  describe('작업 실행', () => {
+    beforeEach(async () => {
+      await batchScheduler.start(db);
+    });
+
+    it('should run cleanup job successfully', async () => {
+      const result = await batchScheduler.runJob('cleanup');
+      expect(result.jobType).toBe('cleanup');
+      expect(result.success).toBe(true);
+      expect(mockForgettingService.executeMemoryCleanup).toHaveBeenCalledWith(db);
+    });
+
+    it('should run monitoring job successfully', async () => {
+      const result = await batchScheduler.runJob('monitoring');
+      expect(result.jobType).toBe('monitoring');
+      expect(result.success).toBe(true);
+      expect(mockPerformanceMonitor.collectMetrics).toHaveBeenCalled();
+    });
+
+    it('should handle unknown job type', async () => {
+      await expect(batchScheduler.runJob('unknown' as any)).rejects.toThrow('Unknown job type: unknown');
+    });
+  });
+
+  describe('메모리 정리', () => {
+    beforeEach(async () => {
+      await batchScheduler.start(db);
+    });
+
+    it('should execute memory cleanup with correct parameters', async () => {
+      const result = await batchScheduler.runJob('cleanup');
+      expect(mockForgettingService.executeMemoryCleanup).toHaveBeenCalledWith(db);
+      expect(result.success).toBe(true);
+    });
+
+    it('should handle cleanup errors gracefully', async () => {
+      mockForgettingService.executeMemoryCleanup.mockRejectedValueOnce(new Error('Cleanup failed'));
+      const result = await batchScheduler.runJob('cleanup');
+      expect(result.success).toBe(false);
+      expect(result.errors).toContain('Cleanup failed');
+    });
+  });
+
+  describe('성능 모니터링', () => {
+    beforeEach(async () => {
+      await batchScheduler.start(db);
+    });
+
+    it('should collect performance metrics', async () => {
+      const result = await batchScheduler.runJob('monitoring');
+      expect(mockPerformanceMonitor.collectMetrics).toHaveBeenCalled();
+      expect(result.success).toBe(true);
+    });
+
+    it('should handle monitoring errors gracefully', async () => {
+      mockPerformanceMonitor.collectMetrics.mockRejectedValueOnce(new Error('Monitoring failed'));
+      const result = await batchScheduler.runJob('monitoring');
+      expect(result.success).toBe(false);
+      expect(result.errors).toContain('Monitoring failed');
+    });
+  });
+
+  describe('중지', () => {
+    it('should stop all jobs', async () => {
+      await batchScheduler.start(db);
+      batchScheduler.stop();
+      
+      const status = batchScheduler.getStatus();
+      expect(status.isRunning).toBe(false);
+      expect(status.activeJobs).toEqual([]);
+    });
+  });
+
+  describe('데이터베이스 통계', () => {
+    beforeEach(async () => {
+      await batchScheduler.start(db);
+      
+      // Add some test data
+      DatabaseUtils.runQuery(db, `
+        INSERT INTO memory_item (id, type, content, importance, created_at)
+        VALUES ('test1', 'episodic', 'Test memory 1', 0.5, datetime('now'))
+      `);
+    });
+
+    it('should collect database statistics', async () => {
+      const result = await batchScheduler.runJob('monitoring');
+      expect(result.success).toBe(true);
+      expect(result.details?.stats).toBeDefined();
+    });
+  });
+});

package/src/test/test-client.ts

@@ -0,0 +1,97 @@
+/**
+ * Memento MCP Client 테스트 스크립트
+ */
+
+import { createMementoClient } from '../client/index.js';
+
+async function testMementoClient() {
+  console.log('🧪 Memento MCP Client 테스트 시작');
+  
+  const client = createMementoClient();
+  
+  try {
+    // 1. 서버 연결
+    console.log('\n1️⃣ 서버 연결 중...');
+    await client.connect();
+    
+    // 2. 기억 저장 테스트
+    console.log('\n2️⃣ 기억 저장 테스트');
+    const memoryId1 = await client.remember({
+      content: "사용자가 React Hook에 대해 질문했고, useState와 useEffect의 차이점을 설명했다.",
+      type: 'episodic',
+      tags: ['react', 'hooks', 'javascript'],
+      importance: 0.8,
+      source: 'test-client'
+    });
+    console.log(`✅ 기억 저장 완료: ${memoryId1}`);
+    
+    const memoryId2 = await client.remember({
+      content: "TypeScript의 타입 시스템에 대해 설명했다. 인터페이스와 타입 별칭의 차이점을 다뤘다.",
+      type: 'semantic',
+      tags: ['typescript', 'types', 'programming'],
+      importance: 0.9,
+      source: 'test-client'
+    });
+    console.log(`✅ 기억 저장 완료: ${memoryId2}`);
+    
+    // 3. 기억 검색 테스트
+    console.log('\n3️⃣ 기억 검색 테스트');
+    const searchResults = await client.recall({
+      query: "React Hook",
+      limit: 5
+    });
+    console.log(`✅ 검색 결과 (${searchResults.length}개):`);
+    searchResults.forEach((result, index) => {
+      console.log(`  ${index + 1}. [${result.type}] ${result.content.substring(0, 50)}... (점수: ${result.score || 'N/A'})`);
+    });
+    
+    // 4. 기억 고정 테스트
+    console.log('\n4️⃣ 기억 고정 테스트');
+    const pinResult = await client.pin({ id: memoryId1 });
+    console.log(`✅ 기억 고정: ${pinResult}`);
+    
+    // 5. 고정된 기억 검색 테스트
+    console.log('\n5️⃣ 고정된 기억 검색 테스트');
+    const pinnedResults = await client.recall({
+      query: "React",
+      filters: { pinned: true },
+      limit: 5
+    });
+    console.log(`✅ 고정된 기억 검색 결과 (${pinnedResults.length}개):`);
+    pinnedResults.forEach((result, index) => {
+      console.log(`  ${index + 1}. [${result.type}] ${result.content.substring(0, 50)}... (고정됨: ${result.pinned})`);
+    });
+    
+    // 6. 기억 고정 해제 테스트
+    console.log('\n6️⃣ 기억 고정 해제 테스트');
+    const unpinResult = await client.unpin({ id: memoryId1 });
+    console.log(`✅ 기억 고정 해제: ${unpinResult}`);
+    
+    // 7. 기억 삭제 테스트 (소프트 삭제)
+    console.log('\n7️⃣ 기억 삭제 테스트 (소프트 삭제)');
+    const forgetResult = await client.forget({ id: memoryId2, hard: false });
+    console.log(`✅ 기억 삭제: ${forgetResult}`);
+    
+    console.log('\n🎉 모든 테스트가 성공적으로 완료되었습니다!');
+    
+  } catch (error) {
+    console.error('❌ 테스트 실패:', error);
+  } finally {
+    // 8. 연결 해제
+    console.log('\n8️⃣ 연결 해제');
+    await client.disconnect();
+  }
+}
+
+// 테스트 실행
+if (process.argv[1] && process.argv[1].endsWith('test-client.ts')) {
+  testMementoClient()
+    .then(() => {
+      console.log('✅ 테스트 완료');
+      process.exit(0);
+    })
+    .catch((error) => {
+      console.error('❌ 테스트 실패:', error);
+      process.exit(1);
+    });
+}

package/src/test/test-embedding.ts

@@ -0,0 +1,153 @@
+/**
+ * 임베딩 기능 테스트
+ * OpenAI API 연동 및 벡터 검색 테스트
+ */
+
+import { createMementoClient } from '../client/index.js';
+
+async function testEmbeddingFunctionality() {
+  console.log('🧠 임베딩 기능 테스트 시작');
+  
+  const client = createMementoClient();
+  
+  try {
+    // 1. 서버 연결
+    console.log('\n1️⃣ 서버 연결 중...');
+    await client.connect();
+    
+    // 2. 다양한 기억 저장 (임베딩 생성 테스트)
+    console.log('\n2️⃣ 임베딩이 포함된 기억 저장');
+    const memories = [
+      {
+        content: "사용자가 React의 useState Hook에 대해 질문했습니다. 상태 관리를 위한 기본 Hook으로, 함수형 컴포넌트에서 상태를 선언하고 업데이트할 수 있습니다.",
+        type: 'episodic' as const,
+        tags: ['react', 'hooks', 'useState', 'javascript'],
+        importance: 0.9
+      },
+      {
+        content: "TypeScript의 인터페이스와 타입 별칭의 차이점을 설명했습니다. 인터페이스는 확장 가능하고, 타입 별칭은 유니온 타입을 표현할 수 있습니다.",
+        type: 'semantic' as const,
+        tags: ['typescript', 'types', 'interface', 'type-alias'],
+        importance: 0.8
+      },
+      {
+        content: "데이터베이스 인덱싱에 대한 질문을 받았습니다. B-tree 인덱스의 작동 원리와 쿼리 성능에 미치는 영향을 설명했습니다.",
+        type: 'episodic' as const,
+        tags: ['database', 'indexing', 'b-tree', 'performance'],
+        importance: 0.7
+      },
+      {
+        content: "MCP(Model Context Protocol)에 대해 학습했습니다. AI 에이전트와 도구 간의 표준화된 통신 프로토콜로, 확장 가능한 아키텍처를 제공합니다.",
+        type: 'semantic' as const,
+        tags: ['mcp', 'protocol', 'ai', 'architecture'],
+        importance: 0.85
+      },
+      {
+        content: "사용자가 Docker 컨테이너화에 대해 질문했습니다. 이미지 빌드, 레이어 캐싱, 멀티스테이지 빌드 등의 개념을 설명했습니다.",
+        type: 'episodic' as const,
+        tags: ['docker', 'containerization', 'devops', 'deployment'],
+        importance: 0.75
+      }
+    ];
+    
+    const memoryIds: string[] = [];
+    for (const memory of memories) {
+      const id = await client.remember(memory);
+      memoryIds.push(id);
+      console.log(`✅ 저장됨: ${id.substring(0, 20)}... - ${memory.content.substring(0, 50)}...`);
+      
+      // 임베딩 생성 시간을 위해 잠시 대기
+      await new Promise(resolve => setTimeout(resolve, 1000));
+    }
+    
+    // 3. 하이브리드 검색 테스트
+    console.log('\n3️⃣ 하이브리드 검색 테스트 (텍스트 + 벡터)');
+    
+    const searchQueries = [
+      { query: "React Hook", description: "React Hook 관련 검색" },
+      { query: "타입 시스템", description: "타입 시스템 관련 검색" },
+      { query: "데이터베이스 성능", description: "데이터베이스 성능 관련 검색" },
+      { query: "AI 프로토콜", description: "AI 프로토콜 관련 검색" },
+      { query: "컨테이너 배포", description: "컨테이너 배포 관련 검색" },
+      { query: "상태 관리", description: "상태 관리 관련 검색" },
+      { query: "확장 가능한 아키텍처", description: "아키텍처 관련 검색" }
+    ];
+    
+    for (const { query, description } of searchQueries) {
+      console.log(`\n🔍 ${description}: "${query}"`);
+      try {
+        const results = await client.recall({ query, limit: 3 });
+        console.log(`   결과: ${results.length}개`);
+        results.forEach((result, index) => {
+          console.log(`   ${index + 1}. [${result.type}] ${result.content.substring(0, 60)}...`);
+          const score = (result as any).finalScore || (result as any).score || 'N/A';
+          console.log(`       점수: ${typeof score === 'number' ? score.toFixed(3) : score}`);
+          if ((result as any).textScore !== undefined && (result as any).vectorScore !== undefined) {
+            console.log(`       텍스트: ${(result as any).textScore.toFixed(3)}, 벡터: ${(result as any).vectorScore.toFixed(3)}`);
+          }
+        });
+      } catch (error) {
+        console.error(`   ❌ 검색 실패: ${error}`);
+      }
+    }
+    
+    // 4. 의미적 유사성 검색 테스트
+    console.log('\n4️⃣ 의미적 유사성 검색 테스트');
+    
+    const semanticQueries = [
+      { query: "프론트엔드 개발", description: "프론트엔드 관련 검색" },
+      { query: "타입 안전성", description: "타입 안전성 관련 검색" },
+      { query: "쿼리 최적화", description: "쿼리 최적화 관련 검색" },
+      { query: "마이크로서비스", description: "마이크로서비스 관련 검색" },
+      { query: "개발 도구", description: "개발 도구 관련 검색" }
+    ];
+    
+    for (const { query, description } of semanticQueries) {
+      console.log(`\n🔍 ${description}: "${query}"`);
+      try {
+        const results = await client.recall({ query, limit: 2 });
+        console.log(`   결과: ${results.length}개`);
+        results.forEach((result, index) => {
+          console.log(`   ${index + 1}. [${result.type}] ${result.content.substring(0, 60)}...`);
+          const score = (result as any).finalScore || (result as any).score || 'N/A';
+          console.log(`       점수: ${typeof score === 'number' ? score.toFixed(3) : score}`);
+        });
+      } catch (error) {
+        console.error(`   ❌ 검색 실패: ${error}`);
+      }
+    }
+    
+    // 5. 임베딩 통계 확인
+    console.log('\n5️⃣ 임베딩 통계 확인');
+    try {
+      // 간단한 검색으로 통계 정보 확인
+      const statsResult = await client.recall({ query: "test", limit: 1 });
+      if (statsResult.length > 0 && (statsResult[0] as any).search_type === 'hybrid') {
+        console.log('   ✅ 하이브리드 검색 활성화됨');
+        console.log(`   📊 벡터 검색 사용 가능: ${(statsResult[0] as any).vector_search_available || false}`);
+      }
+    } catch (error) {
+      console.error(`   ❌ 통계 확인 실패: ${error}`);
+    }
+    
+    console.log('\n🎉 임베딩 기능 테스트 완료!');
+    
+  } catch (error) {
+    console.error('❌ 테스트 실패:', error);
+  } finally {
+    await client.disconnect();
+  }
+}
+
+// 테스트 실행
+if (process.argv[1] && process.argv[1].endsWith('test-embedding.ts')) {
+  testEmbeddingFunctionality()
+    .then(() => {
+      console.log('✅ 임베딩 테스트 완료');
+      process.exit(0);
+    })
+    .catch((error) => {
+      console.error('❌ 임베딩 테스트 실패:', error);
+      process.exit(1);
+    });
+}