Function TestPythonRuntimeToolInjection

pkg/tools/bridge/ptc_integration_test.go:185–219

TestPythonRuntimeToolInjection 测试 Python 运行时工具注入

(t *testing.T)

// TestPythonRuntimeToolInjection 测试 Python 运行时工具注入
func TestPythonRuntimeToolInjection(t *testing.T) {
	runtime := NewPythonRuntime(nil)

	// 测试无工具的简单模式
	t.Run("NoToolsMode", func(t *testing.T) {
		code := "print('Hello World')"
		result, err := runtime.Execute(context.Background(), code, map[string]any{})
		if err != nil {
			t.Fatalf("Failed to execute: %v", err)
		}
		if !result.Success {
			t.Error("Expected success")
		}
	})

	// 测试带工具的 PTC 模式
	t.Run("PTCMode", func(t *testing.T) {
		runtime.SetTools([]string{"Read", "Write"})
		runtime.SetBridgeURL("http://localhost:8080")

		// 验证生成的代码包含工具注入
		wrappedCode := runtime.wrapCode("print('test')", map[string]any{})

		// 检查是否包含关键组件
		if !contains(wrappedCode, "_AsterBridge") {
			t.Error("Expected _AsterBridge class in wrapped code")
		}
		if !contains(wrappedCode, "async def _user_main()") {
			t.Error("Expected async main wrapper")
		}
		if !contains(wrappedCode, "Read") {
			t.Error("Expected Read tool in injected tools")
		}
	})
}

// contains 检查字符串是否包含子串
func contains(s, substr string) bool {