pytest-自定义命令行参数

背景

        在使用pytest做自动化测试时,我们有时候需要切换测试环境,例如在测试环境和生产环境,这时候就遇到环境切换的问题,而且有时候需要根据某个参数不同的参数值,执行不同的业务逻辑。

        上面的场景我们都可以通过“在命令行中输入参数,然后用例中接收这个参数,通过判断这个参数的值来做不同的逻辑”来实现,这时候我们就需要用到pytest的钩子函数:pytest_addoption。

使用方法

        pytest_addoption可以让用户注册一个自定义的命令行参数,方便将用户数据传递给pytest,而且这个方法一般和内置 fixture pytestconfig 配合使用,即pytest_addoption 注册命令行参数,pytestconfig 通过配置对象读取参数的值。

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


# 注册自定义参数 enr 到配置对象
def pytest_addoption(parser):
    """
    pytest_addoption 可以让用户注册一个自定义的命令行参数,方便用户将数据传递给 pytest;
    这个 Hook 方法一般和 内置 fixture pytestconfig 配合使用,pytest_addoption 注册命令行参数,
    pytestconfig 通过配置对象读取参数的值;
    :param parser:
    :return:
    """
    parser.addoption(
        "--enr", action="store", default="test", help="test:表示测试环境,默认test环境"
    )

#从配置对象获取enr的值,
@pytest.fixture(scope='session', autouse=True)
def env(pytestconfig):
    """
    获取运行环境并设置全局变量
    :param pytestconfig:
    :return:
    """
    glb.enr = pytestconfig.getoption("--enr")

上面conftest.py中新增了命令行参数:--enr,然后定义了个fixture,在测试用例中想要获得参数 --enr和 值,就可以调用 enr函数获取对应的值。

参数说明

name:自定义命令行参数的名字,可以是:"foo", "-foo" 或 "--foo";
action:在命令行中遇到此参数时要采取的基本操作类型;
nargs:应该使用的命令行参数的数量;
const:某些操作和nargs选择所需的常量值;
default:如果参数不在命令行中,则生成的默认值。
type:命令行参数应该转换为的类型;
choices:参数允许值的容器;
required:命令行选项是否可以省略(仅可选);
help:对参数作用的简要说明;
metavar:用法消息中参数的名称;
dest:要添加到 parse_args() 返回的对象中的属性的名称;

action="store":默认,只存储参数的值,可以存储任何类型的值,此时 default 也可以是任何类型的值,而且命令行参数多次使用也只能生效一个,最后一个值覆盖之前的值;

action="append":存储一个列表,用 append 模式 将可以同时多次使用自定义参数,并且 default  默认值必须是一个列表,pytest 会把 default  默认参数的值和多个自定义参数的值放在一个列表中。

おすすめ

転載: blog.csdn.net/sl01224318/article/details/121453405
おすすめ