406 lines
16 KiB
JavaScript
406 lines
16 KiB
JavaScript
const http = require('http');
|
||
|
||
// 配置
|
||
const CONFIG = {
|
||
host: '127.0.0.1',
|
||
port: 3456,
|
||
timeout: 5000
|
||
};
|
||
|
||
// 控制台输出颜色
|
||
const colors = {
|
||
reset: "\x1b[0m",
|
||
red: "\x1b[31m",
|
||
green: "\x1b[32m",
|
||
yellow: "\x1b[33m",
|
||
cyan: "\x1b[36m",
|
||
gray: "\x1b[90m"
|
||
};
|
||
|
||
function log(type, msg) {
|
||
const timestamp = new Date().toLocaleTimeString();
|
||
switch (type) {
|
||
case 'info': console.log(`${colors.cyan}[INFO]${colors.reset} ${msg}`); break;
|
||
case 'success': console.log(`${colors.green}[PASS]${colors.reset} ${msg}`); break;
|
||
case 'error': console.log(`${colors.red}[FAIL]${colors.reset} ${msg}`); break;
|
||
case 'warn': console.log(`${colors.yellow}[WARN]${colors.reset} ${msg}`); break;
|
||
case 'group': console.log(`\n${colors.gray}=== ${msg} ===${colors.reset}`); break;
|
||
default: console.log(msg);
|
||
}
|
||
}
|
||
|
||
// HTTP 辅助函数
|
||
function request(method, path, data) {
|
||
return new Promise((resolve, reject) => {
|
||
const options = {
|
||
hostname: CONFIG.host,
|
||
port: CONFIG.port,
|
||
path: path,
|
||
method: method,
|
||
headers: {
|
||
'Content-Type': 'application/json',
|
||
},
|
||
timeout: CONFIG.timeout
|
||
};
|
||
|
||
const req = http.request(options, (res) => {
|
||
let body = '';
|
||
res.on('data', chunk => body += chunk);
|
||
res.on('end', () => {
|
||
if (res.statusCode >= 200 && res.statusCode < 300) {
|
||
try {
|
||
const parsed = JSON.parse(body);
|
||
// MCP 返回 { content: [{ type: 'text', text: "..." }] }
|
||
resolve(parsed);
|
||
} catch (e) {
|
||
// 某些接口可能返回纯文本或非标准 JSON
|
||
resolve(body);
|
||
}
|
||
} else {
|
||
reject(new Error(`HTTP ${res.statusCode}: ${body}`));
|
||
}
|
||
});
|
||
});
|
||
|
||
req.on('error', (e) => reject(new Error(`连接失败: ${e.message}`)));
|
||
req.on('timeout', () => {
|
||
req.destroy();
|
||
reject(new Error('请求超时'));
|
||
});
|
||
|
||
if (data) {
|
||
req.write(JSON.stringify(data));
|
||
}
|
||
req.end();
|
||
});
|
||
}
|
||
|
||
// MCP 工具调用封装
|
||
async function callTool(name, args = {}) {
|
||
const payload = {
|
||
name: name,
|
||
arguments: args
|
||
};
|
||
|
||
try {
|
||
const response = await request('POST', '/call-tool', payload);
|
||
|
||
// 解析复杂的 MCP 响应结构
|
||
// 预期: { content: [ { type: 'text', text: "..." } ] }
|
||
if (response && response.content && Array.isArray(response.content)) {
|
||
const textContent = response.content.find(c => c.type === 'text');
|
||
if (textContent) {
|
||
// 工具结果本身可能是 JSON 字符串，尝试解析它
|
||
try {
|
||
return JSON.parse(textContent.text);
|
||
} catch {
|
||
return textContent.text;
|
||
}
|
||
}
|
||
}
|
||
return response;
|
||
} catch (e) {
|
||
throw new Error(`工具 [${name}] 调用失败: ${e.message}`);
|
||
}
|
||
}
|
||
|
||
// 断言辅助函数
|
||
function assert(condition, message) {
|
||
if (!condition) {
|
||
throw new Error(message || "断言失败");
|
||
}
|
||
}
|
||
|
||
// --- 测试套件 ---
|
||
const tests = {
|
||
async setup() {
|
||
log('group', '连接性检查');
|
||
try {
|
||
const tools = await request('POST', '/list-tools');
|
||
assert(tools && tools.tools && tools.tools.length > 0, "无法获取工具列表");
|
||
log('success', `已连接到 MCP 服务器。发现 ${tools.tools.length} 个工具。`);
|
||
return true;
|
||
} catch (e) {
|
||
log('error', `无法连接服务器。插件是否正在运行？ (${e.message})`);
|
||
return false;
|
||
}
|
||
},
|
||
|
||
async testNodeLifecycle() {
|
||
log('group', '节点生命周期测试');
|
||
const nodeName = `TestNode_${Date.now()}`;
|
||
|
||
try {
|
||
// 1. 创建节点
|
||
log('info', `尝试创建节点: ${nodeName}`);
|
||
const newNodeId = await callTool('create_node', { name: nodeName, type: 'empty' });
|
||
log('info', `create_node 响应: ${JSON.stringify(newNodeId)}`);
|
||
assert(typeof newNodeId === 'string' && newNodeId.length > 0, `create_node 没有返回 UUID。实际返回: ${JSON.stringify(newNodeId)}`);
|
||
log('success', `已创建节点: ${nodeName} (${newNodeId})`);
|
||
|
||
// 2. 查找节点
|
||
log('info', `尝试查找节点: ${nodeName}`);
|
||
const findResult = await callTool('find_gameobjects', { conditions: { name: nodeName } });
|
||
log('info', `find_gameobjects 响应: ${JSON.stringify(findResult)}`);
|
||
assert(Array.isArray(findResult), `find_gameobjects 没有返回数组。实际返回: ${JSON.stringify(findResult)}`);
|
||
assert(findResult.length >= 1, "find_gameobjects 未能找到已创建的节点");
|
||
|
||
// 查找特定节点（防止重名，虽然这里名字包含时间戳）
|
||
const targetNode = findResult.find(n => n.name === nodeName);
|
||
assert(targetNode, "找到节点但名称不匹配？");
|
||
assert(targetNode.uuid === newNodeId, `找到的节点 UUID 不匹配。预期 ${newNodeId}, 实际 ${targetNode.uuid}`);
|
||
log('success', `通过 find_gameobjects 找到节点: ${targetNode.name}`);
|
||
|
||
// 3. 更新变换 (Transform)
|
||
log('info', `尝试更新变换信息`);
|
||
await callTool('update_node_transform', { id: newNodeId, x: 100, y: 200 });
|
||
// 通过查找验证（因为查找会返回位置信息）
|
||
const updatedResult = await callTool('find_gameobjects', { conditions: { name: nodeName } });
|
||
const updatedNode = updatedResult.find(n => n.uuid === newNodeId);
|
||
log('info', `变换更新验证: x=${updatedNode.position.x}, y=${updatedNode.position.y}`);
|
||
assert(updatedNode.position.x === 100 && updatedNode.position.y === 200, `节点位置更新失败。实际: (${updatedNode.position.x}, ${updatedNode.position.y})`);
|
||
log('success', `节点变换已更新至 (100, 200)`);
|
||
|
||
return newNodeId; // 返回以供后续测试使用
|
||
} catch (e) {
|
||
log('error', `节点生命周期测试失败: ${e.message}`);
|
||
throw e;
|
||
}
|
||
},
|
||
|
||
async testComponents(nodeId) {
|
||
log('group', '组件管理测试');
|
||
|
||
// 1. 添加组件
|
||
// 使用 cc.Sprite 因为它最常用
|
||
log('info', `向 ${nodeId} 添加组件 cc.Sprite`);
|
||
const addResult = await callTool('manage_components', {
|
||
nodeId: nodeId,
|
||
action: 'add',
|
||
componentType: 'cc.Sprite'
|
||
});
|
||
log('success', `已添加 cc.Sprite 组件。响应: ${JSON.stringify(addResult)}`);
|
||
|
||
// 2. 获取组件
|
||
log('info', `列出 ${nodeId} 的组件`);
|
||
const components = await callTool('manage_components', { nodeId: nodeId, action: 'get' });
|
||
log('info', `manage_components (get) 响应: ${JSON.stringify(components)}`);
|
||
|
||
assert(Array.isArray(components), `无法获取组件列表。实际返回: ${JSON.stringify(components)}`);
|
||
// 宽松匹配：验证逻辑匹配（检查 type 或 properties.name 中是否包含 Sprite）
|
||
const spriteComp = components.find(c => (c.type && c.type.includes('Sprite')) || (c.properties && c.properties.name && c.properties.name.includes('Sprite')));
|
||
assert(spriteComp, "节点上未找到 cc.Sprite 组件");
|
||
log('success', `验证组件存在: ${spriteComp.uuid} (${spriteComp.type || 'Unknown'})`);
|
||
|
||
// 3. 移除组件
|
||
log('info', `移除组件 ${spriteComp.uuid}`);
|
||
const removeResult = await callTool('manage_components', {
|
||
nodeId: nodeId,
|
||
action: 'remove',
|
||
componentId: spriteComp.uuid
|
||
});
|
||
log('info', `移除结果: ${JSON.stringify(removeResult)}`);
|
||
|
||
// 等待引擎处理移除（异步过程）
|
||
await new Promise(r => setTimeout(r, 200));
|
||
|
||
// 验证移除
|
||
const componentsAfter = await callTool('manage_components', { nodeId: nodeId, action: 'get' });
|
||
log('info', `移除后的组件列表: ${JSON.stringify(componentsAfter)}`);
|
||
|
||
assert(!componentsAfter.find(c => (c.type && c.type.includes('Sprite')) || (c.uuid === spriteComp.uuid)), "组件未被移除");
|
||
log('success', `组件移除成功`);
|
||
},
|
||
|
||
async testEditorSelection(nodeId) {
|
||
log('group', '编辑器选中测试');
|
||
|
||
// 1. 设置选中
|
||
await callTool('manage_editor', {
|
||
action: 'set_selection',
|
||
target: 'node',
|
||
properties: { nodes: [nodeId] }
|
||
});
|
||
|
||
// 2. 获取选中
|
||
const selection = await callTool('manage_editor', { action: 'get_selection' });
|
||
// 预期: { nodes: [...], assets: [...] }
|
||
assert(selection.nodes && selection.nodes.includes(nodeId), "选中状态更新失败");
|
||
log('success', `编辑器选中状态已更新为节点 ${nodeId}`);
|
||
},
|
||
|
||
async testScriptOperations() {
|
||
log('group', '脚本读写与验证测试 (FS Mode)');
|
||
const scriptPath = 'db://assets/auto_test_script.js';
|
||
const initialContent = 'cc.log("Initial Content");';
|
||
const updatedContent = 'cc.log("Updated Content");';
|
||
|
||
// 1. 创建脚本
|
||
try {
|
||
log('info', `创建脚本: ${scriptPath}`);
|
||
await callTool('manage_script', {
|
||
action: 'create',
|
||
path: scriptPath,
|
||
content: initialContent
|
||
});
|
||
log('success', `脚本已创建`);
|
||
} catch (e) {
|
||
if (e.message.includes('exists')) {
|
||
log('warn', `脚本已存在，尝试删除重建...`);
|
||
await callTool('manage_asset', { action: 'delete', path: scriptPath });
|
||
await callTool('manage_script', { action: 'create', path: scriptPath, content: initialContent });
|
||
} else {
|
||
throw e;
|
||
}
|
||
}
|
||
|
||
// 等待资源导入
|
||
await new Promise(r => setTimeout(r, 2000));
|
||
|
||
// 2. 验证读取 (FS Read)
|
||
log('info', `验证读取内容...`);
|
||
const readContent = await callTool('manage_script', { action: 'read', path: scriptPath });
|
||
// 注意：Content 可能会包含一些编辑器自动添加的 meta 信息或者换行，可以宽松匹配
|
||
assert(readContent && readContent.includes("Initial Content"), `读取内容不匹配。实际: ${readContent}`);
|
||
log('success', `脚本读取成功`);
|
||
|
||
// 3. 验证写入 (FS Write + Refresh)
|
||
log('info', `验证写入内容...`);
|
||
await callTool('manage_script', { action: 'write', path: scriptPath, content: updatedContent });
|
||
|
||
// 等待刷新
|
||
await new Promise(r => setTimeout(r, 1000));
|
||
|
||
const readUpdated = await callTool('manage_script', { action: 'read', path: scriptPath });
|
||
assert(readUpdated && readUpdated.includes("Updated Content"), `写入后读取内容不匹配。实际: ${readUpdated}`);
|
||
log('success', `脚本写入成功`);
|
||
|
||
// 4. 验证脚本语法 (Validation)
|
||
log('info', `验证脚本语法...`);
|
||
const validation = await callTool('validate_script', { filePath: scriptPath });
|
||
log('info', `验证结果: ${JSON.stringify(validation)}`);
|
||
assert(validation && validation.valid === true, "脚本验证失败");
|
||
log('success', `脚本语法验证通过`);
|
||
|
||
// 5. 清理
|
||
await callTool('manage_asset', { action: 'delete', path: scriptPath });
|
||
log('success', `清理临时脚本`);
|
||
},
|
||
|
||
async testPrefabOperations(sourceNodeId) {
|
||
log('group', '预制体管理测试 (UUID Mode)');
|
||
const prefabPath = 'db://assets/AutoTestPrefab.prefab';
|
||
|
||
// 确保清理旧的
|
||
try {
|
||
await callTool('manage_asset', { action: 'delete', path: prefabPath });
|
||
} catch (e) { }
|
||
|
||
// 1. 创建预制体
|
||
log('info', `从节点 ${sourceNodeId} 创建预制体: ${prefabPath}`);
|
||
await callTool('prefab_management', {
|
||
action: 'create',
|
||
path: prefabPath,
|
||
nodeId: sourceNodeId
|
||
});
|
||
|
||
// 等待预制体生成和导入 (使用轮询机制)
|
||
log('info', '等待预制体生成...');
|
||
let prefabInfo = null;
|
||
|
||
// 每 200ms 检查一次，最多尝试 30 次 (6秒)
|
||
for (let i = 0; i < 30; i++) {
|
||
try {
|
||
prefabInfo = await callTool('prefab_management', { action: 'get_info', path: prefabPath });
|
||
if (prefabInfo && prefabInfo.exists) {
|
||
break;
|
||
}
|
||
} catch (e) { }
|
||
await new Promise(r => setTimeout(r, 200));
|
||
}
|
||
|
||
// 最终断言
|
||
assert(prefabInfo && prefabInfo.exists, "预制体创建失败或未找到 (超时)");
|
||
log('success', `预制体创建成功: ${prefabInfo.uuid}`);
|
||
|
||
// 2. 实例化预制体 (使用 UUID 加载)
|
||
log('info', `尝试实例化预制体 (UUID: ${prefabInfo.uuid})`);
|
||
const result = await callTool('prefab_management', {
|
||
action: 'instantiate',
|
||
path: prefabPath
|
||
});
|
||
log('info', `实例化结果: ${JSON.stringify(result)}`);
|
||
// 结果通常是一条成功消息字符串
|
||
assert(result && result.toLowerCase().includes('success'), "实例化失败");
|
||
log('success', `预制体实例化成功`);
|
||
|
||
// 3. 清理预制体
|
||
await callTool('manage_asset', { action: 'delete', path: prefabPath });
|
||
log('success', `清理临时预制体`);
|
||
},
|
||
|
||
async testResources() {
|
||
log('group', 'MCP Resource 协议测试');
|
||
|
||
// 1. 列出资源
|
||
log('info', '请求资源列表 (/list-resources)');
|
||
const listRes = await request('POST', '/list-resources');
|
||
log('info', `资源列表响应: ${JSON.stringify(listRes)}`);
|
||
assert(listRes && listRes.resources && Array.isArray(listRes.resources), "资源列表格式错误");
|
||
const hasHierarchy = listRes.resources.find(r => r.uri === 'cocos://hierarchy');
|
||
assert(hasHierarchy, "未找到 cocos://hierarchy 资源");
|
||
log('success', `成功获取资源列表 (包含 ${listRes.resources.length} 个资源)`);
|
||
|
||
// 2. 读取资源: Hierarchy
|
||
log('info', '读取资源: cocos://hierarchy');
|
||
const hierarchyRes = await request('POST', '/read-resource', { uri: 'cocos://hierarchy' });
|
||
assert(hierarchyRes && hierarchyRes.contents && hierarchyRes.contents.length > 0, "读取 Hierarchy 失败");
|
||
const hierarchyContent = hierarchyRes.contents[0].text;
|
||
assert(hierarchyContent && hierarchyContent.startsWith('['), "Hierarchy 内容应该是 JSON 数组");
|
||
log('success', `成功读取场景层级数据`);
|
||
|
||
// 3. 读取资源: Logs
|
||
log('info', '读取资源: cocos://logs/latest');
|
||
const logsRes = await request('POST', '/read-resource', { uri: 'cocos://logs/latest' });
|
||
assert(logsRes && logsRes.contents && logsRes.contents.length > 0, "读取 Logs 失败");
|
||
const logsContent = logsRes.contents[0].text;
|
||
assert(typeof logsContent === 'string', "日志内容应该是字符串");
|
||
log('success', `成功读取编辑器日志`);
|
||
}
|
||
};
|
||
|
||
async function run() {
|
||
console.log(`\n${colors.cyan}正在启动 MCP Bridge 自动化测试...${colors.reset}`);
|
||
console.log(`目标: http://${CONFIG.host}:${CONFIG.port}\n`);
|
||
|
||
const isConnected = await tests.setup();
|
||
if (!isConnected) process.exit(1);
|
||
|
||
try {
|
||
const nodeId = await tests.testNodeLifecycle();
|
||
|
||
await tests.testComponents(nodeId);
|
||
|
||
await tests.testEditorSelection(nodeId);
|
||
|
||
await tests.testEditorSelection(nodeId);
|
||
|
||
await tests.testScriptOperations();
|
||
|
||
await tests.testPrefabOperations(nodeId);
|
||
|
||
await tests.testResources();
|
||
|
||
// 清理：我们在测试中已经尽可能清理了，但保留节点可能有助于观察结果
|
||
// 这里只是打印完成消息
|
||
|
||
console.log(`\n${colors.green}所有测试已成功完成！${colors.reset}\n`);
|
||
} catch (e) {
|
||
console.error(`\n${colors.red}[FATAL ERROR]${colors.reset} 测试套件出错:`);
|
||
console.error(e);
|
||
process.exit(1);
|
||
}
|
||
}
|
||
|
||
run();
|