1、pytest_addoption：动态添加命令行参数（钩子函数）

一、背景：

• 自动化用例需要支持在不同的环境运行：有时候是test测试环境，有时候是dev生产环境
• 执行不同的业务逻辑：给某个参数传入不同的参数值来执行不同的业务逻辑

二、解决方案：

可以通过“自定义一个命令行参数，并在命令行中输入这个参数，然后在测试中用例中接收这个参数，通过判断这个参数的值来做不同的逻辑”来实现

那么我们的需求就变为：pytest中如何自定义一个命令行参数呢？这时候我们就需要用到pytest的钩子函数：pytest_addoption

三、具体实现：

1、新建一个conftest.py文件，然后在conftest.py文件中通过pytest_addoption方法来添加命令行参数，通过定义的fixture来获得参数的值

import pytest


def pytest_addoption(parser):
    
    # 注册自定义参数cmdopt到配置对象
    parser.addoption(
        "--cmdopt", action="store", default="type1", help="my option: type1 or type2"
    )
    # 注册自定义参数env到配置对象
    parser.addoption(
        "--env", action="store", default="dev", help="env：表示测试环境，默认dev环境"
    )


@pytest.fixture()
def cmdopt(pytestconfig):
    # 从配置对象获取cmdopt的值
    return pytestconfig.getoption("--cmdopt")


@pytest.fixture()
def env(request):
    # 从配置对象获取env的值
    return request.config.getoption("--env")

备注：

1）新增了两个命令行参数：--cmdopt和--env

2）然后定义了两个fixture

3）在测试用例中想要获得参数--cmdopt的值、就可以调用cmdopt函数，想要获得参数--env的值、就可以调用env函数

2、编写测试用例test_option.py

import pytest


def test_option(env):
    if env == 'dev':
        print("当前环境为：{}，域名切换为开发环境".format(env))
    elif env == 'test':
        print("当前环境为：{}，域名切换为测试环境".format(env))
    else:
        print("环境错误，当前环境{}不存在".format(env))


if __name__ == '__main__':
    pytest.main(['-s', 'test_option.py'])

备注：获取env的值，针对不同的env值做不同的操作

3、测试用例运行：

1）不带参数运行：默认dev环境

pytest -s test_option.py 
================================================================== test session starts ==============================
platform darwin -- Python 3.11.0, pytest-7.4.0, pluggy-1.2.0
rootdir: /Users/chunyanzhang/Projects/Pycharm/Pytest/testcase
plugins: allure-pytest-2.13.2
collected 1 item                                                                                                                                                                                                                       

test_option.py 当前环境为：dev，域名切换为开发环境
.

============================================================== 1 passed in 0.02s ===============================

2) 带参数运行：可以指定为test环境

pytest -s --env=test test_option.py 
============================================== test session starts ==================================
platform darwin -- Python 3.11.0, pytest-7.4.0, pluggy-1.2.0
rootdir: /Users/chunyanzhang/Projects/Pycharm/Pytest/testcase
plugins: allure-pytest-2.13.2
collected 1 item                                                                                                                                                                                                                       

test_option.py 当前环境为：test，域名切换为测试环境
.

================================= 1 passed in 0.00s ==================================================