【pytest】使用parametrize将参数化变量传递到fixture

简介: 【pytest】使用parametrize将参数化变量传递到fixture

分享一个关于在pytest中,如何将测试用例文件中的变量传递到fixture函数。


一、交代应用场景


  • 目前组内的项目,在根目录下是有一个conftest.py文件的,这里有个生成api token的fixture函数,就叫它gen_token()吧。
  • 每个case包下,也会有个conftest.py,用于存放适用于本模块下测试用例的fixture函数,比如有个叫setup_before()
  • 因为拿token是请求接口的前提,所以在case里,比如有个test_case()里,要传顶层的fixture函数,也就是这样test_case(gen_token)
  • 顶层的gen_token(),是需要3个传参的。因为不同case可能涉及到的生成不同用户的token,所以我们把这个参数放在了case文件里。


ok,大背景是这样的。


现在有小伙伴来需求了,她要在setup_before()里去造数,通过请求另一个接口,这个请求也需要使用token。


那么,问题也就可以转化为:


  • 要将case文件里的参数,传递到fixture函数中。
  • gen_token()里返回的值,setup_before()和test_case()里都要拿到。


二、使用@pytest.mark.parametrize、以及fixture的调用来解决


这里把实际代码抽象一下,转化为简易代码,方便演示和理解:


# 目录结构
-- /demo_top
  -- /demo_sub
      __init__.py
      conftest.py
      test_case.py
  __init__.py
  conftest.py


以下分别


/demo_top/conftest.py/demo_top/demo_sub/conftest.py/demo_top/demo_sub/test_case.py的内容。


1. /demo_top/conftest.py


# content of /demo_top/conftest.py
import pytest
@pytest.fixture()
def gen_token(request):
    params = request.param
    print("\n根目录下gen_token()拿到的参数:", params)
    if params[0] + params[1] == 5:
        return "api_token"
    else:
        return None


这里,模拟生成token的fixture函数,当传过来的值相加等于5,就会返回"api_token",否则返回None


2. /demo_top/demo_sub/conftest.py


# content of /demo_top/demo_sub/conftest.py
import pytest
@pytest.fixture()
def setup_before(request, gen_token):
    print("执行子级setup_before,拿到的传参:", request.param)
    print("执行子级setup_before,拿到gen_token的返回值:", gen_token)
    if gen_token:
        yield "造数完成"
        print("测试用例test_case执行完毕,清理测试数据")
    else:
        pytest.skip("跳过")


这里模拟了给测试用例造数据的fixture函数,如果没拿到token的话,就跳过测试用例。


3. /demo_top/demo_sub/test_case.py


# content of /demo_top/demo_sub/test_case.py
import pytest
test_param = [(1, 4)]
@pytest.mark.parametrize("gen_token", test_param, indirect=True)
@pytest.mark.parametrize("setup_before", test_param, indirect=True)
def test_case1(gen_token, setup_before):
    print("\n测试用例里拿到的gen_token返回值:", gen_token)
    print("测试用例里拿到的setup_before返回值:", setup_before)
    print("执行测试用例test_case1...")
if __name__ == '__main__':
    pytest.main(['-s', 'test_case.py'])


这是测试用例文件了,里面有个测试函数test_case1,因为它需要用到2个fixture函数返回的值,所以gen_token, setup_before都请求。


参数传递


  • @pytest.mark.parametrize:使用pytest内置的parametrize,来把参数传递给目标fixture函数,你希望把参数传递给哪个fixture函数就加哪个。比如这里的gen_tokensetup_before,注意名称与fixture名称一致。
  • indirect=True:作用是让parametrize中的参数名称,也就是"gen_token"当成函数执行,并且后面的参数值test_param,作为"gen_token"的传参。
  • request.param:接受传参的fixture函数,使用request.param来获取值。


fixture调用fixture


fixture之间的相互调用,在之前的文章里已经有过详述了。既然这里setup_before依赖gen_token,之间传递调用即可setup_before(request, gen_token)


在各环节做了些print打印出信息,帮助理解执行过程。


