脚本之家 服务器常用软件
快捷导航
您的位置：首页 → 脚本专栏 → python → Pytest多环境切换

Pytest多环境切换的常见方法介绍

 更新时间：2025年04月11日 15:00:16   作者：朱公子的Note  
Pytest 作为自动化测试的主力框架,如何实现本地、测试、预发、生产环境的灵活切换,本文总结了通过pytest框架实现自由环境切换的几种方法,大家可以根据需要进行选择

你是否也遇到过这种情况：本地测试通过，一到测试环境就翻车？环境变量错乱、接口地址混乱、数据源配置丢失……这些「环境切换」问题简直像定时炸弹，随时引爆你的测试流程！

测试人员每天都跟不同的环境打交道，比如线上环境，测试环境，预上线环境等等，那么作为自动化测试人员写的代码，我们也要具备能自由切换环境的能力，那么今天小编就给大家聊一下，如何能让我们python语言写的测试用例可以自由切换到不同的环境下面去运行呢？

Pytest 作为自动化测试的主力框架，如何实现本地、测试、预发、生产环境的灵活切换？有没有一种通用、实用、可复用的方案？

“Pytest如何实现环境切换，让测试适应不同场景？”这个问题曾让我挠头。配置文件怎么写？参数怎么传？能不能自动化切换而不用手动改代码？带着这些疑问，我开始摸索Pytest的环境管理功能。从简单的命令行参数到复杂的配置文件，我发现它不仅灵活，还能无缝对接团队协作。那么，具体该怎么操作呢？接下来，让我们一起进入实战，揭开Pytest环境切换的奥秘。

今天给他大家聊一下通过pytest框架实现自由环境切换的几种方法。

1.pytest-base-url

pytest有一个插件叫pytest-base-url ，是管理base_url非常好的一款插件，文档资料链接：https://pypi.org/project/pytest-base-url/

下载方式：

pip install pytest-base-url

案例如下：

import requests
 
 
def test_demo(base_url):
    resp = requests.get(base_url)
    status_code = resp.status_code
    assert status_code == 200

第1种使用方式是终端添加--base-url这个命令，如下所示：

第2种使用方式是在pytest.ini配置文件种去配置base_url，然后自动读取url的数据，这样就不用添加--base-url这个命令行参数了：

命令行直接运行pytest 即可。

2.hooks函数

pytest有个hooks函数，可以自定义命令行参数，一般在conftest.py中去引用。

Conftest.py中写pytest_addoption这个hooks函数，可以自定义命令行参数，base_url只能对一个url地址进行传递，那么有的项目不仅需要多项目请求的url地址进行不同环境的切换，还需要对mysql的url地址进行不同的切换，这个时候就可以用自定义命令，定义不同的命令行参数，这样我们在执行pytest的时候就可以自由进行传递。

代码如下：

通过pytest_addoption这个hooks函数，设置了两个命令行参数，一个是--mysql-url,另一个是--project-url，help后面是对这两个命令行的解释，解释说明这两个命令行代表什么含义

import pytest
 
 
def pytest_addoption(parser):
    parser.addoption(
        "--mysql-url", action="store", help="这里可以传递mysql的url地址"
    )
    parser.addoption(
        "--project-url", action="store", help="这里可以传递项目的url地址"
    )

通过fixture函数中的request参数去读取命令行中传递的参数，赋值给一个变量，就可以在pytest中随便使用。

@pytest.fixture(scope='class')
def get_url(request):
    mysql_url = request.config.getoption('--mysql-url')
    project_url = request.config.getoption('--project-url')
    return mysql_url, project_url

项目代码如下：

fixture函数的调用，直接把fixture的名字当成参数传递到测试用例里面即可

import requests
 
 
def test_demo(get_url):
    mysql_url = get_url[0]
    print('mysql_url的地址是{}'.format(mysql_url))
    
    project_url = get_url[1]
    resp = requests.get(project_url)
    status_code = resp.status_code
    
    assert status_code == 200

运行测试的命令为：

pytest -s --mysql -url 192.168.1.11--project-url http://www.example.com

如下图所示：

3.yml和fixture

通过 yml文件 和 fixture 进行配合使用，首先在项目目录下面创建一个conf.yml，然后里面去手动配置现在要运行的环境，截图如下：

然后在创建一个conftest.py，里面定义两个fixture函数，一个用来读取conf.yml里面的内容，另一个用来配置具体的环境信息

比如qa环境，release环境，然后通过conf.yml里面配置的qa还是release，来读取project_url 和 mysql_url，截图如下：

import pytest
import yaml
 
 
@pytest.fixture(scope="session")
def config():
    with open('./conf.yml', 'r', encoding='utf-8') as f:
        conf = yaml.load(f.read(), Loader=yaml.FullLoader)
    return conf
 
 
