快捷導(dǎo)航

pytest解讀fixtures中yield與addfinalizer區(qū)別

更新時間：2022年06月06日 08:53:10 作者：把蘋果咬哭的測試筆記

這篇文章主要為大家介紹了pytest官方解讀fixtures中yield與addfinalizer區(qū)別，有需要的朋友可以借鑒參考下，希望能夠有所幫助，祝大家多多進(jìn)步，早日升職加薪

前言

在上一章中，文末留下了一個坑待填補(bǔ)，疑問是這樣的：

目前從官方文檔中看到的是

We have to be careful though, because pytest will run that finalizer once it’s been added, even if that fixture raises an exception after adding the finalizer.

一旦添加了終結(jié)器，pytest便會執(zhí)行。

但是，當(dāng)我嘗試在setup代碼中進(jìn)行拋錯，終結(jié)器的代碼卻并沒有執(zhí)行。嘗試搜索外網(wǎng)暫時也沒得到有效的幫助，只能在GitHub上向pytest提了issue了，這里算是埋下一個坑，待后續(xù)解決。

一、問題回顧

其實(shí)說到底還是我理解的不對，可能當(dāng)時自己處在疑問中難免就會陷入進(jìn)死循環(huán)，后來在github上經(jīng)過別人提點(diǎn)方才醒悟。先來看下當(dāng)時我嘗試演示出上述結(jié)果的代碼，也就是：setup代碼中進(jìn)行拋錯，終結(jié)器的代碼卻并沒有執(zhí)行。

代碼分為2部分，一個是fixture函數(shù)代碼，另一個則是測試用例。代碼是不能直接copy出來運(yùn)行的，是我在項(xiàng)目的用例中進(jìn)行改造的，在這里僅僅幫助說明意思。

# content of conftest.py
@pytest.fixture()
def init_data_allot_task(request):
    query_sql = """
    SELECT id FROM `sm_purchase_allot` WHERE `status`!=5
    """
    db = DB()
    data = db.fetch_one(query_sql)
    db.close()
    def demo_finalizer():
        print("running finalizer code...")
    request.addfinalizer(demo_finalizer)
    return data

# content of testcase
...
def test_allot_detail(init_data_allot_task):
    """
    """
    payload = {
          "allotId": init_data_allot_task[0]
        }
    r = requests.post(QA_URL + API_URL, json=payload, headers=HEADER)
    result = r.json()
    assert result["result"] == "ok"
    assert result["errmsg"] == "success"
    assert len(result["row"]["taskListOfPage"]["resultData"]) &gt; 0

最開始我想做的是，在fixture函數(shù)中，讓代碼db = DB()拋出一個mysql連接超時的錯誤，然后就能在控制臺中看到"running finalizer code..."的輸出。

但是我執(zhí)行后，并沒有看到預(yù)期的輸出，說明setup代碼拋錯后，addfinalizer代碼并沒有執(zhí)行。

最后經(jīng)過github上朋友指點(diǎn)后，發(fā)現(xiàn)還是我自己理解錯了。

二、問題解決

還是來看下官方的原文：

We have to be careful though, because pytest will run that finalizer once it’s been added, even if that fixture raises an exception after adding the finalizer.

這句話意思其實(shí)是說，當(dāng)finalizer 一旦添加成功后，pytest就會去執(zhí)行它。就算是fixture函數(shù)在添加了finalizer之后拋出了異常。

按照這樣理解的話，那我在fixture函數(shù)中的代碼就有問題了。因?yàn)閐b = DB()代碼在request.addfinalizer(demo_finalizer)之前就拋錯了，那么實(shí)際上并沒有執(zhí)行到添加終結(jié)器的這行代碼，所以終結(jié)器都還沒添加成功，又怎么會去執(zhí)行呢？

終于我明白過來了，于是調(diào)整了代碼順序，把request.addfinalizer(demo_finalizer)放到前面去，然后再接上fixture的代碼：

# content of conftest.py
@pytest.fixture()
def init_data_allot_task(request):
    query_sql = """
    SELECT id FROM `sm_purchase_allot` WHERE `status`!=5 
    """
    def demo_finalizer():
        print("running finalizer code...")
    request.addfinalizer(demo_finalizer)
    print("running setup code...")
    db = DB()
    data = db.fetch_one(query_sql)
    db.close()
    return data

如此來看，我們會先看到"running setup code..."的輸出，然后看到mysql拋錯，最后仍然可以看到"running setup code..."的輸出。

運(yùn)行代碼驗(yàn)證一下：