Pytest框架之fixture詳解（一）

更新時(shí)間：2022年06月30日 10:18:07 作者：小旭2021

本文詳細(xì)講解了Pytest框架之fixture，文中通過(guò)示例代碼介紹的非常詳細(xì)。對(duì)大家的學(xué)習(xí)或工作具有一定的參考借鑒價(jià)值，需要的朋友可以參考下

pytest的fixture實(shí)現(xiàn)方式一：xunit-style

pytest的xunit-style有三個(gè)級(jí)別的fixture：測(cè)試模塊、測(cè)試類(lèi)、測(cè)試函數(shù)。

1、測(cè)試函數(shù)/方法級(jí)別：每一個(gè)測(cè)試函數(shù)都會(huì)執(zhí)行的前置和后置。

測(cè)試類(lèi)內(nèi)部的測(cè)試方法：

前置函數(shù)名稱(chēng)：setup_method
后置函數(shù)名稱(chēng)：teardown_method

模塊下的測(cè)試函數(shù)：

前置函數(shù)名稱(chēng)：setup_function
后置函數(shù)名稱(chēng)：teardown_function

2、測(cè)試類(lèi)級(jí)別：一個(gè)測(cè)試類(lèi)只執(zhí)行一次前置和后置。

前置函數(shù)名稱(chēng)：setup_class
后置函數(shù)名稱(chēng)：teardown_class

注意：用@classmethod裝飾

3、測(cè)試模塊級(jí)別：一個(gè)測(cè)試模塊只執(zhí)行一次前置和后置。

前置函數(shù)名稱(chēng)：setup_module
后置函數(shù)名稱(chēng)：teardown_module

from selenium import webdriver
from time import sleep
from random import randint
?
?
def setup_module():
    print("====  模塊級(jí)的 setup 操作  ====")
?
?
def teardown_module():
    print("====  模塊級(jí)的 teardown 操作  ====")
?
?
def test_random():
    assert randint(1, 5) == 3
?
?
class TestWeb:
?
    @classmethod
    def setup_class(cls):
        print("====  測(cè)試類(lèi)級(jí)的 setup 操作  ====")
?
    @classmethod
    def teardown_class(cls):
        print("====  測(cè)試類(lèi)級(jí)的 teardown 操作  ====")
?
    def setup_method(self):
        print("====  測(cè)試用例級(jí)的 setup 操作  ====")
        self.driver = webdriver.Chrome()
?
    def teardown_method(self):
        print("====  測(cè)試用例級(jí)的 teardown 操作  ====")
        self.driver.quit()
?
    def test_search(self):
        self.driver.get("https://www.baidu.com/")
        self.driver.find_element_by_id("kw").send_keys("檸檬班")
        self.driver.find_element_by_id("su").click()
        sleep(1)

pytest也支持運(yùn)行unittest的測(cè)試用例。支持unittest以下特性：

@unittest.skip
setUp/tearDown;
setUpClass/tearDownClass;
setUpModule/tearDownModule;

pytest的fixture實(shí)現(xiàn)方式二：fixture機(jī)制

通過(guò)@pytest.fixture裝飾器來(lái)定義fixture。一個(gè)函數(shù)被@pytest.fixture裝飾，那么這個(gè)函數(shù)就是fixture。

使用fixture時(shí)，分為二個(gè)部分：fixture定義、fixture調(diào)用。

除此之外，還有fixture的共享機(jī)制，嵌套調(diào)用機(jī)制。

1、定義fixture。

1）fixture通過(guò)函數(shù)實(shí)現(xiàn)。

2）使用@pytest.fixture進(jìn)行裝飾

import pytest
?
@pytest.fixture
def init():
    pass

3）前置準(zhǔn)備工作代碼和后置清理工作代碼，都寫(xiě)在一個(gè)函數(shù)里面。

4）通過(guò)yeild關(guān)鍵字，區(qū)分前置代碼和后置代碼。yeild之前的代碼為前置代碼，yeild之后的代碼為后置代碼

