KuzuMem-MCP

import { detectHandler } from '../../../mcp/services/handlers/unified/detect-handler'; import { ToolHandlerContext } from '../../../mcp/types/sdk-custom'; import { GraphAnalysisService } from '../../../services/domain/graph-analysis.service'; import { MemoryService } from '../../../services/memory.service'; // Define discriminated union type for detect handler results type DetectResult = | { type: 'strongly-connected'; status: 'complete'; projectedGraphName: string; components: Array<{ componentId: number; nodes: string[] }>; totalComponents: number; message?: string; } | { type: 'weakly-connected'; status: 'complete'; projectedGraphName: string; components: Array<{ componentId: number; nodes: string[] }>; totalComponents: number; message?: string; } | { type: 'strongly-connected' | 'weakly-connected'; status: 'error'; message: string; components: []; }; describe('Detect Tool Tests', () => { let mockMemoryService: jest.Mocked<MemoryService>; let mockGraphAnalysisService: jest.Mocked<GraphAnalysisService>; let mockContext: jest.Mocked<ToolHandlerContext>; beforeEach(() => { mockGraphAnalysisService = { getStronglyConnectedComponents: jest.fn(), getWeaklyConnectedComponents: jest.fn(), shortestPath: jest.fn(), } as any; mockMemoryService = { graphAnalysis: mockGraphAnalysisService, services: { graphAnalysis: mockGraphAnalysisService, }, } as any; // Mock context with session mockContext = { logger: { info: jest.fn(), error: jest.fn(), warn: jest.fn(), debug: jest.fn(), }, session: { clientProjectRoot: '/test/project', }, sendProgress: jest.fn(), request: {} as any, meta: {} as any, signal: {} as any, } as any; }); describe('Strongly Connected Components Detection', () => { it('should detect strongly connected components', async () => { const mockResult = { type: 'strongly-connected' as const, status: 'complete', projectedGraphName: 'circular-deps', components: [ { componentId: 0, nodes: ['comp-1', 'comp-2', 'comp-3'] }, { componentId: 1, nodes: ['comp-4', 'comp-5'] }, ], totalComponents: 2, message: 'Found circular dependencies', }; mockGraphAnalysisService.getStronglyConnectedComponents.mockResolvedValue(mockResult); const result = (await detectHandler( { type: 'strongly-connected', repository: 'test-repo', branch: 'main', projectedGraphName: 'circular-deps', nodeTableNames: ['Component'], relationshipTableNames: ['DEPENDS_ON'], }, mockContext, mockMemoryService, )) as DetectResult; if (result.type === 'strongly-connected' && result.status === 'complete') { expect(result.type).toBe('strongly-connected'); expect(result.status).toBe('complete'); expect(result.components).toHaveLength(2); expect(result.components[0]).toEqual({ componentId: 0, nodes: ['comp-1', 'comp-2', 'comp-3'], }); expect(result.totalComponents).toBe(2); } else { fail('Expected result to be strongly-connected with complete status'); } expect(mockGraphAnalysisService.getStronglyConnectedComponents).toHaveBeenCalledWith( mockContext, '/test/project', { type: 'strongly-connected', repository: 'test-repo', branch: 'main', projectedGraphName: 'circular-deps', nodeTableNames: ['Component'], relationshipTableNames: ['DEPENDS_ON'], }, ); }); it('should handle empty results', async () => { mockGraphAnalysisService.getStronglyConnectedComponents.mockResolvedValue({ type: 'strongly-connected' as const, status: 'complete', projectedGraphName: 'deps', components: [], totalComponents: 0, }); const result = (await detectHandler( { type: 'strongly-connected', repository: 'test-repo', projectedGraphName: 'deps', nodeTableNames: ['Component'], relationshipTableNames: ['DEPENDS_ON'], }, mockContext, mockMemoryService, )) as DetectResult; if (result.type === 'strongly-connected' && result.status === 'complete') { expect(result.components).toEqual([]); expect(result.totalComponents).toBe(0); } else { fail('Expected result to be strongly-connected with complete status'); } }); }); describe('Weakly Connected Components Detection', () => { it('should detect weakly connected components', async () => { const mockResult = { type: 'weakly-connected' as const, status: 'complete', projectedGraphName: 'isolated-systems', components: [ { componentId: 0, nodes: ['comp-1', 'comp-2'] }, { componentId: 1, nodes: ['comp-3'] }, { componentId: 2, nodes: ['comp-4', 'comp-5', 'comp-6'] }, ], totalComponents: 3, message: 'Found isolated subsystems', }; mockGraphAnalysisService.getWeaklyConnectedComponents.mockResolvedValue(mockResult); const result = (await detectHandler( { type: 'weakly-connected', repository: 'test-repo', projectedGraphName: 'isolated-systems', nodeTableNames: ['Component'], relationshipTableNames: ['DEPENDS_ON'], }, mockContext, mockMemoryService, )) as DetectResult; if (result.type === 'weakly-connected' && result.status === 'complete') { expect(result.type).toBe('weakly-connected'); expect(result.components).toHaveLength(3); expect(result.components[2]).toEqual({ componentId: 2, nodes: ['comp-4', 'comp-5', 'comp-6'], }); expect(result.totalComponents).toBe(3); } else { fail('Expected result to be weakly-connected with complete status'); } }); }); describe('Session Validation', () => { it('should throw error if no active session', async () => { const contextNoSession = { logger: { info: jest.fn(), error: jest.fn(), warn: jest.fn(), debug: jest.fn(), }, session: {}, sendProgress: jest.fn(), request: {} as any, meta: {} as any, signal: {} as any, } as any; await expect( detectHandler( { type: 'strongly-connected', repository: 'test-repo', projectedGraphName: 'deps', nodeTableNames: ['Component'], relationshipTableNames: ['DEPENDS_ON'], }, contextNoSession, mockMemoryService, ), ).rejects.toThrow('No active session for detect tool'); }); }); describe('Error Handling', () => { it('should handle and rethrow service errors', async () => { const error = new Error('Detection service error'); mockGraphAnalysisService.getStronglyConnectedComponents.mockRejectedValue(error); const result = await detectHandler( { type: 'strongly-connected', repository: 'test-repo', projectedGraphName: 'deps', nodeTableNames: ['Component'], relationshipTableNames: ['DEPENDS_ON'], }, mockContext, mockMemoryService, ); expect(result).toEqual({ type: 'strongly-connected', status: 'error', message: 'Detection service error', components: [], }); expect(mockContext.sendProgress).toHaveBeenCalledWith({ status: 'error', message: 'Failed to execute strongly-connected detection: Detection service error', percent: 100, isFinal: true, }); }); it('should throw error for unknown detection type', async () => { await expect( detectHandler( { type: 'unknown' as any, repository: 'test-repo', projectedGraphName: 'deps', nodeTableNames: ['Component'], relationshipTableNames: ['DEPENDS_ON'], }, mockContext, mockMemoryService, ), ).rejects.toThrow(); }); }); describe('Progress Reporting', () => { it('should report progress for strongly connected detection', async () => { mockGraphAnalysisService.getStronglyConnectedComponents.mockResolvedValue({ type: 'strongly-connected' as const, status: 'complete', projectedGraphName: 'deps', components: [], totalComponents: 0, }); await detectHandler( { type: 'strongly-connected', repository: 'test-repo', projectedGraphName: 'deps', nodeTableNames: ['Component'], relationshipTableNames: ['DEPENDS_ON'], }, mockContext, mockMemoryService, ); expect(mockContext.sendProgress).toHaveBeenCalledWith({ status: 'in_progress', message: 'Finding strongly connected components...', percent: 50, }); expect(mockContext.sendProgress).toHaveBeenCalledWith({ status: 'complete', message: 'Strongly connected components detection complete. Found 0 components', percent: 100, isFinal: true, }); }); it('should report progress for weakly connected detection', async () => { mockGraphAnalysisService.getWeaklyConnectedComponents.mockResolvedValue({ type: 'weakly-connected' as const, status: 'complete', projectedGraphName: 'deps', components: [], totalComponents: 0, }); await detectHandler( { type: 'weakly-connected', repository: 'test-repo', projectedGraphName: 'deps', nodeTableNames: ['Component'], relationshipTableNames: ['DEPENDS_ON'], }, mockContext, mockMemoryService, ); expect(mockContext.sendProgress).toHaveBeenCalledWith({ status: 'in_progress', message: 'Finding weakly connected components...', percent: 50, }); expect(mockContext.sendProgress).toHaveBeenCalledWith({ status: 'complete', message: 'Weakly connected components detection complete. Found 0 components', percent: 100, isFinal: true, }); }); }); });