/**
 * @file test_mcp_tools_extended.cpp
 * @brief 测试MCP工具相关功能的扩展测试
 * 
 * 本文件包含对MCP工具模块的扩展单元测试，基于规范2024-11-05。
 */

#include "mcp_tool.h"
#include "mcp_server.h"
#include "mcp_client.h"
#include <gtest/gtest.h>
#include <gmock/gmock.h>
#include <string>
#include <thread>
#include <future>
#include <chrono>

// 测试类，用于设置服务器和客户端
class McpToolsExtendedTest : public ::testing::Test {
protected:
    void SetUp() override {
        // 创建服务器
        server = std::make_unique<mcp::server>("localhost", 8098);
        server->set_server_info("TestServer", "2024-11-05");
        
        // 设置服务器能力
        mcp::json capabilities = {
            {"tools", {{"listChanged", true}}}
        };
        server->set_capabilities(capabilities);
        
        // 注册计算器工具
        mcp::tool calculator = mcp::tool_builder("calculator")
            .with_description("计算器工具")
            .with_string_param("operation", "操作类型 (add, subtract, multiply, divide)")
            .with_number_param("a", "第一个操作数")
            .with_number_param("b", "第二个操作数")
            .build();
        
        server->register_tool(calculator, [](const mcp::json& params) -> mcp::json {
            std::string operation = params["operation"];
            double a = params["a"];
            double b = params["b"];
            double result = 0;
            
            if (operation == "add") {
                result = a + b;
            } else if (operation == "subtract") {
                result = a - b;
            } else if (operation == "multiply") {
                result = a * b;
            } else if (operation == "divide") {
                if (b == 0) {
                    return {{"error", "除数不能为零"}};
                }
                result = a / b;
            } else {
                return {{"error", "未知操作: " + operation}};
            }
            
            return {{"result", result}};
        });
        
        // 注册文本处理工具
        mcp::tool text_processor = mcp::tool_builder("text_processor")
            .with_description("文本处理工具")
            .with_string_param("text", "要处理的文本")
            .with_string_param("operation", "操作类型 (uppercase, lowercase, reverse)")
            .build();
        
        server->register_tool(text_processor, [](const mcp::json& params) -> mcp::json {
            std::string text = params["text"];
            std::string operation = params["operation"];
            std::string result;
            
            if (operation == "uppercase") {
                result = text;
                std::transform(result.begin(), result.end(), result.begin(), ::toupper);
            } else if (operation == "lowercase") {
                result = text;
                std::transform(result.begin(), result.end(), result.begin(), ::tolower);
            } else if (operation == "reverse") {
                result = text;
                std::reverse(result.begin(), result.end());
            } else {
                return {{"error", "未知操作: " + operation}};
            }
            
            return {{"result", result}};
        });
        
        // 注册列表处理工具
        mcp::tool list_processor = mcp::tool_builder("list_processor")
            .with_description("列表处理工具")
            .with_array_param("items", "要处理的项目列表", "string")
            .with_string_param("operation", "操作类型 (sort, reverse, count)")
            .build();
        
        server->register_tool(list_processor, [](const mcp::json& params) -> mcp::json {
            auto items = params["items"].get<std::vector<std::string>>();
            std::string operation = params["operation"];
            
            if (operation == "sort") {
                std::sort(items.begin(), items.end());
                return {{"result", items}};
            } else if (operation == "reverse") {
                std::reverse(items.begin(), items.end());
                return {{"result", items}};
            } else if (operation == "count") {
                return {{"result", items.size()}};
            } else {
                return {{"error", "未知操作: " + operation}};
            }
        });
    }
    
    void TearDown() override {
        // 停止服务器
        if (server && server_thread.joinable()) {
            server->stop();
            server_thread.join();
        }
    }
    
    // 启动服务器
    void start_server() {
        server_thread = std::thread([this]() {
            server->start(false);
        });
        // 等待服务器启动
        std::this_thread::sleep_for(std::chrono::milliseconds(100));
    }
    
    std::unique_ptr<mcp::server> server;
    std::thread server_thread;
};

// 测试获取工具列表
TEST_F(McpToolsExtendedTest, GetToolsTest) {
    // 启动服务器
    start_server();
    
    // 创建客户端
    mcp::client client("localhost", 8098);
    client.set_timeout(5);
    client.initialize("TestClient", mcp::MCP_VERSION);
    
    // 获取工具列表
    auto tools = client.get_tools();
    
    // 验证工具列表
    EXPECT_EQ(tools.size(), 3);
    
    // 验证工具名称
    std::vector<std::string> tool_names;
    for (const auto& tool : tools) {
        tool_names.push_back(tool.name);
    }
    
    EXPECT_THAT(tool_names, ::testing::UnorderedElementsAre("calculator", "text_processor", "list_processor"));
}

// 测试调用计算器工具
TEST_F(McpToolsExtendedTest, CalculatorToolTest) {
    // 启动服务器
    start_server();
    
    // 创建客户端
    mcp::client client("localhost", 8098);
    client.set_timeout(5);
    client.initialize("TestClient", mcp::MCP_VERSION);
    
    // 调用加法
    mcp::json add_result = client.call_tool("calculator", {
        {"operation", "add"},
        {"a", 5},
        {"b", 3}
    });
    EXPECT_EQ(add_result["result"], 8);
    
    // 调用减法
    mcp::json subtract_result = client.call_tool("calculator", {
        {"operation", "subtract"},
        {"a", 10},
        {"b", 4}
    });
    EXPECT_EQ(subtract_result["result"], 6);
    
    // 调用乘法
    mcp::json multiply_result = client.call_tool("calculator", {
        {"operation", "multiply"},
        {"a", 6},
        {"b", 7}
    });
    EXPECT_EQ(multiply_result["result"], 42);
    
    // 调用除法
    mcp::json divide_result = client.call_tool("calculator", {
        {"operation", "divide"},
        {"a", 20},
        {"b", 5}
    });
    EXPECT_EQ(divide_result["result"], 4);
    
    // 测试除以零
    mcp::json divide_by_zero = client.call_tool("calculator", {
        {"operation", "divide"},
        {"a", 10},
        {"b", 0}
    });
    EXPECT_TRUE(divide_by_zero.contains("error"));
}

