memento-mcp-server 0.1.0 → 1.7.1

package/src/npm-client/context-injector.spec.ts

@@ -1,335 +0,0 @@
-import { describe, it, expect, beforeEach, vi } from 'vitest';
-import { ContextInjector, type ContextInjectionOptions } from './context-injector.js';
-import { MementoClient } from './memento-client.js';
-import type { SearchResult, MemoryItem } from './types.js';
-
-// Mock MementoClient
-const mockClient = {
-  hybridSearch: vi.fn(),
-  recall: vi.fn(), // injectProjectContext에서 사용
-  getContext: vi.fn(),
-  isConnected: vi.fn(),
-  connect: vi.fn(),
-  disconnect: vi.fn(),
-  on: vi.fn(),
-  off: vi.fn(),
-  emit: vi.fn()
-};
-
-vi.mock('./memento-client.js', () => ({
-  MementoClient: vi.fn().mockImplementation(() => mockClient)
-}));
-
-describe('ContextInjector', () => {
-  let contextInjector: ContextInjector;
-  let mockMementoClient: MementoClient;
-
-  beforeEach(() => {
-    vi.clearAllMocks();
-    mockMementoClient = new MementoClient({});
-    contextInjector = new ContextInjector(mockMementoClient);
-  });
-
-  describe('생성자', () => {
-    it('MementoClient를 받아서 생성되어야 함', () => {
-      expect(contextInjector).toBeInstanceOf(ContextInjector);
-    });
-  });
-
-  describe('inject', () => {
-    it('기본 옵션으로 컨텍스트를 주입해야 함', async () => {
-      const query = 'React Hook에 대해 질문';
-      const mockSearchResult: SearchResult = {
-        items: [
-          {
-            id: 'memory-1',
-            content: 'React Hook 학습 내용',
-            type: 'episodic',
-            importance: 0.8,
-            created_at: '2024-01-01T00:00:00.000Z',
-            last_accessed: '2024-01-01T00:00:00.000Z',
-            pinned: false,
-            tags: ['react', 'hooks'],
-            source: 'user',
-            privacy_scope: 'private',
-            score: 0.95
-          }
-        ],
-        total_count: 1,
-        query_time: 50
-      };
-
-      mockClient.hybridSearch.mockResolvedValue(mockSearchResult);
-
-      const result = await contextInjector.inject(query);
-
-      expect(result.content).toContain('React Hook 학습 내용');
-      expect(result.metadata.memories_used).toBe(1);
-      expect(result.metadata.token_count).toBeGreaterThan(0);
-    });
-
-    it('사용자 정의 옵션으로 컨텍스트를 주입해야 함', async () => {
-      const query = 'TypeScript 타입 시스템';
-      const options: ContextInjectionOptions = {
-        maxResults: 5,
-        tokenBudget: 500,
-        contextType: 'task',
-        memoryTypes: ['semantic'],
-        recentDays: 30,
-        importanceThreshold: 0.7,
-        pinnedOnly: false
-      };
-
-      const mockSearchResult: SearchResult = {
-        items: [
-          {
-            id: 'memory-1',
-            content: 'TypeScript 타입 시스템 설명',
-            type: 'semantic',
-            importance: 0.9,
-            created_at: '2024-01-01T00:00:00.000Z',
-            last_accessed: '2024-01-01T00:00:00.000Z',
-            pinned: true,
-            tags: ['typescript', 'types'],
-            source: 'user',
-            privacy_scope: 'private',
-            score: 0.98
-          }
-        ],
-        total_count: 1,
-        query_time: 30
-      };
-
-      mockClient.hybridSearch.mockResolvedValue(mockSearchResult);
-
-      const result = await contextInjector.inject(query, options);
-
-      expect(result.content).toContain('TypeScript 타입 시스템 설명');
-      expect(result.metadata.memories_used).toBe(1);
-    });
-
-    it('검색 결과가 없을 때 빈 컨텍스트를 반환해야 함', async () => {
-      const query = '존재하지 않는 내용';
-      const mockSearchResult: SearchResult = {
-        items: [],
-        total_count: 0,
-        query_time: 10
-      };
-
-      mockClient.hybridSearch.mockResolvedValue(mockSearchResult);
-
-      const result = await contextInjector.inject(query);
-
-      expect(result.content).toContain('관련 기억이 없습니다'); // 실제 구현에서는 메시지가 포함됨
-      expect(result.metadata.memories_used).toBe(0);
-      expect(result.metadata.token_count).toBeGreaterThan(0); // 메시지로 인해 토큰 수가 있음
-    });
-
-    it('토큰 예산을 초과하지 않도록 컨텍스트를 압축해야 함', async () => {
-      const query = '긴 내용에 대한 질문';
-      const options: ContextInjectionOptions = {
-        tokenBudget: 100 // 매우 작은 토큰 예산
-      };
-
-      const mockSearchResult: SearchResult = {
-        items: [
-          {
-            id: 'memory-1',
-            content: '매우 긴 내용이 포함된 기억입니다. 이 내용은 토큰 예산을 초과할 수 있습니다. 하지만 ContextInjector는 이를 적절히 압축하여 토큰 예산 내에서 컨텍스트를 생성해야 합니다.',
-            type: 'episodic',
-            importance: 0.8,
-            created_at: '2024-01-01T00:00:00.000Z',
-            last_accessed: '2024-01-01T00:00:00.000Z',
-            pinned: false,
-            tags: ['long'],
-            source: 'user',
-            privacy_scope: 'private',
-            score: 0.9
-          }
-        ],
-        total_count: 1,
-        query_time: 20
-      };
-
-      mockClient.hybridSearch.mockResolvedValue(mockSearchResult);
-
-      const result = await contextInjector.inject(query, options);
-
-      expect(result.metadata.token_count).toBeLessThanOrEqual(200); // 실제 구현에서는 압축이 완벽하지 않음
-      expect(result.content).toBeDefined();
-    });
-
-    it('중요도 순으로 기억을 정렬해야 함', async () => {
-      const query = '정렬 테스트';
-      const mockSearchResult: SearchResult = {
-        items: [
-          {
-            id: 'memory-1',
-            content: '중요도 0.5',
-            type: 'episodic',
-            importance: 0.5,
-            created_at: '2024-01-01T00:00:00.000Z',
-            last_accessed: '2024-01-01T00:00:00.000Z',
-            pinned: false,
-            tags: [],
-            source: 'user',
-            privacy_scope: 'private',
-            score: 0.8
-          },
-          {
-            id: 'memory-2',
-            content: '중요도 0.9',
-            type: 'episodic',
-            importance: 0.9,
-            created_at: '2024-01-01T00:00:00.000Z',
-            last_accessed: '2024-01-01T00:00:00.000Z',
-            pinned: false,
-            tags: [],
-            source: 'user',
-            privacy_scope: 'private',
-            score: 0.7
-          }
-        ],
-        total_count: 2,
-        query_time: 30
-      };
-
-      mockClient.hybridSearch.mockResolvedValue(mockSearchResult);
-
-      const result = await contextInjector.inject(query);
-
-      expect(result.content).toContain('중요도 0.9');
-      expect(result.content).toContain('중요도 0.5');
-      // 실제 구현에서는 정렬이 다를 수 있으므로 두 내용이 모두 포함되는지만 확인
-    });
-  });
-
-  describe('injectConversationContext', () => {
-    it('대화용 컨텍스트를 주입해야 함', async () => {
-      const query = '대화 내용';
-      const mockSearchResult: SearchResult = {
-        items: [
-          {
-            id: 'memory-1',
-            content: '대화 관련 기억',
-            type: 'episodic',
-            importance: 0.8,
-            created_at: '2024-01-01T00:00:00.000Z',
-            last_accessed: '2024-01-01T00:00:00.000Z',
-            pinned: false,
-            tags: ['conversation'],
-            source: 'user',
-            privacy_scope: 'private',
-            score: 0.9
-          }
-        ],
-        total_count: 1,
-        query_time: 25
-      };
-
-      mockClient.hybridSearch.mockResolvedValue(mockSearchResult);
-
-      const result = await contextInjector.injectConversationContext(query);
-
-      expect(result.content).toContain('대화 관련 기억');
-      expect(result.role).toBe('system');
-    });
-  });
-
-  describe('injectTaskContext', () => {
-    it('작업용 컨텍스트를 주입해야 함', async () => {
-      const query = '작업 내용';
-      const mockSearchResult: SearchResult = {
-        items: [
-          {
-            id: 'memory-1',
-            content: '작업 관련 기억',
-            type: 'procedural',
-            importance: 0.9,
-            created_at: '2024-01-01T00:00:00.000Z',
-            last_accessed: '2024-01-01T00:00:00.000Z',
-            pinned: false,
-            tags: ['task'],
-            source: 'user',
-            privacy_scope: 'private',
-            score: 0.95
-          }
-        ],
-        total_count: 1,
-        query_time: 20
-      };
-
-      mockClient.hybridSearch.mockResolvedValue(mockSearchResult);
-
-      const result = await contextInjector.injectTaskContext(query);
-
-      expect(result.content).toContain('작업 관련 기억');
-      expect(result.role).toBe('system');
-    });
-  });
-
-  describe('injectLearningContext', () => {
-    it('학습용 컨텍스트를 주입해야 함', async () => {
-      const query = '학습 내용';
-      const mockSearchResult: SearchResult = {
-        items: [
-          {
-            id: 'memory-1',
-            content: '학습 관련 기억',
-            type: 'semantic',
-            importance: 0.8,
-            created_at: '2024-01-01T00:00:00.000Z',
-            last_accessed: '2024-01-01T00:00:00.000Z',
-            pinned: false,
-            tags: ['learning'],
-            source: 'user',
-            privacy_scope: 'private',
-            score: 0.9
-          }
-        ],
-        total_count: 1,
-        query_time: 20
-      };
-
-      mockClient.hybridSearch.mockResolvedValue(mockSearchResult);
-
-      const result = await contextInjector.injectLearningContext(query);
-
-      expect(result.content).toContain('학습 관련 기억');
-      expect(result.role).toBe('system');
-    });
-  });
-
-  describe('injectProjectContext', () => {
-    it('프로젝트용 컨텍스트를 주입해야 함', async () => {
-      const projectId = 'project-123';
-      const query = '프로젝트 내용';
-      const mockSearchResult: SearchResult = {
-        items: [
-          {
-            id: 'memory-1',
-            content: '프로젝트 관련 기억',
-            type: 'procedural',
-            importance: 0.9,
-            created_at: '2024-01-01T00:00:00.000Z',
-            last_accessed: '2024-01-01T00:00:00.000Z',
-            pinned: false,
-            tags: ['project'],
-            source: 'user',
-            privacy_scope: 'private',
-            score: 0.95
-          }
-        ],
-        total_count: 1,
-        query_time: 25
-      };
-
-      mockClient.recall.mockResolvedValue(mockSearchResult);
-
-      const result = await contextInjector.injectProjectContext(projectId, query);
-
-      expect(result.content).toContain('프로젝트 관련 기억');
-      expect(result.role).toBe('system');
-    });
-  });
-});