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

前言
一、修改 conftest.py 文件
二、创建配置文件
三、修改search_steps.py文件
四、运行测试
总结

前言

本文教程知识点：

支持在多浏览器、多环境中执行测试

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

使用命令行参数灵活配置测试环境和浏览器选项；
实现多环境测试；
掌握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)

代码说明如下：

在 conftest.py 文件中定义了一个名为 base_url 的 fixture，它会根据命令行参数 --env 来选择不同的 URL；
在 navigate_to_search_page 函数中，base_url 作为参数传入，来获取当前环境的URL，让测试可以在不同环境中运行。

四、运行测试

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

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