test_case.py                                                            [100%]
============================== 1 passed in 0.08s ==============================
根目录下gen_token()拿到的参数: (1, 4)
执行子级setup_before,拿到的传参: (1, 4)
执行子级setup_before,拿到gen_token的返回值: api_token
.
测试用例里拿到的gen_token返回值: api_token
执行测试用例test_case1...
测试用例test_case执行完毕,清理测试数据
Process finished with exit code 0


再看下gen_token不返回token的情况,改下传参test_param = [(2, 4)]


test_case.py                                                            [100%]
============================= 1 skipped in 0.08s ==============================s
根目录下gen_token()拿到的参数: (2, 4)
执行子级setup_before,拿到的传参: (2, 4)
执行子级setup_before,拿到gen_token的返回值: None
Skipped: 跳过
Process finished with exit code 0


测试用例不执行。

相关文章
|
人工智能 JSON 自然语言处理
基于阿里云通义千问的AI模型应用开发指南
阿里云通义千问是阿里巴巴集团推出的多模态大语言模型平台,提供了丰富的API和接口,支持多种AI应用场景,如文本生成、图像生成和对话交互等。本文将详细介绍阿里云通义千问的产品功能,并展示如何使用其API来构建一个简单的AI应用,包括程序代码和具体操作流程,以帮助开发者快速上手。
4600 3
|
机器学习/深度学习 数据可视化 JavaScript
Plotly可视化介绍
Plotly可视化介绍
351 0
|
7月前
|
Oracle Java 关系型数据库
Java 17 采用率增长 430%
1995年,Sun Microsystems发布Java语言,推动现代多媒体应用发展。凭借“一次编写,到处运行”的优势,Java迅速成为主流编程语言。New Relic最新发布的《2023年Java生态系统现状》报告显示,Java 11以超56%的使用率稳居榜首,Java 8仍占近33%。尽管Oracle每半年更新一次Java版本,但开发者更倾向使用长期支持(LTS)版本。Java 17的采用率在过去一年增长430%,潜力巨大。此外,Amazon已成为最受欢迎的JDK供应商,市场份额达31%。容器化应用也已成为主流,70%的Java应用来自容器。
pip install -U weditor 安装报错 error: subprocess-exited-with-error
pip install -U weditor 安装报错 error: subprocess-exited-with-error
719 0
|
6月前
|
算法 机器人 定位技术
【机器人路径规划】基于流场寻路算法(Flow Field Pathfinding)的机器人路径规划(Python代码实现)
【机器人路径规划】基于流场寻路算法(Flow Field Pathfinding)的机器人路径规划(Python代码实现)
414 4
|
8月前
|
Web App开发 开发框架 .NET
Playwright 自动化测试系列(6)| 第三阶段:测试框架集成​指南:参数化测试 + 多浏览器并行执行
Pytest 与 Playwright 集成可提升自动化测试效率,支持参数化测试、多浏览器并行执行及统一报告生成。通过数据驱动、Fixture 管理和并行优化,显著增强测试覆盖率与执行速度,适用于复杂 Web 应用测试场景。
|
Python
使用Python计算字符串的SHA-256散列值
使用Python计算字符串的SHA-256散列值
438 7
|
监控
USB 2.0 规范摘录1
USB 2.0 规范摘录
531 1
|
编译器 图形学 C语言
SSE2 指令集简介以及与SSE的差别
SSE2,Intel在2001年为Pentium 4引入的扩展,增强了SSE的功能,添加了对双精度浮点和64位整数运算的支持,新增144条指令,提升向量处理能力。SSE2的C代码示例展示了如何通过`_mm_add_ps`加速向量加法。启用SSE2编译器支持可优化处理图像、音频和视频等大量计算任务的性能。
|
SQL 存储 分布式计算
AnalyticDB MySQL带你深入浅出SQL优化器原理
SQL优化器是数据库、数据仓库、大数据等相关领域中最复杂的内核模块之一,它是影响查询性能的关键因素。比如大家熟知的开源产品 MySQL、PostgreSQL、Greenplum DB、Hive、Spark、Presto,都有自己的优化器。本文将由浅入深地带读者了解其中技术原理。