Pytest-Bdd-Playwright 系列教程（2）：支持在多浏览器、多环境中执行测试

前言

学会本文教程内容后，你将掌握：

通过本文的学习，你将能够:

• 使用命令行参数灵活配置测试环境和浏览器选项；
• 实现多环境测试；
• 掌握Pytest fixture的高级用法；
• 学会使用配置文件管理测试参数；
• 优化项目结构,提高代码复用性和可维护性。

一、 修改 conftest.py 文件

conftest.py是Pytest的一个特殊文件,用于存放共享的fixture和hook函数。

我们对 conftest.py 进行以下修改:

• 添加命令行选项解析
• 实现动态浏览器选择
• 支持无头模式运行
• 引入环境配置

# conftest.py
# Author: blues_Cimport pytest
import loggingfrom pathlib import Path
from utils.config import Config
from playwright.sync_api import sync_playwrightlog = logging.getLogger(__name__)
root_path = Path(__file__).resolve().parentdef pytest_addoption(parser):parser.addoption("--browser", action="store", default="chromium", help="要使用的浏览器: chromium, firefox, or webkit")parser.addoption("--headless", action="store_true", default=False, help="运行浏览器无头模式")parser.addoption("--env", action="store", default=Config.DEFAULT_ENV, help="要测试的环境: test, staging, or prod")@pytest.fixture(scope="function")
def playwright():with sync_playwright() as playwright:yield playwright@pytest.fixture(scope="function")
def browser_type(playwright, request):browser_name = request.config.getoption("--browser")if browser_name == "chromium":return playwright.chromiumelif browser_name == "firefox":return playwright.firefoxelif browser_name == "webkit":return playwright.webkitelse:raise ValueError(f"不支持的浏览器: {browser_name}")@pytest.fixture(scope="function")
def browser(browser_type, request):headless = request.config.getoption("--headless")browser = browser_type.launch(headless=headless)log.info(f"启动浏览器: {browser_type.name}, 无头模式: {headless}")yield browserbrowser.close()@pytest.fixture(scope="function")
def base_url(request):env = request.config.getoption("--env")if env not in Config.ENVIRONMENTS:raise ValueError(f"不支持的环境: {env}")return Config.ENVIRONMENTS[env]@pytest.fixture(scope="function")
def page(browser):page = browser.new_page()yield pagepage.close()

代码修改说明如下：

• pytest_addoption: 添加自定义命令行选项
• browser_type: 根据命令行参数动态选择浏览器
• browser: 支持无头模式运行
• base_url: 根据环境参数选择测试URL

二、创建配置文件

为了更好地管理测试参数,我们创建一个专门的配置文件。
在utils/中创建一个config.py文件：

项目结构如下：

│
├── features/                # BDD特性文件
│   └── search.feature       # 搜索功能的特性文件
│
├── pages/                   # 页面对象
│   └── search_page.py       # 搜索页面对象
│
├── steps/                   # 步骤定义
│   └── search_steps.py      # 搜索相关步骤定义
│
├── tests/                   # 测试脚本
│   └── test_search.py
│
├── reports/                 # 测试报告（自动生成）
│   ├── logs/                # 日志文件目录（自动生成）
│   │   └── test_log.log     # 测试日志文件（自动生成）
│   └── test_report.html     # HTML格式的测试报告（自动生成）
│
├── utils/          # 工具函数
│   └── config.py   # 配置管理
│
├── conftest.py              # Pytest fixtures和全局配置
├── pytest.ini               # Pytest配置文件
├── requirements.txt         # 项目依赖
└── README.md                # 项目说明文档

# config.pyclass Config:ENVIRONMENTS = {'test': 'https://www.baidu.com','stage': 'https://example.com','prod': 'https://blog.csdn.net/weixin_48321392'}DEFAULT_ENV = 'test'config = Config()

三、修改search_steps.py文件

我们需要修改步骤定义文件以适应新的配置：

@given('我在百度搜索页面')
def navigate_to_search_page(search_page, base_url):search_page.navigate(base_url)

这里，我们使用base_url fixture来获取当前环境的URL，让测试可以在不同环境中运行。

四、运行测试

现在，我们可以使用不同的命令行参数来运行测试：

pytest # 默认使用chromium浏览器,非无头模式,测试环境
pytest --browser firefox --env stage # 使用Firefox浏览器,测试预发布环境
pytest --browser webkit --env prod --headless # 使用WebKit浏览器,无头模式,测试生产环境