test-mcp-server.js•4.49 kB
#!/usr/bin/env node
import { spawn } from 'child_process';
import { writeFileSync, unlinkSync } from 'fs';
import path from 'path';
console.log('🧪 开始MCP服务器完整功能测试...\n');
// 创建测试文档
const testDocPath = path.join(process.cwd(), 'test-document.txt');
const testContent = `这是一个测试文档。
人工智能（Artificial Intelligence，AI）是计算机科学的一个分支，它企图了解智能的实质，并生产出一种新的能以人类智能相似的方式做出反应的智能机器。
该领域的研究包括机器人、语言识别、图像识别、自然语言处理和专家系统等。人工智能从诞生以来，理论和技术日益成熟，应用领域也不断扩大，可以设想，未来人工智能带来的科技产品，将会是人类智慧的"容器"。
人工智能可以对人的意识、思维的信息过程的模拟。人工智能不是人的智能，但能像人那样思考、也可能超过人的智能。`;
writeFileSync(testDocPath, testContent, 'utf-8');
console.log('✅ 测试文档创建成功');
// 启动MCP服务器
console.log('\n🚀 启动MCP服务器...');
const server = spawn('node', ['dist/index.js'], {
stdio: ['pipe', 'pipe', 'pipe']
});
let serverReady = false;
let responses = [];
// 监听服务器输出
server.stdout.on('data', (data) => {
const output = data.toString();
console.log('📤 服务器输出:', output.trim());
if (output.includes('知识库MCP服务器已启动')) {
serverReady = true;
console.log('✅ 服务器启动成功！');
}
});
server.stderr.on('data', (data) => {
const output = data.toString();
console.log('⚠️ 服务器错误:', output.trim());
if (output.includes('知识库MCP服务器已启动')) {
serverReady = true;
console.log('✅ 服务器启动成功！');
}
});
// 发送MCP请求
function sendMCPRequest(method, params) {
const request = {
jsonrpc: '2.0',
id: Date.now(),
method,
params
};
console.log(`📤 发送请求: ${method}`);
server.stdin.write(JSON.stringify(request) + '\n');
}
// 监听服务器响应
server.stdout.on('data', (data) => {
const output = data.toString();
try {
const lines = output.trim().split('\n');
for (const line of lines) {
if (line.trim() && line.startsWith('{')) {
const response = JSON.parse(line);
responses.push(response);
console.log(`📥 收到响应: ${response.method || 'result'}`);
}
}
} catch (error) {
// 忽略非JSON输出
}
});
// 测试流程
setTimeout(() => {
if (!serverReady) {
console.log('❌ 服务器启动超时');
cleanup();
return;
}
console.log('\n📋 开始测试MCP服务器功能...\n');
// 测试1: 添加文档
console.log('1️⃣ 测试添加文档...');
sendMCPRequest('tools/call', {
name: 'add_document',
arguments: {
file_path: testDocPath
}
});
// 测试2: 查询知识库
setTimeout(() => {
console.log('\n2️⃣ 测试查询知识库...');
sendMCPRequest('tools/call', {
name: 'query_knowledge_base',
arguments: {
question: '什么是人工智能？'
}
});
}, 2000);
// 测试3: 获取统计信息
setTimeout(() => {
console.log('\n3️⃣ 测试获取统计信息...');
sendMCPRequest('tools/call', {
name: 'get_stats',
arguments: {}
});
}, 4000);
// 测试4: 列出文档
setTimeout(() => {
console.log('\n4️⃣ 测试列出文档...');
sendMCPRequest('tools/call', {
name: 'list_documents',
arguments: {}
});
}, 6000);
// 测试5: 清空知识库
setTimeout(() => {
console.log('\n5️⃣ 测试清空知识库...');
sendMCPRequest('tools/call', {
name: 'clear_knowledge_base',
arguments: {}
});
}, 8000);
// 完成测试
setTimeout(() => {
console.log('\n📊 测试结果统计:');
console.log(` 收到 ${responses.length} 个响应`);
console.log('🎉 MCP服务器功能测试完成！');
cleanup();
}, 10000);
}, 3000);
function cleanup() {
try {
unlinkSync(testDocPath);
console.log('✅ 测试文件已清理');
} catch (error) {
console.log('⚠️ 清理测试文件失败:', error.message);
}
server.kill();
console.log('✅ 服务器已停止');
process.exit(0);
}
// 处理进程退出
process.on('SIGINT', cleanup);
process.on('SIGTERM', cleanup);