Pytest框架之fixture詳解（一）

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

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

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

pytest的xunit-style有三個級別的fixture：測試模塊、測試類、測試函數(shù)。

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

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

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

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

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

2、測試類級別：一個測試類只執(zhí)行一次前置和后置。

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

注意：用@classmethod裝飾

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

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

from selenium import webdriver
from time import sleep
from random import randint
?
?
def setup_module():
    print("====  模塊級的 setup 操作  ====")
?
?
def teardown_module():
    print("====  模塊級的 teardown 操作  ====")
?
?
def test_random():
    assert randint(1, 5) == 3
?
?
class TestWeb:
?
    @classmethod
    def setup_class(cls):
        print("====  測試類級的 setup 操作  ====")
?
    @classmethod
    def teardown_class(cls):
        print("====  測試類級的 teardown 操作  ====")
?
    def setup_method(self):
        print("====  測試用例級的 setup 操作  ====")
        self.driver = webdriver.Chrome()
?
    def teardown_method(self):
        print("====  測試用例級的 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也支持運行unittest的測試用例。支持unittest以下特性：

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

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

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

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

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

1、定義fixture。

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

2）使用@pytest.fixture進行裝飾

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

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

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

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

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

5）fixture有4個作用域：測試會話(session)、測試模塊(module)、測試類(class)、測試用例(function)

測試會話：pytest執(zhí)行測試用例的整個過程，稱為會話。

比如pytest收集到了100條用例并執(zhí)行完成，這個過程稱為測試會話。

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

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

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

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

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

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