當我們運行測驗函式時，我們希望確保測驗函式在運行結束后，可以自己清理掉對環境的影響，
這樣的話，它們就不會干擾任何其他的測驗函式，更不會榷訓月累的留下越來越多的測驗資料，
用過unittest的朋友相信都知道teardown這個函式，做的是一樣的事情，那么下面姑且就把這種“善后”作業的代碼
叫做teardown代碼吧，
而pytest中的fixture，也提供了這樣一個非常有用的系統，我們可以在里面定義teardown代碼，
這里可以使用2種方式來實作，分別是yield和addfinalizer
一、yield fixtures（推薦）
1, yield 和 return
在有yield的fixtures函式中，關鍵字yield 可以代替 return，可以把fixture里的一些物件傳遞給呼叫它們的fixture函式或者測驗函式，
就像其他普通的fixture函式一樣，區別僅僅是：
yield替換掉了return- teardown代碼放置在
yield之后
2, yield的執行順序
pytest在執行fixture函式時，會根據fixture函式之間的線性關系順序呼叫的，
但是，當測驗函式運行結束的時候，pytest又會按照之前的順序反方向來執行fixture中yield之后的代碼，
結合示例看下，這里沒有參考官方示例了，手寫一個直觀些的：
import pytest
@pytest.fixture
def fixture_one():
print("\n執行fixture_one")
return 1
@pytest.fixture
def fixture_two(fixture_one):
print("\n執行fixture_two")
yield 2
print("\n執行fixture_two的teardown代碼")
@pytest.fixture
def fixture_adding(fixture_one, fixture_two):
print("\n執行fixture_adding")
result = fixture_one + fixture_two
yield result
print("\n執行fixture_adding的teardown代碼")
def test_demo(fixture_two, fixture_adding):
print("\n執行測驗函式test_demo")
assert fixture_adding == 3
代碼中，fixture中呼叫多個fixture，測驗函式中呼叫多個fixture，通過前面幾章的接觸，
相信大家這時候已經可以梳理出前后呼叫順序了：
- test_demo 測驗函式，先去呼叫fixture函式 fixture_two，然后呼叫 fixture_adding，
- 在fixture函式 fixture_two中，又會去呼叫另一個fixture函式 fixture_one，
- 在fixture函式 fixture_adding中，呼叫了 fixture_one、fixture_two，
所以，fixture函式的先后順序是：fixture_one、fixture_two、fixture_adding，
那么，可以得知測驗結束后的teardown代碼執行順序：fixture_adding、fixture_two，
運行一下代碼，驗證下結果是否符合我們的梳理：
============================= test session starts =============================
platform win32 -- Python 3.6.8, pytest-5.4.3, py-1.9.0, pluggy-0.13.1
rootdir: D:\練習\demo_fixture
plugins: allure-pytest-2.8.32, celery-4.3.0, Faker-4.14.2, base-url-1.4.2, html-2.1.1, metadata-1.10.0collected 1 item
test_module.py
執行fixture_one
執行fixture_two
執行fixture_adding
.
執行測驗函式test_demo
執行fixture_adding的teardown代碼
執行fixture_two的teardown代碼
[100%]
============================== 1 passed in 0.09s ==============================
結果與我們剛才梳理的一致，
但是，值得注意的是，就算是teardown的代碼是按照正確的順序執行，也不能保證代碼能正常執行的，
比如說teardown里的某些代碼執行例外了，導致別的清理動作也沒法執行，
這里就涉及到另一個點了：健壯的fixture結構應該是什么樣子，這個官方檔案另起進行說明，這里同樣，
二、addfinalizer
1.request.addfinalizer把函式變成終結器
在pytest中想要做teardown的處理，除了使用帶有yield的fixture函式，還可以直接添加終結器，
直接來看示例代碼：
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===")
看下運行結果：
============================= test session starts =============================
platform win32 -- Python 3.6.8, pytest-5.4.3, py-1.9.0, pluggy-0.13.1
rootdir: D:\練習\demo_fixture
plugins: allure-pytest-2.8.32, celery-4.3.0, Faker-4.14.2, base-url-1.4.2, html-2.1.1, metadata-1.10.0collected 2 items
test_module.py
這個fixture在每個case前執行一次
.
===執行了case: test_01===
在每個case完成后執行的teardown
這個fixture在每個case前執行一次
.
===執行了case: test_02===
在每個case完成后執行的teardown
[100%]
============================== 2 passed in 0.10s ==============================
Process finished with exit code 0
運行結果可以看出，效果與yield是一致的，這算是一個固定寫法，關于request檔案中也有另外的講解，屆時再分享，
2.request.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")
#注冊demo_finalizer為終結函式
request.addfinalizer(demo_finalizer)
request.addfinalizer(demo_finalizer2)
def test_01(demo_fixture):
print("\n===執行了case: test_01===")
def test_02(demo_fixture):
print("\n===執行了case: test_02===")
if __name__ == '__main__':
pytest.main(['-s', 'test_module.py'])
運行結果：
============================= test session starts =============================
platform win32 -- Python 3.6.8, pytest-5.4.3, py-1.9.0, pluggy-0.13.1
rootdir: D:\練習\demo_fixture
plugins: allure-pytest-2.8.32, celery-4.3.0, Faker-4.14.2, base-url-1.4.2, html-2.1.1, metadata-1.10.0collected 2 items
test_module.py
這個fixture在每個case前執行一次
.
===執行了case: test_01===
在每個case完成后執行的teardown2
在每個case完成后執行的teardown
這個fixture在每個case前執行一次
.
===執行了case: test_02===
在每個case完成后執行的teardown2
在每個case完成后執行的teardown
[100%]
============================== 2 passed in 0.09s ==============================
Process finished with exit code 0
這里要注意的是，多個終結器的情況下，執行的順序是與注冊時候相反的，
3.yield和addfinalizer的區別
目前從官方檔案中看到的是
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.
一旦添加了終結器，pytest便會執行，
但是，當我嘗試在setup代碼中進行拋錯，終結器的代碼卻并沒有執行，
嘗試搜索外網暫時也沒得到有效的幫助，只能在GitHub上向pytest提了issue了，這里算是埋下一個坑，待后續解決，
轉載請註明出處，本文鏈接：https://www.uj5u.com/qita/266935.html
標籤：其他
下一篇：那些年我們一起追過的大佬