SeleniumBase是一款全功能的浏览器自动化框架,专为Web测试、爬虫、数据抓取和"隐身"操作设计。
它构建在Selenium WebDriver的基础上,但提供了更简洁的API和强大的扩展功能,大幅提高了Web自动化任务的效率和可靠性。
作为一款专业的Web自动化工具包,SeleniumBase其核心目标是简化 Web 测试、数据抓取、隐身浏览等复杂任务,提供从单机脚本到分布式测试的全流程支持。
核心特性
-
简化的API:相比原生Selenium,提供更简洁、更直观的命令语法
-
多种运行模式:支持UC Mode和CDP Mode等特殊模式,具备"隐身"能力
-
多浏览器支持:兼容Chrome、Edge、Firefox和Safari等主流浏览器
-
跨平台兼容:支持Linux/Ubuntu、macOS和Windows操作系统
-
CI/CD集成:可在GitHub Actions、Jenkins、Azure、Google Cloud和AWS等平台上运行
-
自动等待:自动等待页面元素加载完成,无需手动添加
time.sleep()
语句 -
测试报告:生成丰富的HTML报告和仪表盘可视化测试结果
-
调试工具:提供Demo Mode和Debug Mode便于测试调试和可视化
项目结构与快速上手
目录解析
SeleniumBase/
├── examples/ # 200+实战示例(含测试、爬虫、隐身模式案例)
├── help_docs/ # 官方文档与客户案例
├── integrations/ # 集成模块(如Selenium Grid、MySQL日志存储)
├── sbase/ # 核心库代码(含CDP模式、驱动管理)
├── tests/ # 单元测试与集成测试
├── .github/ # 自动化工作流配置(如CI/CD脚本)
├── LICENSE # MIT开源协议
└── requirements.txt # 依赖清单(支持Python 3.8-3.13)
安装与初始化
# 从PyPI安装(推荐)
pip install seleniumbase
# 从GitHub克隆开发版
git clone https://github.com/seleniumbase/SeleniumBase.git
cd SeleniumBase
pip install -e .
# 初始化测试项目
sbase mkdir my_tests # 自动生成配置文件与示例脚本
代码示例
基于BaseCase类的测试示例
以下示例展示了SeleniumBase中最常用的测试方式,通过继承BaseCase类创建测试用例。
这段代码模拟了在电商网站上完成完整购物流程的场景,从登录到下单确认。
from seleniumbase import BaseCase
BaseCase.main(__name__, __file__)
class MyTestClass(BaseCase):
def test_swag_labs(self):
self.open("https://www.saucedemo.com")
self.type("#user-name", "standard_user")
self.type("#password", "secret_sauce")
self.click("input#login-button")
self.assert_element("div.inventory_list")
self.click("button#add-to-cart-sauce-labs-backpack")
self.click("#shopping_cart_container a")
self.assert_text("Backpack", "div.cart_item")
self.click("button#checkout")
self.type("#first-name", "SeleniumBase")
self.type("#last-name", "Automation")
self.type("#postal-code", "77123")
self.click("input#continue")
self.assert_text("Thank you for your order!")
BaseCase提供了丰富的内置方法如open()、type()、click()和assert_element()等,使测试代码简洁易读。
适合结构化的功能测试,可与pytest框架无缝集成,支持自动发现和运行。继承BaseCase的测试类还能获得丰富的内置功能,如截图、日志记录和详细的失败报告。
使用SB上下文管理器的简洁写法
以下代码展示了SeleniumBase的上下文管理器(Context Manager)用法,是快速执行短期自动化任务的理想选择。
通过with SB() as sb:语法,可以自动处理浏览器的启动和关闭,无需手动管理资源。
from seleniumbase import SB
with SB(uc=True, test=True, locale="en") as sb:
sb.open("https://nowsecure.nl/users/sign_in")
sb.activate_cdp_mode(url)
sb.xp_click_captcha()
sb.sleep(2)
参数uc=True启用了"无痕模式"(Undetected Chrome),能有效绕过现代网站的反爬机制。test=True启用测试模式,提供更详细的日志。
上下文管理器结束时会自动关闭浏览器,避免资源泄露,是处理验证码和安全挑战页面的利器。
使用Driver管理器
Driver管理器提供了比原生Selenium更简洁但功能相当的API,适合那些熟悉Selenium但希望减少代码量的开发者。
from seleniumbase import Driver
driver = Driver()
try:
driver.open("seleniumbase.io/simple/login")
driver.type("#username", "demo_user")
driver.type("#password", "secret_pass")
driver.click("a:contains('Sign in')")
driver.assert_exact_text("Welcome!", "h1")
driver.assert_element("img#image1")
driver.highlight("#image1")
driver.click_link("Sign out")
driver.assert_text("signed out", "#top_message")
finally:
driver.quit()
使用try-finally结构确保测试无论成功失败都会正确关闭浏览器,防止资源泄露。Driver类封装了许多实用方法,如open()、type()、click()等,同时保留了访问底层driver对象的能力,使开发者能在需要时直接调用原生Selenium方法。
特色功能详解
Demo模式
Demo模式可以帮助可视化测试过程,使测试步骤变得清晰可见:
pytest my_first_test.py --demo
在Demo模式下,测试会在浏览器操作之间短暂暂停,并高亮显示正在操作的页面元素,同时显示断言信息。当测试速度过快时,这个模式非常有用。
仪表盘和报告
SeleniumBase提供了强大的仪表盘功能,可实时显示测试结果:
pytest --dashboard --rs --headless
此命令会生成一个HTML仪表盘,自动更新测试运行结果,包括通过、失败和跳过的测试数量,以及详细的测试信息。您还可以使用以下命令生成HTML报告:
pytest test_suite.py --htmlreport.html
生成测试报告
使用–htmlreport.html参数可生成详细的HTML测试报告
pytest test_suite.py --dashboard --html=report.html
使用–junit-xml=report.xml生成XML格式报告
pytest test_suite.py --junit-xml=report.xml
还支持pynose格式的报告(–report选项),提供简洁的测试结果摘要,并支持使用–show-report即时显示
pynose test_suite.py --report
behave测试框架的用户可以使用类似的仪表盘和报告功能,通过–dashboard和–junit参数生成可视化结果。
behave behave_bdd/features/ -D dashboard -D headless
命令行选项
--browser=BROWSER
:指定使用的浏览器(默认为Chrome)--headless
:无头模式运行测试,不显示浏览器界面--demo
:以演示模式运行测试,展示每个步骤--rs
或--reuse-session
:重用浏览器会话,提高效率--dashboard
:生成实时更新的测试仪表盘--proxy=IP_ADDRESS:PORT
:使用代理服务器进行测试
SeleniumBase VS 原生Selenium
对比维度 | SeleniumBase | 原生Selenium |
---|---|---|
开发效率 | 单方法实现多步操作,语法极简 | 需要手动处理等待、异常等底层逻辑 |
反爬能力 | 内置UC/CDP隐身模式 | 需自行封装CDP协议 |
测试报告 | 支持HTML、仪表盘、JUnit XML | 需额外插件生成报告 |
团队协作 | 提供CasePlans用例管理、GUI工具 | 依赖第三方工具链 |
总结
SeleniumBase是一款卓越的Web自动化框架,它在Selenium WebDriver的基础上提供了显著的功能增强和简化的API。这个框架不仅支持常规的网站测试,还能有效处理验证码绕过、数据抓取和自动化任务执行。其直观的命令语法大幅降低了Web自动化的学习曲线,同时内置的Demo模式和调试工具让测试过程更加透明可控。SeleniumBase兼容所有主流浏览器和操作系统,可以无缝集成到各种CI/CD平台,使其成为团队自动化工作流的理想选择。
GitHub项目地址:https://github.com/seleniumbase/SeleniumBase
更多资料获取
📚 个人网站:ipengtao.com
如果还想要领取更多更丰富的资料,可以点击文章下方名片,回复【优质资料】,即可获取 全方位学习资料包。
点击文章下方链接卡片,回复【优质资料】,可直接领取资料大礼包。