pytest解讀fixtures中yield與addfinalizer區別是什么

本文小編為大家詳細介紹“pytest解讀fixtures中yield與addfinalizer區別是什么”，內容詳細，步驟清晰，細節處理妥當，希望這篇“pytest解讀fixtures中yield與addfinalizer區別是什么”文章能幫助大家解決疑惑，下面跟著小編的思路慢慢深入，一起來學習新知識吧。

一、問題回顧

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

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

# 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"]) > 0

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

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

最后經過github上朋友指點后，發現還是我自己理解錯了。

二、問題解決

還是來看下官方的原文：

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. 

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

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

終于我明白過來了，于是調整了代碼順序，把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..."的輸出。

運行代碼驗證一下：

讀到這里，這篇“pytest解讀fixtures中yield與addfinalizer區別是什么”文章已經介紹完畢，想要掌握這篇文章的知識點還需要大家自己動手實踐使用過才能領會，如果想了解更多相關內容的文章，歡迎關注億速云行業資訊頻道。