Pytest之fixture的常见用法

文章目录

1.前言
2.使用fixture执行前置操作
3.使用conftest共享fixture
4.使用yield执行后置操作

1.前言

在pytest中，fixture是一个非常强大和灵活的功能，用于为测试函数提供固定的测试数据、测试环境或执行一些前置和后置操作等，
与setup和teardown类似。

2.使用fixture执行前置操作

示例：

import pytest@pytest.fixture()
def fixture():print('执行前置操作')def test_01(fixture):assert 1==1def test_02():assert 2==2if __name__ == '__main__':pytest.main()

运行结果中可以看到第一条测试用例执行时，输出了“执行前置操作”：
在这里插入图片描述

定义fixture：使用@pytest.fixture装饰器来定义一个fixture函数。

注意测试用例中传递的fixture是将定义好的函数名传过去的

作用域：fixture的作用域决定了它在测试中的生命周期和应用范围，通过scope参数来指定，有以下几种常见的作用域：

function（默认）：每个测试函数都会调用一次fixture。
class：每个测试类中的所有测试方法共享同一个fixture实例，在测试类的所有测试方法执行前创建，执行完后销毁。
module：在整个测试模块中只创建一次fixture实例，模块中的所有测试函数和测试类共享。
session：在整个测试会话期间只创建一次fixture实例，所有测试模块、测试类和测试函数都共享。

除了之外，可以通过params参数为fixture传递不同的参数值，实现参数化测试。

import pytest@pytest.fixture(params=[1, 2, 3])
def parameter_fixture(request):return request.paramdef test_parameterized(parameter_fixture):print(f"测试参数: {parameter_fixture}")assert parameter_fixture > 0

fixture函数中的变量名必须为request。

运行结果：
在这里插入图片描述

如果觉得每次使用fixture函数的时候都需要将fixture函数当参数传入测试用例中比较麻烦，可以在fixture

import pytest@pytest.fixture(autouse = True)
def test_fixture():print('执行前置操作')def test_01():assert 1==1def test_02():assert 2==2if __name__ == '__main__':pytest.main()

运行结果：
在这里插入图片描述

3.使用conftest共享fixture

在 pytest 中，conftest.py 是一个非常重要的文件，它主要用于在多个测试文件之间共享 fixture、钩子函数等，帮助组织和管理测试代码。
使用conftest.py注意事项：

conftest.py为固定写法，不能修改名字
使用conftest.py文件方法无需倒入
位于项目的根目录，那么它定义的 fixture 和钩子函数对整个项目的测试文件都有效。
如果 conftest.py 位于某个子目录下，那么它定义的内容只对该子目录及其子目录下的测试文件有效。

示例：
创建conftest.py文件在根目录下，并输入以下内容

import pytest
from selenium import webdriver
from selenium.webdriver.chrome.service import Service@pytest.fixture(scope='session')
def driver():driver = webdriver.Chrome()driver.maximize_window()return driver

创建一个文件测试conftest中fixture函数
代码如下：

import pytestdef test_01(driver):driver.get('https://www.baidu.com/')title = driver.titleassert title == '百度一下，你就知道'def test_02(driver):driver.get('https://www.bilibili.com/')url = driver.current_urlassert url == 'https://www.bilibili.com/'if __name__ == '__main__':pytest.main()

执行结果：
在这里插入图片描述
可以看出fixture函数成功运行了，但是上述代码中虽然有前置获取浏览器驱动的操作，但没有关闭浏览器驱动，虽然会自动关闭，但最好还是自己手动释放一下。如果想要执行后置操作，就需要使用yield

4.使用yield执行后置操作

在 pytest 的 fixture 函数里，yield 关键字用于分隔测试的前置和后置操作。在 yield 之前的代码会在测试用例执行前运行，起到初始化环境、准备数据等前置操作的作用；yield 之后的代码会在测试用例执行完毕后运行，用于清理资源、恢复环境等后置操作。

将conftest.py中代码修改一下：

@pytest.fixture(scope='session')
def driver():driver = webdriver.Chrome()driver.maximize_window()print('打开浏览器')yield driverprint('关闭浏览器')driver.quit()

注意这里是yield driver而不是return driver

当 fixture 函数中使用 return driver 时，函数执行到 return 语句就会立即返回 driver 对象并终止函数的执行，不会再执行 return 语句之后的代码。
yield 关键字使 fixture 函数成为一个生成器函数。当执行到 yield driver 时，函数会暂停执行并返回 driver 对象给测试用例使用。当测试用例执行完毕后，fixture 函数会从 yield 语句的下一行继续执行。

再次执行刚才的测试用例
在这里插入图片描述
可以看到有输出语句，说明yield后面的代码被执行到了