// IDE过滤功能测试
import { Client } from '@modelcontextprotocol/sdk/client/index.js';
import { StdioClientTransport } from '@modelcontextprotocol/sdk/client/stdio.js';
import { dirname, join } from 'path';
import { fileURLToPath } from 'url';
import { expect, test } from 'vitest';
const __filename = fileURLToPath(import.meta.url);
const __dirname = dirname(__filename);
// Helper function to wait for delay
const delay = (ms) => new Promise(resolve => setTimeout(resolve, ms));
test('downloadTemplate tool supports IDE filtering', async () => {
let transport = null;
let client = null;
try {
console.log('Testing downloadTemplate IDE filtering functionality...');
// Create client
client = new Client({
name: "test-client-ide-filtering",
version: "1.0.0",
}, {
capabilities: {}
});
// Use the CJS CLI for integration testing
const serverPath = join(__dirname, '../mcp/dist/cli.cjs');
transport = new StdioClientTransport({
command: 'node',
args: [serverPath],
// Only enable the minimal plugin set to speed up server startup in tests
env: { ...process.env, CLOUDBASE_MCP_PLUGINS_ENABLED: "setup" }
});
// Connect client to server
await client.connect(transport);
await delay(500);
console.log('Testing downloadTemplate tool availability...');
// List tools to find downloadTemplate
const toolsResult = await client.listTools();
expect(toolsResult.tools).toBeDefined();
expect(Array.isArray(toolsResult.tools)).toBe(true);
const downloadTemplateTool = toolsResult.tools.find(t => t.name === 'downloadTemplate');
expect(downloadTemplateTool).toBeDefined();
console.log('✅ downloadTemplate tool found');
// Check if the tool has IDE parameter
const toolSchema = downloadTemplateTool.inputSchema;
expect(toolSchema).toBeDefined();
// Check if ide parameter exists
const ideParam = toolSchema.properties?.ide;
expect(ideParam).toBeDefined();
expect(ideParam.description).toContain('指定要下载的IDE类型');
console.log('✅ IDE parameter found in tool schema');
// Check if ide parameter has correct enum values
expect(ideParam.enum).toBeDefined();
expect(Array.isArray(ideParam.enum)).toBe(true);
expect(ideParam.enum).toContain('all');
expect(ideParam.enum).toContain('cursor');
expect(ideParam.enum).toContain('windsurf');
expect(ideParam.enum).toContain('codebuddy');
expect(ideParam.enum).toContain('claude-code');
expect(ideParam.enum).toContain('cline');
expect(ideParam.enum).toContain('gemini-cli');
expect(ideParam.enum).toContain('opencode');
expect(ideParam.enum).toContain('qwen-code');
expect(ideParam.enum).toContain('baidu-comate');
expect(ideParam.enum).toContain('openai-codex-cli');
expect(ideParam.enum).toContain('augment-code');
expect(ideParam.enum).toContain('github-copilot');
expect(ideParam.enum).toContain('roocode');
expect(ideParam.enum).toContain('tongyi-lingma');
expect(ideParam.enum).toContain('trae');
expect(ideParam.enum).toContain('vscode');
console.log('✅ All supported IDE types found in enum');
console.log('✅ downloadTemplate IDE filtering test passed');
} catch (error) {
console.error('❌ downloadTemplate IDE filtering test failed:', error);
throw error;
} finally {
if (client) {
await client.close();
}
if (transport) {
await transport.close();
}
}
}, 90000);
test('downloadTemplate tool requires IDE parameter when not detected', async () => {
let transport = null;
let client = null;
try {
console.log('Creating client...');
// Create client
client = new Client({
name: "test-client-ide-required",
version: "1.0.0",
}, {
capabilities: {}
});
// Use the CJS CLI for integration testing
const serverPath = join(__dirname, '../mcp/dist/cli.cjs');
const env = { ...process.env };
delete env.INTEGRATION_IDE; // Ensure INTEGRATION_IDE is not set
env.CLOUDBASE_MCP_PLUGINS_ENABLED = "setup";
env.NODE_ENV = "test";
env.VITEST = "true";
transport = new StdioClientTransport({
command: 'node',
args: [serverPath],
env
});
console.log('Connecting to MCP server...');
// Connect client to server
await client.connect(transport);
console.log('Connected. Waiting 500ms...');
await delay(500);
// Verify tool is available
const toolsResult = await client.listTools();
const downloadTemplateTool = toolsResult.tools.find(t => t.name === 'downloadTemplate');
if (!downloadTemplateTool) {
throw new Error('downloadTemplate tool not found');
}
console.log('✅ downloadTemplate tool found');
console.log('Calling downloadTemplate (missing ide)...');
// Call downloadTemplate without ide parameter
const result = await Promise.race([
client.callTool({
name: 'downloadTemplate',
arguments: {
template: 'rules',
overwrite: false
}
}),
new Promise((_, reject) =>
setTimeout(() => reject(new Error('callTool timeout after 10s')), 10000)
)
]);
console.log('Call completed. Result:', result.content?.[0]?.text?.substring(0, 100));
// Verify error message
expect(result.content).toBeDefined();
expect(result.content.length).toBeGreaterThan(0);
expect(result.content[0].text).toContain('必须指定 IDE 参数');
expect(result.content[0].text).toContain('请传入 `ide` 参数');
} catch (error) {
console.error('Test failed:', error.message);
throw error;
} finally {
if (client) {
await client.close();
}
if (transport) {
await transport.close();
}
}
}, 60000);
