Python中pytest命令行实现环境切换
前言
在自动化测试过程中经常需要在不同的环境下进行测试验证，所以写自动化测试代码时需要考虑不同环境切换的情况。pytest钩子函数pytest_addoption可以很好帮我们解决这个痛点。
pytest_addoption(parser, pluginmanager)
注册argparse样式选项和ini样式配置值，在测试运行开始时调用一次。
注意：
由于pytest在启动过程中如何发现插件，因此该函数只能在位于测试根目录的插件或conftest.py文件中实现。
参数
parser（pytest.parser）–若要添加命令行选项，请调用parser.addoption（…）。若要添加ini文件值，请调用解析器.addini（…）。
pluginmanager（pytest.PytestPluginManager）–pytest插件管理器，可用于安装hookspec（）或hookpimpl（），并允许一个插件调用另一个插件的钩子来更改命令行选项的添加方式。
以后可以分别通过配置对象访问选项：
config.getoption（name）来检索命令行选项的值。
config.getini（name）来检索从ini样式文件中读取的值。
config对象通过.config属性在许多内部对象上传递，或者可以作为pytestconfig fixture检索。
在conftest.py文件中定义命令行参数
def pytest_addoption(parser):
"""
添加命令行参数
parser.addoption为固定写法
default 设置一个默认值，此处设置默认值为sit
choices 参数范围，传入其他值无效
help 帮助信息
"""
parser.addoption(
"--env", default="sit", choices=["dev", "sit", "uat"], help="环境参数"
)我们定义了不同环境下的命令参数：dev、sit、uat，我们怎么获取运行的命令行参数呢？
获取命令行参数
@pytest.fixture(scope="session")
def get_env(request):
return request.config.getoption("--env")设置不同环境的全局变量
在不同的测试环境下，URL、用户信息等数据都是不一样的，建议在conftest中给全局变量赋值可以减少代码冗余。
先定义一个数据文件，data_util.py分别获取用户信息和URL信息
def get_env():
env = {
'sit': 'www.baidu.com',
'uat': 'www.hao123.com'
}
return env
def get_user():
users = {
'sit': ['user1', 'pwd1'],
'uat': ['user2', 'pwd2']
}
return users然后在conftest中根据环境设置全局变量值
# 设置不同环境下的全局变量
@pytest.fixture(scope="session")
def set_env(get_env):
if get_env == 'sit':
env_url = data_util.get_env()['sit']
user = data_util.get_user()['sit']
if get_env == 'uat':
env_url = data_util.get_env()['uat']
user = data_util.get_user()['uat']
return {'env_url': env_url, 'user': user}注意fixture的使用范围为整个测试会话。
以下是完整的conftest
import pytest
import data_util
def pytest_addoption(parser):
"""
添加命令行参数
parser.addoption为固定写法
default 设置一个默认值，此处设置默认值为sit
choices 参数范围，传入其他值无效
help 帮助信息
"""
parser.addoption(
"--env", default="sit", choices=["dev", "sit", "uat"], help="环境参数"
)
@pytest.fixture(scope="session")
def get_env(request):
return request.config.getoption("--env")
# 设置不同环境下的全局变量
@pytest.fixture(scope="session")
def set_env(get_env):
if get_env == 'sit':
env_url = data_util.get_env()['sit']
user = data_util.get_user()['sit']
if get_env == 'uat':
env_url = data_util.get_env()['uat']
user = data_util.get_user()['uat']
return {'env_url': env_url, 'user': user}定义测试类及测试方法
注意fixture不能在x-unit风格下的setup\teardown中引用，因此需要使用fixture定义setup、teardown方法才能引用到conftest里的fixture，一般我们在setup方法中初始化环境变量具体如下：
@pytest.fixture()
def class_fixture(set_env):
print('setup_class')
url = set_env.get('env_url')
user = set_env.get('user')
print(url, user)
yield
print('teardown class')这样我们就在测试前把环境信息设置OK了。
测试验证
以下是测试方法
import pytest
pytestmark = pytest.mark.usefixtures("module_fixture")
@pytest.fixture(scope="module", params=["test_fixture"])
def module_fixture(request):
param = request.param
print(" SETUP module", param)
yield param
print(" TEARDOWN module", param)
@pytest.fixture()
def class_fixture(set_env):
print('setup_class')
url = set_env.get('env_url')
user = set_env.get('user')
print(url, user)
yield
print('teardown class')
@pytest.fixture(scope="function", params=[1, 2])
def function_fixture(request):
param = request.param
print(" SETUP function", param)
yield param
print(" TEARDOWN function", param)
@pytest.mark.usefixtures('class_fixture')
class TestFixture:
def test_0(self, function_fixture):
print(" RUN test0 with function_fixture", function_fixture)
def test_1(self, module_fixture):
print(" RUN test1 with module_fixture", module_fixture)
def test_2(self, function_fixture, module_fixture):
print(f" RUN test2 with function_fixture {function_fixture} and module_fixture {module_fixture}")
def test_env(self, get_env):
print(f"The current environment is: get_env")
if __name__ == '__main__':
pytest.main(['-v', '-s','--env=uat', 'test_fixture.py::TestFixture::test_0'])我们首先填的uat命令运行，查看输出：
可以看出输出是正确的，我们再切换成sit试试：
if __name__ == '__main__':
pytest.main(['-v', '-s','--env=uat', 'test_fixture.py::TestFixture::test_0'])
可以看出在不同的命令下获得的测试数据也不一样，这样我们就达到了环境切换的目的了~
到此这篇关于Python中pytest命令行实现环境切换的文章就介绍到这了,更多相关pytest 环境切换内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家！
相关文章
这篇文章主要介绍了Python学习之用pygal画世界地图实例，具有一定借鉴价值，需要的朋友可以参考下。2017-12-12
这篇文章主要介绍了带你一文读懂Python垃圾回收机制,如果对其垃圾回收机制不了解，很多时候写出的 Python 代码会非常低效,需要的朋友可以参考下2023-04-04
本文主要介绍了PyTorch的MNIST数据集，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧2021-06-06
文字转音频可以帮助视觉障碍者通过听取声音来获取信息，也可以帮助人们方便地听取一些长篇文章或学习资料，节省阅读时间和疲劳，这篇文章主要介绍了通过Python的gtts库将文字转为音频的方法,需要的朋友可以参考下2023-05-05
下面小编就为大家带来一篇python字典的常用操作方法小结。小编觉得挺不错的，现在分享给大家，也给大家做个参考，一起跟随小编过来看看吧2016-05-05
Python hashlib模块与subprocess模块使用详细介绍
这篇文章主要介绍了Python hashlib模块与subprocess模块使用，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习吧2022-10-10
Python绘制三角函数图(sin\cos\tan)并标注特定范围的例子
今天小编就为大家分享一篇Python绘制三角函数图(sin\cos\tan)并标注特定范围的例子，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2019-12-12
这篇文章主要为大家介绍了使用python画一个圣诞树实现示例详解，有需要的朋友可以借鉴参考下，希望能够有所帮助，祝大家多多进步，早日升职加薪2023-05-05
这篇文章主要介绍了python 包之 multiprocessing 多进程，首先创建一个进程。实例化 Process 类创建一个进程对象、然后调用它的 start 方法即可生成一个子进程，下文更多相关资料,需要的朋友可以参考一下2022-04-04
这篇文章主要介绍了如何利用python实现图片转化字符画，文章围绕主题展开详细的内容介绍，具有一定的参考价值，需要的小伙伴可以参考一下2022-06-06
最新评论