深度解析 pytest 参数化与 --count 执行顺序的奥秘

_可乐无糖2025-01-15 19:50

有这样一个业务场景，登录不同地区的账号，重复500遍，以验证登录功能是否正常。

登录的代码如下，其中login_data是一个fixture，用来组织数据：

python 复制代码

@pytest.mark.parametrize("login_data", ['cn_test', 'us', 'eu'], indirect=True) 
def test_login_change_region(appium_service, ios_driver, login_data):
    ...

那么如何实现重复登录500遍呢？按照之前文章介绍过的内容来说，在设置--count 500即可，但这样你会发现，'cn_test'登录了500遍、'us'登录了500遍、'eu'登录了500遍。

之前文章讲过，两个parametrize中的参数会按照笛卡尔积进行排序，当时我们并没有关心它们的执行顺序，例如像下面这样写，执行结果会怎样呢？

python 复制代码

@pytest.mark.parametrize("repeat_count", list(range(2)))
@pytest.mark.parametrize("login_data", ['cn_test', 'us', 'eu'], indirect=True) 
def test_login_change_region(appium_service, ios_driver, login_data, repeat_count: int):
    ...

首先总次数肯定是6次，没问题，那么顺序呢？实际执行的顺序是这样的：

'cn_test'登录了2遍、'us'登录了2遍、'eu'登录了2遍。不是第1遍登录'cn_test'、'us'、'eu'，第2遍登录'cn_test'、'us'、'eu'。

那么如果我就是想要第1遍登录'cn_test'、'us'、'eu'，第2遍登录'cn_test'、'us'、'eu' 如何实现呢？

这里介绍一种方法，手动组合参数以获得所需的执行顺序：

python 复制代码

@pytest.mark.parametrize("login_data", ['cn_test', 'us', 'eu'] * 2， indirect=True)  # 重复2次
def test_login_change_region(appium_service, ios_driver, login_data):
    # 登录逻辑

这样得到的顺序就是：第1遍登录'cn_test'、'us'、'eu'，第2遍登录'cn_test'、'us'、'eu'了。

【总结】

同样的操作，但是不同的执行顺序可能会发现不同的问题，这样就需要我们控制测试方法的执行顺序，所以我们不仅要考虑如何编写测试方法，还有考虑它们的执行顺序。