自動化測試框架中的fixture

我們在編寫測試用例，都會涉及到用例執行之前的環境准備工作，和用例執行之后的環境清理工作。

代碼版的測試用例也不例外。在自動化測試框架當中，我們也需要編寫：

用例執行之前的環境准備工作代碼(前置工作代碼)

用例執行之后的環境清理工作(后置工作代碼)

通常，在自動化測試框架當中，都叫做fixture。

pytest作為python語言的測試框架，它的fixture有2種實現方式。

一種是xunit-style，跟unittest框架的機制非常相似，即setup/teardown系列

一種是它自己的fixture機制，以@pytest.fixture裝飾器來申明。

pytest的fixture實現方式一：xunit-style

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

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

測試類內部的測試方法：

前置函數名稱：setup_method

后置函數名稱：teardown_method

模塊下的測試函數：

前置函數名稱：setup_function

后置函數名稱：teardown_function

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

前置函數名稱：setup_class

后置函數名稱：teardown_class

注意：用@classmethod裝飾

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

前置函數名稱：setup_module

后置函數名稱：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實現方式二：fixture機制

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

使用fixture時，分為二個部分：fixture定義、fixture調用。

除此之外，還有fixture的共享機制，嵌套調用機制。

1、定義fixture。

1）fixture通過函數實現。

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

import pytest

@pytest.fixture
def init():
    pass

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

4）通過yeild關鍵字，區分前置代碼和后置代碼。yeild之前的代碼為前置代碼，yeild之后的代碼為后置代碼

在實際應用場景當中，可以只有前置准備工作代碼，也可以只有后置清理工作代碼。

import pytest


@pytest.fixture
def init():
    print("用例執行之前，執行的代碼")  # 前置代碼 
    yield 
    print("用例執行之后，執行的代碼")  # 后置代碼
 

@pytest.fixture
def init2():
    print("用例執行之前，執行的代碼")  # 只有用例執行之前的前置准備代碼
    

@pytest.fixture
def init3():
    yield
    print("用例執行之后，執行的代碼")  # 只有用例執行之后的后置清理代碼

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

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

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

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

import pytest


# 沒有設置scope，默認為測試函數級別。即調用此fixture的測試類/模塊/函數下，每個測試函數都會執行一次這個fixture
@pytest.fixture
def init():
    print("用例執行之前，執行的代碼")  # 前置代碼
    yield
    print("用例執行之后，執行的代碼")  # 后置代碼


# 設置scope為class。調用此fixture的測試類下，只執行一次這個fixture.
@pytest.fixture(scope="class")
def init2():
    print("用例執行之前，執行的代碼")  # 只有用例執行之前的前置准備代碼


# 設置scope為session。autouse表示自動使用。
# 那么在pytest收集用例后，開始執行用例之前會自動化執行這個fixture當中的前置代碼，
# 當所有用例執行完成之后，自動化執行這個fixture的后置代碼。
@pytest.fixture(scope="session",autouse=True)
def init3():
    yield
    print("用例執行之后，執行的代碼")  # 只有用例執行之后的后置清理代

6）fixture的返回值設置：yeild 返回值

當測試用例當中，要使用fixture里生成的數據時，則需要fixture返回數據。

若有數據返回則：yeild 返回值

import pytest
from selenium import webdriver
from time import sleep


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