文章目录
- 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==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注意事项：
- 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 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后面的代码被执行到了