@pytest.fixture(scope="session")
def env_vars(config):
    env = config["env"]
    mapping = {
        "qa": {
            "project_url": "http://121.42.15.146:9090",
            "mysql_url": "192.168.1.1"
        },
        "release": {
            "project_url": "https://release.com",
            "mysql_url": "127.0.0.1"
        }
    }
    project_url = mapping[env]["project_url"]
    mysql_url = mapping[env]["mysql_url"]
    return mysql_url, project_url

项目代码(还是之前的测试用例)如下：

import requests
 
 
def test_demo(env_vars):
    mysql_url = env_vars[0]
    print('mysql_url的地址是{}'.format(mysql_url))
    
    project_url = env_vars[1]
    resp = requests.get(project_url)
    status_code = resp.status_code
    
    assert status_code == 200

命令行运行方式，正常运行即可，截图如下:

结论

随着 CI/CD 流水线普及，多环境并存已成常态。测试人员面临的环境复杂度越来越高，若不能实现配置统一、切换自如，将严重拖慢测试效率，甚至引发线上事故。

环境切换看似是小问题，实则影响自动化测试流程的稳定性与扩展性。掌握 Pytest 中的配置管理，就等于掌握了自动化测试的方向盘。

总的来说，Pytest的环境切换通过命令行参数、配置文件和插件支持，让测试灵活适应各种场景。从单人开发到团队协作，它都能大幅提升效率。这不仅是技术实践，更是现代测试思维的体现。掌握环境切换，你的测试流程将更加顺畅，代码质量也能更上一层楼。

测试环境千万条，配置管理第一条；配置不统一，Bug泪两行。

以上就是Pytest多环境切换的常见方法介绍的详细内容，更多关于Pytest多环境切换的资料请关注脚本之家其它相关文章！

您可能感兴趣的文章:

相关文章

  • Python实战之异步获取中国天气信息

    Python实战之异步获取中国天气信息

    这篇文章主要介绍了如何利用Python爬虫异步获取天气信息，用的API是中国天气网。文中的示例代码讲解详细，感兴趣的小伙伴可以动手试一试
    2022-03-03
  • python通过文件头判断文件类型

    python通过文件头判断文件类型

    这篇文章主要介绍了python通过文件头判断文件类型，需要的朋友可以参考下
    2015-10-10
  • 深入理解 Python 中的多线程 新手必看

    深入理解 Python 中的多线程 新手必看

    你应当将下边的例子运行多次，以便可以注意到线程是不可预测的和线程每次运行出的不同结果。声明：从这里开始忘掉你听到过的关于GIL的东西，因为GIL不会影响到我想要展示的东西
    2016-11-11
  • Python虚拟环境创建和使用方法(使用自带的venv模块)

    Python虚拟环境创建和使用方法(使用自带的venv模块)

    这篇文章主要如何在Python中使用虚拟环境,包括创建、激活、使用、生成requirements.txt文件、卸载包和删除虚拟环境,虚拟环境有助于隔离项目依赖,避免版本冲突,并便于部署,需要的朋友可以参考下
    2024-12-12
  • Python环境使用OpenCV检测人脸实现教程

    Python环境使用OpenCV检测人脸实现教程

    这篇文章主要介绍了Python环境使用OpenCV检测人脸实现教程,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下
    2020-10-10
  • python 基于pygame实现俄罗斯方块

    python 基于pygame实现俄罗斯方块

    这篇文章主要介绍了python 基于pygame实现俄罗斯方块的方法，帮助大家更好的理解和学习使用python，感兴趣的朋友可以了解下
    2021-03-03
  • 使用Python绘制3D堆叠条形图全解析

    使用Python绘制3D堆叠条形图全解析

    在数据可视化的工具箱里,3D 图表总能带来眼前一亮的效果,本文就来和大家聊聊如何使用Python实现绘制3D 堆叠条形图,感兴趣的小伙伴可以了解下
    2025-06-06
  • Python删除Java源文件中全部注释的实现方法

    Python删除Java源文件中全部注释的实现方法

    这篇文章主要介绍了Python删除Java源文件中全部注释的实现方法,涉及Python读取文件、正则匹配、字符串查找、替换等相关操作技巧,需要的朋友可以参考下
    2017-08-08
  • Python读写Excel大数据文件的3种有效方式对比

    Python读写Excel大数据文件的3种有效方式对比

    Python处理Excel大数据有很多方式,不过Excel撑死才104万行,能有多大的数据,本文主要介绍了三种常见方式,感兴趣的小伙伴可以跟随小编一起学习一下
    2025-09-09
  • Python深度学习之使用Albumentations对图像做增强

    Python深度学习之使用Albumentations对图像做增强

    诸如RandomCrop和CenterCrop之类的某些增强功能可能会变换图像,使其不包含所有原始边界框. 本示例说明如何使用名为RandomSizedBBoxSafeCrop的变换来裁剪图像的一部分,但保留原始图像的所有边界框,需要的朋友可以参考下
    2021-05-05

最新评论