【pytest】teardown里的yield和addfinalizer

在之前介绍pytest中的fixture用法的文章中https://zhuanlan.zhihu.com/p/87775743，提到了teardown的实现。
最近在翻pytest官方文档的时候，又发现了addfinalizer这个函数，跟yield一样，也可以实现在case结束后运行关键字之后的代码。那今天就来捋一下这2者的用法和区别。

一、yield

再来简单回顾下pytest里的setUp和tearDown的用法，我们可以看到，下方代码里有三个case用例，分别是test_开头。
而在demo_fixture函数里，有一个yield关键字。那么在yield之前的代码，会在case之前执行，yield之后的代码，则会在case运行结束后执行。

#yield_finalizer_demo.py

import pytest

@pytest.fixture()
def demo_fixture():
    print("\n这个fixture在每个case前执行一次")
    yield
    print("\n在每个case完成后执行的teardown")

def test_01(demo_fixture):
    print("\n===执行了case: test_01===")

def test_02(demo_fixture):
    print("\n===执行了case: test_02===")

def test_03(demo_fixture):
    print("\n===执行了case: test_03===")

好了，现在我3个case都传入这个demo_fixture，运行一下，看下效果。这3个红色框中标出的分别就是每一个case执行前后的代码输出，符合我们的预期。

二、addfinalizer

现在，我们再来看addfinalizer，这里我姑且叫它终结器。在用法上，addfinalizer跟yield是不同的，需要你去注册作为终结器使用的函数。
这里还是用上方的代码去修改，去掉yield关键字，增加一个新的函数demo_finalizer，并且注册成终结函数：

import pytest


@pytest.fixture()
def demo_fixture(request):
    print("\n这个fixture在每个case前执行一次")
    def demo_finalizer():
        print("\n在每个case完成后执行的teardown")

    #注册demo_finalizer为终结函数    
    request.addfinalizer(demo_finalizer)

def test_01(demo_fixture):
    print("\n===执行了case: test_01===")

def test_02(demo_fixture):
    print("\n===执行了case: test_02===")

def test_03(demo_fixture):
    print("\n===执行了case: test_03===")

接下来，运行一下，可以看到结果与使用yield的时候一致。

三、yield与addfinalizer的区别

那么，除了在使用上的区别之外，yield与addfinalizer还有什么不同呢？

1. addfinalizer可以注册多个终结函数。

import pytest


@pytest.fixture()
def demo_fixture(request):
    print("\n这个fixture在每个case前执行一次")
    def demo_finalizer():
        print("\n在每个case完成后执行的teardown")
    def demo_finalizer2():
        print("\n在每个case完成后执行的teardown2")
    def demo_finalizer3():
        print("\n在每个case完成后执行的teardown3")

    #注册demo_finalizer为终结函数    
    request.addfinalizer(demo_finalizer)
    request.addfinalizer(demo_finalizer2)
    request.addfinalizer(demo_finalizer3)

def test_01(demo_fixture):
    print("\n===执行了case: test_01===")

def test_02(demo_fixture):
    print("\n===执行了case: test_02===")

def test_03(demo_fixture):
    print("\n===执行了case: test_03===")

在代码里增加demo_finalizer2，demo_finalizer3，这2个终结函数。运行一下：

可以看到，注册的3个函数都被执行了，但是要注意的是执行顺序，与注册的顺序相反。

2. 当setUp的代码执行错误，addfinalizer依旧会执行

这里接官方文档上的例子说明一下：

@pytest.fixture
def equipments(request):
    r = []
    for port in ('C1', 'C3', 'C28'):
    equip = connect(port)
    request.addfinalizer(equip.disconnect)
    r.append(equip)
    return r

比如，C1，C3，C28这3个端口连接，如果C28这个端口失败了，这时候会抛出一个连接异常，但是在执行teardown关闭连接的时候，C1和C3的依然可以正常关闭。