Ambiance MCP Server

/** * Integration tests for lightweight tools * These tests verify the three core tools work correctly without mocking */ import { handleSemanticCompact, handleProjectHints, handleFileSummary } from '../localTools'; import * as fs from 'fs'; import { join } from 'path'; import { tmpdir } from 'os'; // Mock sqlite3 to avoid binding issues jest.mock('sqlite3', () => ({ Database: jest.fn().mockImplementation(() => ({ exec: jest.fn().mockImplementation((sql: any, callback: any) => callback(null)), prepare: jest.fn().mockReturnValue({ run: jest.fn().mockImplementation((params: any, callback: any) => callback(null)), all: jest.fn().mockImplementation((params: any, callback: any) => callback(null, [])), get: jest.fn().mockImplementation((params: any, callback: any) => callback(null, null)), finalize: jest.fn(), }), close: jest.fn().mockImplementation((callback: any) => callback(null)), })), Statement: jest.fn(), })); // Don't mock fs - this is an integration test that needs real file system access describe('Lightweight Tools Integration', () => { let testProjectPath: string; beforeAll(() => { // Create a temporary test project testProjectPath = join(tmpdir(), 'test-lightweight-tools'); if (fs.existsSync(testProjectPath)) { fs.rmSync(testProjectPath, { recursive: true, force: true }); } fs.mkdirSync(testProjectPath, { recursive: true }); fs.mkdirSync(join(testProjectPath, 'src'), { recursive: true }); // Create sample TypeScript files fs.writeFileSync( join(testProjectPath, 'src', 'index.ts'), ` export interface User { id: string; name: string; email: string; } export class UserService { private users: User[] = []; addUser(user: User): void { this.users.push(user); } getUser(id: string): User | undefined { return this.users.find(u => u.id === id); } } export const createUserService = (): UserService => { return new UserService(); }; ` ); fs.writeFileSync( join(testProjectPath, 'src', 'auth.ts'), ` export interface AuthToken { token: string; expiresAt: Date; } export class AuthService { authenticate(username: string, password: string): AuthToken | null { // Simple auth logic if (username && password) { return { token: 'sample-token', expiresAt: new Date(Date.now() + 3600000) }; } return null; } validateToken(token: string): boolean { return token === 'sample-token'; } } ` ); fs.writeFileSync( join(testProjectPath, 'package.json'), JSON.stringify( { name: 'test-project', version: '1.0.0', main: 'src/index.ts', }, null, 2 ) ); }); afterAll(() => { // Clean up test project if (fs.existsSync(testProjectPath)) { fs.rmSync(testProjectPath, { recursive: true, force: true }); } }); describe('local_context (handleSemanticCompact)', () => { it('should compress project context successfully', async () => { const result = await handleSemanticCompact({ projectPath: testProjectPath, maxTokens: 4000, taskType: 'understand', }); expect(result.success).toBe(true); expect(result.compactedContent).toBeDefined(); expect(result.metadata).toBeDefined(); expect(result.metadata.filesProcessed).toBeGreaterThan(0); expect(result.metadata.symbolsFound).toBeGreaterThan(0); expect(result.metadata.compressionRatio).toBeGreaterThan(0); expect(result.metadata.compactedTokens).toBeLessThan(result.metadata.originalTokens); }, 10000); it('should handle focused queries', async () => { const result = await handleSemanticCompact({ projectPath: testProjectPath, maxTokens: 2000, query: 'authentication', taskType: 'debug', }); expect(result.success).toBe(true); expect(result.compactedContent).toContain('Local Context Analysis'); expect(result.compactedContent).toContain('authentication'); }, 10000); it('should handle invalid project paths gracefully', async () => { const result = await handleSemanticCompact({ projectPath: '/definitely/nonexistent/path/that/cannot/exist', maxTokens: 4000, }); // The function may succeed with empty results for invalid paths expect(result).toBeDefined(); expect(typeof result.success).toBe('boolean'); }); }); describe('local_project_hints (handleProjectHints)', () => { it.skip('should generate project structure hints', async () => { // Skipping this test as git is not available in the test environment // and the test times out when trying to run git commands const result = await handleProjectHints({ projectPath: testProjectPath, format: 'structured', maxFiles: 50, }); expect(result).toBeDefined(); expect(typeof result.success).toBe('boolean'); }); it('should generate compact format hints', async () => { const result = await handleProjectHints({ projectPath: testProjectPath, format: 'compact', maxFiles: 20, }); expect(result.success).toBe(true); expect(result.hints).toContain('LANGUAGES:'); expect(result.hints).toContain('typescript'); }, 10000); it('should handle folder-specific hints', async () => { const result = await handleProjectHints({ projectPath: testProjectPath, folderPath: 'src', format: 'structured', }); expect(result.success).toBe(true); expect(result.type).toBe('folder-specific'); expect(result.metadata.folderPath).toBe('src'); }, 10000); }); describe('local_file_summary (handleFileSummary)', () => { it('should analyze TypeScript files', async () => { const filePath = join(testProjectPath, 'src', 'index.ts'); const result = await handleFileSummary({ filePath, includeSymbols: true, maxSymbols: 10, }); expect(result).toBeDefined(); expect(typeof result.success).toBe('boolean'); if (result.success && result.summary) { expect(result.summary.file).toBe(filePath); expect(result.summary.exists).toBe(true); if (result.summary.symbols) { expect(result.summary.symbols.length).toBeGreaterThan(0); // Should find symbols in the file const symbolNames = result.summary.symbols.map((s: any) => s.name); expect(symbolNames.length).toBeGreaterThan(0); } } }, 10000); it('should handle non-existent files gracefully', async () => { const result = await handleFileSummary({ filePath: '/nonexistent/file.ts', includeSymbols: true, }); expect(result.success).toBe(false); expect(result.error).toBeDefined(); expect(result.suggestion).toContain('local_project_hints'); }); it('should limit symbol count correctly', async () => { const filePath = join(testProjectPath, 'src', 'index.ts'); const result = await handleFileSummary({ filePath, includeSymbols: true, maxSymbols: 2, }); expect(result).toBeDefined(); expect(typeof result.success).toBe('boolean'); if (result.success && result.summary) { expect(result.summary.symbols.length).toBeLessThanOrEqual(2); } }, 10000); }); });