pytest接口测试（九）--环境变量(自定义命令行参数)

前言

我们希望只维护一个host地址，这样测试环境部署环境发生改变，或者有多套测试环境的时候，我们可以快速切换测试环境。环境变量，起到全局的变量作用

自定义命令行参数

定义一个全局的conftest.py文件

import os
import pytest
# 添加命令行参数
def pytest_addoption(parser):
    parser.addoption(
        "--cmdhost",
        action="store",
        # default: 默认值，命令行没有指定host时，默认用该参数值
        default="http://xx.xx.xx.xx:xxxx",
        help="test case project host address"
    )

# autouse=True自动执行该前置操作
@pytest.fixture(scope="session", autouse=True)
def host(request):
    '''获取命令行参数'''
    # 获取命令行参数给到环境变量
    os.environ["host"] = request.config.getoption("--cmdhost")
    print("当前用例运行测试环境:%s"%os.environ["host"])

定义完成后在项目目录下，输入pytest -h帮助文档中有刚刚设置的参数

如果想更换运行的环境变量，输入pytest --cmdhost="新的host地址"即可