Pytest教程__Hook函数pytest_addoption(parser)：定义自己的命令行参数（14-1）

最新推荐文章于 2024-11-08 18:29:04 发布

bug捕手

最新推荐文章于 2024-11-08 18:29:04 发布

阅读量1.9k

点赞数 2

CC 4.0 BY-SA版权

分类专栏： pytest 文章标签： pytest python 软件测试自动化测试自动化测试框架

本文链接：https://blog.youkuaiyun.com/GDYY3721/article/details/131246594

pytest 专栏收录该内容

18 篇文章

订阅专栏

文章介绍了如何在pytest框架下，通过conftest.py文件定义钩子函数pytest_addoption来自定义命令行参数，如`--env_opt`和`--run_level`，并使用fixture获取这些参数值，以实现不同测试环境的切换和不同业务逻辑的执行。在测试用例中通过调用fixture函数获取参数，实现动态配置和控制测试流程。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

考虑场景：

我们的自动化用例需要支持在不同测试环境运行，有时候在dev环境运行，有时候在test环境运行；
有时候需要根据某个参数不同的参数值，执行不同的业务逻辑；

上面的场景我们都可以通过“在命令行中输入参数，然后用例中接收这个参数，通过判断这个参数的值来做不同的逻辑”来实现。那么我们的需求就变为pytest中如何自定义一个命令行参数呢？这时候我们就需要用到pytest的钩子函数：pytest_addoption

在conftest.py文件中定义命令名

新建一个conftest.py文件

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

import pytest


# pytest_addoption(parser) 定义自己的命令行参数的固定写法
def pytest_addoption(parser):
    # 定义 --env_opt 参数名
    parser.addoption("--env_opt")

    # 定义 --run_level 参数名
    # 参数说明：
    #   default：当命令行不调用参数时的默认值
    #   help：在帮助中显示的说明
    parser.addoption("--run_level", default=1, help="执行用例的级别", action="store")


# 获取--env_opt参数值
@pytest.fixture(scope="session")
def env_opt(request):
    return request.config.getoption("--env_opt")


# 获取--run_level参数值
@pytest.fixture(scope="session")
def run_level(request):
    return request.config.getoption("run_level")


@pytest.fixture(scope="session", autouse=True)
def start_appium_desired(env_opt, run_level):
    print(f"\n{env_opt}")
    print(f"\n{run_level}")

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

在用例中获取定义命令的值

class TestDemo:

    def test_001(self, env_opt):
        print(f"\nenv_opt参数值：{env_opt}")

    def test_002(self, run_level):
        print(f"\nrun_level参数值：{run_level}")

调用定义的命令

import pytest
import os

if __name__ == '__main__':
    env = {
        "host": "127.0.0.1",
        "port": "6789"
        }

    pytest.main(["-vs", "--alluredir=./temp", f"--env_opt={env}"])
    os.system("allure generate ./temp -o ./report/ --clean")

执行结果