Pytest-Bdd-Playwright 系列教程(2):支持在多浏览器、多环境中执行测试
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)
这里,我们使用base_url fixture来获取当前环境的URL,让测试可以在不同环境中运行。
四、运行测试
现在,我们可以使用不同的命令行参数来运行测试:
pytest # 默认使用chromium浏览器,非无头模式,测试环境
pytest --browser firefox --env stage # 使用Firefox浏览器,测试预发布环境
pytest --browser webkit --env prod --headless # 使用WebKit浏览器,无头模式,测试生产环境