// 测试调用文本处理工具
TEST_F(McpToolsExtendedTest, TextProcessorToolTest) {
    // 启动服务器
    start_server();
    
    // 创建客户端
    mcp::client client("localhost", 8098);
    client.set_timeout(5);
    client.initialize("TestClient", mcp::MCP_VERSION);
    
    // 测试转大写
    mcp::json uppercase_result = client.call_tool("text_processor", {
        {"text", "Hello World"},
        {"operation", "uppercase"}
    });
    EXPECT_EQ(uppercase_result["result"], "HELLO WORLD");
    
    // 测试转小写
    mcp::json lowercase_result = client.call_tool("text_processor", {
        {"text", "Hello World"},
        {"operation", "lowercase"}
    });
    EXPECT_EQ(lowercase_result["result"], "hello world");
    
    // 测试反转
    mcp::json reverse_result = client.call_tool("text_processor", {
        {"text", "Hello World"},
        {"operation", "reverse"}
    });
    EXPECT_EQ(reverse_result["result"], "dlroW olleH");
}

// 测试调用列表处理工具
TEST_F(McpToolsExtendedTest, ListProcessorToolTest) {
    // 启动服务器
    start_server();
    
    // 创建客户端
    mcp::client client("localhost", 8098);
    client.set_timeout(5);
    client.initialize("TestClient", mcp::MCP_VERSION);
    
    // 准备测试数据
    std::vector<std::string> items = {"banana", "apple", "orange", "grape"};
    
    // 测试排序
    mcp::json sort_result = client.call_tool("list_processor", {
        {"items", items},
        {"operation", "sort"}
    });
    std::vector<std::string> sorted_items = sort_result["result"];
    EXPECT_THAT(sorted_items, ::testing::ElementsAre("apple", "banana", "grape", "orange"));
    
    // 测试反转
    mcp::json reverse_result = client.call_tool("list_processor", {
        {"items", items},
        {"operation", "reverse"}
    });
    std::vector<std::string> reversed_items = reverse_result["result"];
    EXPECT_THAT(reversed_items, ::testing::ElementsAre("grape", "orange", "apple", "banana"));
    
    // 测试计数
    mcp::json count_result = client.call_tool("list_processor", {
        {"items", items},
        {"operation", "count"}
    });
    EXPECT_EQ(count_result["result"], 4);
}

// 测试工具参数验证
TEST_F(McpToolsExtendedTest, ToolParameterValidationTest) {
    // 启动服务器
    start_server();
    
    // 创建客户端
    mcp::client client("localhost", 8098);
    client.set_timeout(5);
    client.initialize("TestClient", mcp::MCP_VERSION);
    
    // 测试缺少必需参数
    try {
        client.call_tool("calculator", {
            {"a", 5}
            // 缺少 operation 和 b
        });
        FAIL() << "应该抛出异常，因为缺少必需参数";
    } catch (const mcp::mcp_exception& e) {
        EXPECT_EQ(e.code(), mcp::error_code::invalid_params);
    }
    
    // 测试参数类型错误
    try {
        client.call_tool("calculator", {
            {"operation", "add"},
            {"a", "not_a_number"}, // 应该是数字
            {"b", 3}
        });
        FAIL() << "应该抛出异常，因为参数类型错误";
    } catch (const mcp::mcp_exception& e) {
        EXPECT_EQ(e.code(), mcp::error_code::invalid_params);
    }
}

// 测试工具注册和注销
TEST_F(McpToolsExtendedTest, ToolRegistrationAndUnregistrationTest) {
    // 创建工具注册表
    mcp::tool_registry& registry = mcp::tool_registry::instance();
    
    // 创建一个测试工具
    mcp::tool test_tool = mcp::tool_builder("test_tool")
        .with_description("测试工具")
        .with_string_param("input", "输入参数")
        .build();
    
    // 注册工具
    registry.register_tool(test_tool, [](const mcp::json& params) -> mcp::json {
        return {{"output", "处理结果: " + params["input"].get<std::string>()}};
    });
    
    // 验证工具已注册
    auto registered_tool = registry.get_tool("test_tool");
    ASSERT_NE(registered_tool, nullptr);
    EXPECT_EQ(registered_tool->first.name, "test_tool");
    EXPECT_EQ(registered_tool->first.description, "测试工具");
    
    // 调用工具
    mcp::json result = registry.call_tool("test_tool", {{"input", "测试输入"}});
    EXPECT_EQ(result["output"], "处理结果: 测试输入");
    
    // 注销工具
    bool unregistered = registry.unregister_tool("test_tool");
    EXPECT_TRUE(unregistered);
    
    // 验证工具已注销
    EXPECT_EQ(registry.get_tool("test_tool"), nullptr);
    
    // 尝试调用已注销的工具
    try {
        registry.call_tool("test_tool", {{"input", "测试输入"}});
        FAIL() << "应该抛出异常，因为工具已注销";
    } catch (const mcp::mcp_exception& e) {
        EXPECT_EQ(e.code(), mcp::error_code::method_not_found);
    }
}