在實(shí)際應(yīng)用場(chǎng)景當(dāng)中，可以只有前置準(zhǔn)備工作代碼，也可以只有后置清理工作代碼。

import pytest
?
?
@pytest.fixture
def init():
    print("用例執(zhí)行之前，執(zhí)行的代碼")  # 前置代碼
    yield
    print("用例執(zhí)行之后，執(zhí)行的代碼")  # 后置代碼
  
?
@pytest.fixture
def init2():
    print("用例執(zhí)行之前，執(zhí)行的代碼")  # 只有用例執(zhí)行之前的前置準(zhǔn)備代碼
     
?
@pytest.fixture
def init3():
    yield
    print("用例執(zhí)行之后，執(zhí)行的代碼")  # 只有用例執(zhí)行之后的后置清理代碼

5）fixture有4個(gè)作用域：測(cè)試會(huì)話(session)、測(cè)試模塊(module)、測(cè)試類(lèi)(class)、測(cè)試用例(function)

測(cè)試會(huì)話：pytest執(zhí)行測(cè)試用例的整個(gè)過(guò)程，稱(chēng)為會(huì)話。

比如pytest收集到了100條用例并執(zhí)行完成，這個(gè)過(guò)程稱(chēng)為測(cè)試會(huì)話。

設(shè)置fixture的作用域：通過(guò)@pytest.fixture(scope=作用域)來(lái)設(shè)置。默認(rèn)情況下，scope=function

import pytest
?
?
# 沒(méi)有設(shè)置scope，默認(rèn)為測(cè)試函數(shù)級(jí)別。即調(diào)用此fixture的測(cè)試類(lèi)/模塊/函數(shù)下，每個(gè)測(cè)試函數(shù)都會(huì)執(zhí)行一次這個(gè)fixture
@pytest.fixture
def init():
    print("用例執(zhí)行之前，執(zhí)行的代碼")  # 前置代碼
    yield
    print("用例執(zhí)行之后，執(zhí)行的代碼")  # 后置代碼
?
?
# 設(shè)置scope為class。調(diào)用此fixture的測(cè)試類(lèi)下，只執(zhí)行一次這個(gè)fixture.
@pytest.fixture(scope="class")
def init2():
    print("用例執(zhí)行之前，執(zhí)行的代碼")  # 只有用例執(zhí)行之前的前置準(zhǔn)備代碼
?
?
# 設(shè)置scope為session。autouse表示自動(dòng)使用。
# 那么在pytest收集用例后，開(kāi)始執(zhí)行用例之前會(huì)自動(dòng)化執(zhí)行這個(gè)fixture當(dāng)中的前置代碼，
# 當(dāng)所有用例執(zhí)行完成之后，自動(dòng)化執(zhí)行這個(gè)fixture的后置代碼。
@pytest.fixture(scope="session",autouse=True)
def init3():
    yield
    print("用例執(zhí)行之后，執(zhí)行的代碼")  # 只有用例執(zhí)行之后的后置清理代

6）fixture的返回值設(shè)置：yeild 返回值

當(dāng)測(cè)試用例當(dāng)中，要使用fixture里生成的數(shù)據(jù)時(shí)，則需要fixture返回?cái)?shù)據(jù)。

若有數(shù)據(jù)返回則：yeild 返回值

import pytest
from selenium import webdriver
from time import sleep
?
?
# 設(shè)置scope為class。調(diào)用此fixture的測(cè)試類(lèi)下，只執(zhí)行一次這個(gè)fixture.
@pytest.fixture(scope="class")
def init2():
    print("==== 測(cè)試類(lèi)下，執(zhí)行所有用例之前，執(zhí)行的代碼 ====")
    driver = webdriver.Chrome()
    yield driver   # 返回driver對(duì)象
    print("==== 測(cè)試類(lèi)下，執(zhí)行所有用例之后，執(zhí)行的代碼 ====")
    driver.quit()