【pytest系列】- fixture测试夹具详解

如果想从头学起pytest，可以去看看这个系列的文章！

https://www.cnblogs.com/miki-peng/category/1960108.html

fixture的优势

​ pytest框架的fixture测试夹具就相当于unittest框架的setup、teardown，但相对之下它的功能更加强大和灵活。

• 命名方式灵活，不限于unittest的setup、teardown
• 可以实现数据共享，多个模块跨文件共享前置后置
• 可以实现多个模块跨文件使用一个session来完成多个用例
• 可以实现unittest不能实现的功能，比如unittest中的测试用例和测试用例之间是无法传递参数和数据的，但是fixture却可以解决这个问题

fixture定义及调用

​ 关键代码：@pytest.fixture()，用于声明函数是处理前置后置的测试夹具函数。用法如下：

@pytest.fixture()
def my_fixture():	# 记住这个夹具名
    print("我的测试夹具")

​ 测试夹具已经定义好了，那测试用例如何调用呢？有以下三种方式：

• 方式一：将fixture名称作为参数传给测试用例，可以传多个fixture，按先后顺序执行
• 方式二：装饰器：@pytest.mark.usefixtures(fixture_name)
  • 使用在类上，代表这个类所有测试用例都会调用这个fixture
  • 使用在用例上，代表只有这个用例调用这个fixture
  • 同样可以使用多个fixture，按先后顺序执行
  • 如果fixture有返回值，用这个装饰器是无法获取到返回值的，也就无法给用例传递数据，只能用方法一
• 方式三：fixture设置autouse=True自动调用，同样fixture有返回值时，无法获取返回值

import pytest

@pytest.fixture()
def my_fixture():	# 记住这个夹具名
    print("我的测试夹具")

# 方式一
def test_fix1(my_fixture):
    print("这是测试用例1111")
    print("-------分割线------")

# 方式二
# 类中应用
@pytest.mark.usefixtures("my_fixture")
class TestLogin:

    def test_fix2(self):
        print("这是测试用例2222")
        print("-------分割线------")

    def test_fix3(self):
        print("这是测试用例3333")
        print("-------分割线------")

# 测试用例应用
@pytest.mark.usefixtures("my_fixture")
def test_fix4():
    print("这是测试用例4444")
    print("-------分割线------")

# 方式三
def test_fix5():  # 未声明使用测试夹具
    print("这是测试用例5555")

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

​ 运行结果：

Testing started at 23:12 ...
C:\software\python\python.exe ...

test.py 我的测试夹具
.这是测试用例1111
-------分割线------
我的测试夹具
.这是测试用例2222
-------分割线------
我的测试夹具
.这是测试用例3333
-------分割线------
我的测试夹具
.这是测试用例4444
-------分割线------
.这是测试用例5555
                                                              [100%]

============================== 5 passed in 0.02s ==============================

Process finished with exit code 0

​ 为什么方式三没有用到测试夹具呢，因为还没有设置测试夹具自动调用。fixture里面有个参数autouse(自动使用的意思)，默认是False，当设置为True时，用例就会自动调用该fixture功能，这样的话写用例时就不用每次都去传参了。

@pytest.fixture(autouse=True)	# 设置用例自动调用该fixture
def my_fixture():
    print("我的测试夹具")

fixture作用域

​ 在unittest中还有一个setUpClass和tearDownClass，是作用于类上，在每个测试用例类执行前去执行前置，用例类执行完再执行后置，pytest的fixture同样有这样的作用域，且使用更广泛更灵活。

​ 关键代码：@pytest.fixture(scope='作用范围')，参数如下：

• function：默认作用域，每个测试用例都运行一次
• class：每个测试类只执行一次
• module：每个模块只执行一次
• package：每个python包只执行一次
• session：整个会话只执行一次，即运行项目时整个过程只执行一次

​ 优先级：session > package > module > class > function，优先级高的会在低的fixture之前先实例化。

​ 相同作用域的fixture遵循函数中声明的先后顺序，并遵循fixture之间的依赖关系，如在fixture_A里面依赖的fixture_B优先实例化，然后再到fixture_A实例化

​ 我们前面举例fixture的使用时都是使用的默认作用域，下面举例一下作用域为class的场景：

# 因为用于演示，因此测试夹具直接写在py文件中
import pytest
from selenium import webdriver

@pytest.fixture(scope='class')
def my_fixture():
    """前置条件"""
    print("前置条件-启动浏览器")
    driver = webdriver.Chrome()
    yield driver
    driver.quit()
    print("后置条件-关闭浏览器")

class TestCase:

    def test_case01(self, my_fixture):    # 这里通过参数传入my_fixture函数，用例执行前会先去执行my_fixture
        driver = my_fixture    # my_fixture不需要加括号
        driver.get('http://www.baidu.com')
        print('第一个用例')
        assert 1 == 1

    def test_case02(self, my_fixture):    # 这里通过参数传入my_fixture函数，用例执行前会先去执行my_fixture
        driver = my_fixture    # my_fixture不需要加括号
        driver.get('http://www.cnblogs.com/')
        print('第二个用例')
        assert 1 == 1

if __name__ == '__main__':
    pytest.main(['test.py', '-s'])

​ 运行结果如下，从运行结果可以看出，整个类只打开了一次浏览器。

C:\software\python\python.exe D:/learn/test.py
============================= test session starts =============================
platform win32 -- Python 3.7.3, pytest-5.2.2, py-1.8.0, pluggy-0.13.0
rootdir: D:\learn
plugins: html-2.0.0, metadata-1.8.0
collected 2 items

test.py 前置条件-启动浏览器
第一个用例
.第二个用例
.后置条件-关闭浏览器

============================== 2 passed in 9.76s ==============================

Process finished with exit code 0

​ 前面也提到了fixture的另外一个参数autouse，当autouse=True时，用例会自动执行测试夹具，但无法获取fixture的返回值，如上示例返回的driver就无法传递给用例了。

​ 如果你想让用例自动执行测试夹具且希望driver等参数可以返回给用例时，可以试一下在测试夹具中使用yield关键字返回值时把值存储到临时变量中，再让用例去取临时变量中的值，这里不作举例，以下是autouse=True的一个简单示例：（yield关键字在后面的章节会讲解）

# 因为用于演示，因此测试夹具直接写在py文件中
import pytest
from selenium import webdriver

@pytest.fixture(scope='class', autouse=True)	# 所有类自动执行该测试夹具
def my_fixture():
    """前置条件"""
    print("前置条件-启动浏览器")
    driver = webdriver.Chrome()
    yield driver
    driver.quit()
    print("后置条件-关闭浏览器")

class TestCase:

    def test_case01(self):	# 不需要传入测试夹具
        print('第一个用例')
        assert 1 == 1

    def test_case02(self):
        print('第二个用例')
        assert 1 == 1

if __name__ == '__main__':
    pytest.main(['test.py', '-s'])

​