python自动化-pytest-标记

• 标记通常搭配装饰器使用
  • pytest 的标记（Markers） 是一个非常强大和灵活的功能，允许你为测试用例“打标签”，从而实现分类、控制执行、修改行为等目的。
  • 可以使用 @pytest.mark.标记名 来装饰测试函数或测试类。

1、内置标记（Built-in Marks）

pytest 提供了一些开箱即用的内置标记：

• @pytest.mark.parametrize 详解

这是最常用的标记之一，用于数据驱动测试。数据驱动测试 = 参数化测试+数据文件

基本语法

@pytest.mark.parametrize("参数名", [值1, 值2, ...])
def test_function(参数名):
    ...

示例

import pytest

# 测试平方函数
@pytest.mark.parametrize("input,expected", [
    (2, 4),
    (3, 9),
    (-2, 4),
    (0, 0),
])
def test_square(input, expected):
    assert input ** 2 == expected

# 测试不同浏览器（字符串参数）
@pytest.mark.parametrize("browser", ["chrome", "firefox", "safari"])
def test_login(browser):
    print(f"Testing login with {browser}")
    # 启动对应浏览器并测试登录
    
-- 运行结果会显示为多个独立的测试用例：

test_square[2-4] PASSED
test_square[3-9] PASSED
test_square[-2-4] PASSED
test_login[chrome] PASSED
test_login[firefox] PASSED

2、自定义标记（Custom Marks）

可以创建自己的标记来分类测试，如 smoke, integration, e2e, slow 等。

• 2.1 在配置文件中声明自定义标记

如果不声明，pytest 会发出警告（PytestUnknownMarkWarning）。

在 pyproject.toml 中声明：

[tool.pytest.ini_options]
markers = [
    "smoke: 快速冒烟测试",
    "integration: 集成测试",
    "e2e: 端到端测试",
    "slow: 运行时间较长的测试",
    "db: 涉及数据库操作",
    "performance: 性能测试"
]

或在 pytest.ini 中：

[tool:pytest]
markers =
    smoke: 快速冒烟测试
    integration: 集成测试
    slow: 运行时间较长的测试

• 2.2 使用自定义标记

import pytest

@pytest.mark.smoke
def test_login():
    assert True

@pytest.mark.integration
@pytest.mark.db
class TestUserAPI:
    def test_create_user(self):
        assert True

    @pytest.mark.performance
    def test_user_list_speed(self):
        assert True

• 2.3 使用标记来控制测试执行

标记的强大之处在于你可以选择性地运行或跳过测试。

3、内置标记和自定义标记的区别

4、类标记

• 类上的标记会应用到类中所有的测试方法。
• 可以在方法上添加额外的标记。

@pytest.mark.db
class TestDatabase:
    def test_read(self):  # 有 db 标记
        pass

    @pytest.mark.write
    def test_write(self):  # 有 db 和 write 两个标记
        pass

    @pytest.mark.skip
    def test_delete(self):  # 有 db 标记，但会被跳过
        pass

5、查看所有已注册的标记

pytest --markers

输出示例：

@pytest.mark.smoke - 快速冒烟测试
@pytest.mark.integration - 集成测试
@pytest.mark.slow - 运行时间较长的测试
@pytest.mark.skip - skip this test unconditionally
@pytest.mark.skipif - skip the given test if condition is True
...

6、实践

1. 为自定义标记在配置文件中声明，避免警告。
2. 使用有意义的标记名，如 smoke, e2e, ui, api。
3. 结合 CI/CD 使用：例如，在快速流水线中只运行 smoke 测试，在完整流水线中运行所有测试。
4. 避免过度标记：标记太多会增加管理成本。
5. 利用 parametrize + mark：可以为参数化测试的特定数据点打标记。

总结

pytest 的标记系统是其灵活性和可扩展性的核心之一。通过标记，可以：

• 组织和分类测试用例。
• 控制执行流程（跳过、预期失败）。
• 实现参数化测试。
• 集成到自动化流程中（如只运行冒烟测试）。