Function TestHTTPBridgeServerEndpoints

pkg/tools/bridge/ptc_integration_test.go:118–182

TestHTTPBridgeServerEndpoints 测试 HTTP 服务器端点

(t *testing.T)

// TestHTTPBridgeServerEndpoints 测试 HTTP 服务器端点
func TestHTTPBridgeServerEndpoints(t *testing.T) {
	// 创建工具注册表和桥接
	registry := tools.NewRegistry()
	registry.Register("MockTool", NewMockTool)
	toolBridge := NewToolBridge(registry)

	// 创建并启动服务器
	server := NewHTTPBridgeServer(toolBridge, "localhost:18081")
	if err := server.StartAsync(); err != nil {
		t.Fatalf("Failed to start server: %v", err)
	}
	defer func() {
		ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
		defer cancel()
		_ = server.Shutdown(ctx)
	}()

	// 等待服务器启动
	time.Sleep(200 * time.Millisecond)

	// 测试各个端点
	t.Run("ToolList", func(t *testing.T) {
		tools := toolBridge.ListAvailableTools()
		if len(tools) == 0 {
			t.Error("Expected at least one tool")
		}
		if tools[0] != "MockTool" {
			t.Errorf("Expected MockTool, got %s", tools[0])
		}
	})

	t.Run("ToolSchema", func(t *testing.T) {
		schema, err := toolBridge.GetToolSchema("MockTool")
		if err != nil {
			t.Fatalf("Failed to get schema: %v", err)
		}
		if schema["name"] != "MockTool" {
			t.Error("Schema name mismatch")
		}
	})

	t.Run("ToolCall", func(t *testing.T) {
		ctx := context.Background()
		result, err := toolBridge.CallTool(ctx, "MockTool", map[string]any{
			"message": "Test message",
		}, nil)

		if err != nil {
			t.Fatalf("Failed to call tool: %v", err)
		}

		if !result.Success {
			t.Errorf("Tool call failed: %s", result.Error)
		}

		resultMap, ok := result.Result.(map[string]any)
		if !ok {
			t.Fatal("Expected map result")