@xiaohui-wang/mcpadvisor
Version:
MCP Advisor & Installation - Find the right MCP server for your needs
60 lines (59 loc) • 2.54 kB
JavaScript
import { SearchService } from '../services/searchService.js';
/**
* 测试 SearchService
*/
describe('SearchService', () => {
// 模拟SearchService.searchGetMcp方法
beforeEach(() => {
// 为测试创建一个模拟的searchGetMcp方法
jest.spyOn(SearchService, 'searchGetMcp').mockImplementation(async (query, options) => {
const mockResults = [
{
title: 'Test Server 1',
description: 'A test server for AI applications',
github_url: 'https://github.com/test/server1',
similarity: 0.95
},
{
title: 'Test Server 2',
description: 'Another test server for data processing',
github_url: 'https://github.com/test/server2',
similarity: 0.85
},
{
title: 'Test Server 3',
description: 'A third test server for various tasks',
github_url: 'https://github.com/test/server3',
similarity: 0.75
}
];
// 如果设置了limit选项，则限制结果数量
if (options?.limit) {
return mockResults.slice(0, options.limit);
}
// 如果设置了minSimilarity选项，则过滤结果
if (options?.minSimilarity !== undefined) {
const minSimilarity = options.minSimilarity; // 创建本地变量，TypeScript能正确推断类型
return mockResults.filter(result => result.similarity >= minSimilarity);
}
return mockResults;
});
});
afterEach(() => {
jest.restoreAllMocks();
});
test('should return search results for a query', async () => {
const results = await SearchService.searchGetMcp('AI');
expect(results).toHaveLength(3);
expect(results[0].title).toBe('Test Server 1');
});
test('should limit results when limit option is provided', async () => {
const results = await SearchService.searchGetMcp('data', { limit: 2 });
expect(results).toHaveLength(2);
});
test('should filter results by minimum similarity', async () => {
const results = await SearchService.searchGetMcp('test', { minSimilarity: 0.8 });
expect(results.every(result => result.similarity >= 0.8)).toBe(true);
expect(results).toHaveLength(2);
});
});