Agent/test_browser_use.py

"""
最简单的 browser-use 测试
用于验证 browser-use 库本身是否能正常工作
"""

from browser_use import BrowserSession
import asyncio


async def test_simple():
    """最简单的测试：只启动和停止浏览器"""
    print("🧪 测试 1: 最简单的浏览器启动")
    print("="*60)

    try:
        # 最简单的配置
        session = BrowserSession(headless=False)
        print("✅ BrowserSession 创建成功")

        print("⏳ 正在启动浏览器...")
        await session.start()
        print("✅ 浏览器启动成功！")

        print("⏳ 等待 3 秒...")
        await asyncio.sleep(3)

        print("⏳ 正在停止浏览器...")
        await session.stop()
        print("✅ 浏览器停止成功！")

        print("\n" + "="*60)
        print("🎉 测试通过！browser-use 工作正常")
        print("="*60)

    except Exception as e:
        print(f"\n❌ 测试失败: {str(e)}")
        import traceback
        traceback.print_exc()
        print("\n" + "="*60)
        print("⚠️  browser-use 库存在问题")
        print("="*60)


async def test_with_user_data_dir():
    """测试 2: 使用 user_data_dir"""
    print("\n🧪 测试 2: 使用 user_data_dir")
    print("="*60)

    try:
        from pathlib import Path
        user_data_dir = str(Path.home() / ".browser_use" / "test_profile")
        Path(user_data_dir).mkdir(parents=True, exist_ok=True)

        session = BrowserSession(
            headless=False,
            user_data_dir=user_data_dir
        )
        print("✅ BrowserSession 创建成功（with user_data_dir）")

        print("⏳ 正在启动浏览器...")
        await session.start()
        print("✅ 浏览器启动成功！")

        print("⏳ 等待 3 秒...")
        await asyncio.sleep(3)

        print("⏳ 正在停止浏览器...")
        await session.stop()
        print("✅ 浏览器停止成功！")

        print("\n" + "="*60)
        print("🎉 测试通过！user_data_dir 工作正常")
        print("="*60)

    except Exception as e:
        print(f"\n❌ 测试失败: {str(e)}")
        import traceback
        traceback.print_exc()
        print("\n" + "="*60)
        print("⚠️  user_data_dir 配置存在问题")
        print("="*60)


async def main():
    """运行所有测试"""
    print("\n" + "="*60)
    print("🚀 开始测试 browser-use 库")
    print("="*60 + "\n")

    # 测试 1: 最简单的配置
    await test_simple()

    # 等待一下
    await asyncio.sleep(2)

    # 测试 2: 使用 user_data_dir
    await test_with_user_data_dir()

    print("\n" + "="*60)
    print("✅ 所有测试完成")
    print("="*60)


if __name__ == "__main__":
    asyncio.run(main())