Pytest之fixture的常见用法

1.前言

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

2.使用fixture执行前置操作

示例：

import pytest

@pytest.fixture()
def fixture():
    print('执行前置操作')

def test_01(fixture):
    assert 1==1

def test_02():
    assert 2==2

if __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.param

def 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==1

def test_02():
    assert 2==2

if __name__ == '__main__':
    pytest.main()

运行结果：

3.使用conftest共享fixture

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

1. conftest.py为固定写法，不能修改名字
2. 使用conftest.py文件方法无需倒入
3. 位于项目的根目录，那么它定义的 fixture 和钩子函数对整个项目的测试文件都有效。
4. 如果 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 pytest


def test_01(driver):
    driver.get('https://www.baidu.com/')
    title = driver.title
    assert title == '百度一下，你就知道'


def test_02(driver):
    driver.get('https://www.bilibili.com/')
    url = driver.current_url
    assert 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 driver
    print('关闭浏览器')
    driver.quit()

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

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

再次执行刚才的测试用例

可以看到有输出语句，说明yield后面的代码被执行到了