pytest---fixture簡單使用

頭號碼甲 2022-08-26 發(fā)布于北京

展開全文

　　上一篇介紹了如何通過setup和teardown來幫助我們做自動化的前置和后置內(nèi)容，那么如果有一個場景，有的用例需要用到登錄的方法，有的卻用不到登錄的方法，這時如果用setup和teardown的方法就不能寫在一個測試類中，但是pytest中還有更好的前置內(nèi)容，不需要每個用例都能用到。

fixture

fixture屬于pytest中的一個方法，fixture是在測試函數(shù)運行前后進(jìn)行執(zhí)行的，fixture的命名不規(guī)范沒有強制要求，以及里面的代碼內(nèi)容可以自己進(jìn)行定制從而滿足多種測試需求，配置測試前的數(shù)據(jù)清理，以及測試完成后的數(shù)據(jù)管理。fixture中的代碼處理測試后的代碼為 yield 測試開始前會執(zhí)行yield前面的代碼（setup），測試完成后將執(zhí)行yield后面的代碼（teardown），在fixture中不是強制必須有yield，比如，我只有前置代碼，沒有后置代碼，那就不需要yield內(nèi)容?；蛘咧挥泻笾么a，沒有前置代碼。無論測試過程中發(fā)生什么錯誤，yield后續(xù)的代碼都會被執(zhí)行

源碼：

def fixture(  # noqa: F811
    fixture_function: Optional[_FixtureFunction] = None,
    *,
    scope: "Union[_Scope, Callable[[str, Config], _Scope]]" = "function",
    params: Optional[Iterable[object]] = None,
    autouse: bool = False,
    ids: Optional[
        Union[
            Iterable[Union[None, str, float, int, bool]],
            Callable[[Any], Optional[object]],
        ]
    ] = None,
    name: Optional[str] = None
) -> Union[FixtureFunctionMarker, _FixtureFunction]:

通過上面的源碼可以看到fixture一共有5個參數(shù)分別是：name，scope，params，autouse，ids。每個參數(shù)在后面都會介紹到，這里安靜就先不做太多介紹。

栗子：

安靜拿開頭說的那個案例，如果一個測試類下用例，有的需要登錄，有的必須要登錄操作

import pytest

@pytest.fixture()
def login():
    print('輸入賬號,輸入密碼')
    print('完成登錄功能?。。?！')
    yield
    print('---退出登錄---')

class Test_Login():

    def test_01(self,login):
        print('這是用例01')
        print('需要用到登錄！')

    def test_02(self):
        print('這是用例02')
        print('不需要登錄！')

    def test_03(self,login):
        print('這是用例03')
        print('這里需要用到登錄！')

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

這里會發(fā)現(xiàn)，我在用例1和用例3中進(jìn)行通過fixture進(jìn)行使用，然后用例1和用例3都執(zhí)行了配置的登錄和退出登錄內(nèi)容。用例2，我沒用使用fixture方法，所以沒用執(zhí)行對應(yīng)代碼。

也可以通過--setup-show的方法來查看詳細(xì)的fixture信息

多個fixture使用

在一個測試用例中可以使用多個fixture的方法，執(zhí)行順序根據(jù)你傳入的順序進(jìn)行執(zhí)行。

import pytest

@pytest.fixture()
def login():
    print('輸入賬號,輸入密碼')
    print('完成登錄功能?。。。?/span>')
    yield
    print('---退出登錄---')

@pytest.fixture()
def add():
    print('測試開始執(zhí)行！')

class Test_Login():

    def test_01(self,login,add):
        print('這是用例01')
        print('需要用到登錄！')

    def test_02(self):
        print('這是用例02')
        print('不需要登錄！')

    def test_03(self,add,login):
        print('這是用例03')
        print('這里需要用到登錄！')

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

結(jié)果可以看出來，用例1：把login方法的fixture放在了前面，先執(zhí)行的login方法后執(zhí)行的add，用例3：add方法在前面，login方法在后面，執(zhí)行也是add先執(zhí)行，login后執(zhí)行。

fixture相互調(diào)用

其實fixture之間也是可以進(jìn)行相互調(diào)用的。

import pytest


@pytest.fixture()
def login():
    print('輸入賬號,輸入密碼')
    print('完成登錄功能！?。?！')
    yield
    print('---退出登錄---')


@pytest.fixture()
def add(login):
    print('測試開始執(zhí)行！')
    yield
    print('測試結(jié)束！')

class Test_Login:

    def test_01(self, add):
        print('------用例01------')

    def test_02(self):
        print('------用例02------')

    def test_03(self,login):
        print('------用例03------')


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

通過上面的執(zhí)行結(jié)果可以看到，我們fixture中的login函數(shù)被add函數(shù)進(jìn)行調(diào)用了，然后在用例中直接執(zhí)行add，login中的內(nèi)容也進(jìn)行了執(zhí)行

yield遇到異常后繼續(xù)執(zhí)行

在前面介紹fixture的時候說過，fixture執(zhí)行過程中，無論遇到什么異常，都會繼續(xù)執(zhí)行yeild后面的代碼（teardown），安靜舉例給大家看看

import pytest

@pytest.fixture()
def login():
    print('輸入賬號,輸入密碼')
    print('完成登錄功能！?。?！')
    yield
    print('---退出登錄---')

class Test_Login():

    def test_01(self,login):
        print('這是用例01')
        print('需要用到登錄！')
        assert 1==2

    def test_02(self):
        print('這是用例02')
        print('不需要登錄！')

    def test_03(self,login):
        print('這是用例03')
        print('這里需要用到登錄！')

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