#!/usr/bin/env python3
"""
MCP工具本地测试
测试MCP服务器的各种工具功能
"""
import sys
import unittest
from pathlib import Path
# 添加测试工具路径
sys.path.insert(0, str(Path(__file__).parent.parent / 'utils'))
from test_helpers import BaseTestCase, create_test_server_config, create_test_docker_config
class TestMCPTools(BaseTestCase):
"""测试MCP工具功能"""
def test_mcp_server_import(self):
"""测试MCP服务器模块导入"""
try:
import mcp_server
# 检查关键函数是否存在
self.assertTrue(hasattr(mcp_server, 'create_tools_list'), "MCP服务器应该有create_tools_list函数")
self.assertTrue(hasattr(mcp_server, 'handle_request'), "MCP服务器应该有handle_request函数")
self.assertTrue(True, "MCP服务器模块导入成功")
except ImportError as e:
self.fail(f"MCP服务器模块导入失败: {e}")
def test_config_manager_tools(self):
"""测试配置管理工具"""
from config_manager.main import EnhancedConfigManager
# 使用临时配置目录
config_manager = EnhancedConfigManager()
# 测试关键方法存在
critical_methods = [
'quick_setup',
'guided_setup',
'get_existing_servers',
'preview_docker_wizard_command'
]
for method_name in critical_methods:
self.assert_method_exists(config_manager, method_name)
def test_docker_config_tools(self):
"""测试Docker配置工具"""
from docker_config_manager import DockerConfigManager
docker_manager = DockerConfigManager()
# 测试关键方法存在
critical_methods = [
'create_from_template',
'create_default_templates'
]
for method_name in critical_methods:
self.assert_method_exists(docker_manager, method_name)
def test_server_config_creation(self):
"""测试服务器配置创建"""
from config_manager.main import EnhancedConfigManager
config_manager = EnhancedConfigManager()
# 创建测试配置
test_config = {
'name': 'test-server',
'host': '192.168.1.100',
'username': 'testuser',
'port': 22,
'connection_type': 'ssh'
}
try:
# 测试获取现有服务器配置不抛出异常
result = config_manager.get_existing_servers()
self.assertIsNotNone(result, "获取服务器配置应该返回结果")
except Exception as e:
self.fail(f"获取服务器配置失败: {e}")
def test_docker_command_generation(self):
"""测试Docker命令生成"""
from config_manager.main import EnhancedConfigManager
config_manager = EnhancedConfigManager()
test_config = create_test_docker_config()
try:
# 这个方法会打印Docker命令，我们只需要确保它不抛出异常
config_manager.preview_docker_wizard_command(test_config)
self.assertTrue(True, "Docker命令生成正常执行")
except Exception as e:
self.fail(f"Docker命令生成失败: {e}")
def test_config_file_operations(self):
"""测试配置文件操作"""
from config_manager.main import EnhancedConfigManager
config_manager = EnhancedConfigManager()
# 测试配置文件路径
config_path = config_manager.config_path
self.assertTrue(str(config_path).endswith('.remote-terminal/config.yaml'),
"配置文件路径应该使用.remote-terminal目录")
def test_mcp_tool_availability(self):
"""测试MCP工具的可用性"""
try:
import mcp_server
# 获取工具列表
tools = mcp_server.create_tools_list()
# 检查关键工具是否注册
expected_tools = [
'list_servers',
'connect_server',
'execute_command',
'create_server_config',
'update_server_config'
]
tool_names = [tool['name'] for tool in tools]
for expected_tool in expected_tools:
self.assertIn(expected_tool, tool_names,
f"MCP工具列表应该包含{expected_tool}")
self.assertTrue(True, "MCP工具可用性检查成功")
except Exception as e:
self.fail(f"MCP工具可用性检查失败: {e}")
class TestConfigurationConsistency(BaseTestCase):
"""测试配置一致性"""
def test_config_directory_consistency(self):
"""测试配置目录一致性"""
from config_manager.main import EnhancedConfigManager
from docker_config_manager import DockerConfigManager
enhanced_manager = EnhancedConfigManager()
docker_manager = DockerConfigManager()
# 检查两个管理器使用相同的配置目录
enhanced_config_dir = enhanced_manager.config_path.parent
docker_config_dir = docker_manager.config_dir
self.assertEqual(enhanced_config_dir.name, docker_config_dir.name,
"所有配置管理器应该使用相同的配置目录名称")
# 确保都使用.remote-terminal目录
self.assertEqual(enhanced_config_dir.name, '.remote-terminal',
"配置目录应该是.remote-terminal")
self.assertEqual(docker_config_dir.name, '.remote-terminal',
"Docker配置目录应该是.remote-terminal")
if __name__ == '__main__':
# 运行测试
unittest.main(verbosity=2)
