1.前言

微服務架構下，由于各類服務開發進度的不一致，導致聯調工作經常會存在不確定性，進而導致項目延期

在實際工作中，為了保證項目進度，我們經常需要針對部分未完成模塊及不穩定模塊采用Mock方式，以驗證已開發完的模塊

本篇文章將介紹Python實現Mock的幾種常見方式

2.Mock介紹

Mock測試：在測試驗證過程中，對于那些尚未完成或不穩定的對象，用一個虛擬對象來替代，以便測試的測試方法

因此，這個虛擬的對象是Mock對象，Mock對象是真實對象在調試期間的代替品

它的優勢包含：

·前、后端并行開發

·模擬無法訪問的資源

·隔離系統，避免臟數據干擾測試結果

3.1mock

在Python3.3之前使用mock，需要先安裝依賴

#安裝mock依賴

pip3installmock

項目地址：

https://github.com/testing-cabal/mock

假設Product類中有2個方法

·get_product_status_by_id

·buy_product

其中，get_product_status_by_id方法還沒有實現;buy_product方法依賴于get_product_status_by_id方法的返回值。

#product_impl.py

classProduct(object):

def__init__(self):

pass

defget_product_status_by_id(self,product_id):

"""

通過商品id獲取產品信息(Mock)

:return:

"""

#待實現查詢數據庫的業務邏輯

pass

defbuy_product(self,product_id):

"""

購買產品(真實邏輯)

:return:

"""

#產品信息

#{"id":1,"name":"蘋果","num":23}

product=self.get_product_status_by_id(product_id)

ifproduct.get("num")>=1:

result={"status":0,"msg":"購買成功!"}

else:

result={"status":1,"msg":"購買失敗，庫存不足!"}

returnresult

Mock的步驟如下：

·導入使用mock中的patch方法

·作為測試方法的裝飾器，對get_product_status_by_id方法進行Mock，方法參數為Mock對象

·測試方法中，對該Mock對象設置一個返回值

·調用并斷言

frommockimportpatch

frommock_.product_implimportProduct

@patch('mock_.product_impl.Product.get_product_status_by_id')

deftest_succuse(mock_get_product_status_by_id):

#Mock方法，指定一個返回值

mock_get_product_status_by_id.return_value={"id":1,"name":"蘋果","num":23}

product=Product()

assertproduct.buy_product(1).get("status")==0

需要注意的是，Mock此方法的時候，必須制定該方法的完整路徑

使用@patch.object同樣能完成Mock，不同的是，@patch.object包含2個參數

第一個參數為該方法所在的類;第二個參數為方法名

frommockimportpatch

frommock_.product_implimportProduct

#Mock一個方法

#@patch.object：對象、方法名

@patch.object(Product,'get_product_status_by_id')

deftest_succuse(mock_get_product_status_by_id):

#Mock方法，指定一個返回值

mock_get_product_status_by_id.return_value={"id":1,"name":"蘋果","num":23}

product=Product()

assertproduct.buy_product(1).get("status")==0

3.2unittest.mock

Python3.3之后，mock作為標準庫，已經內置到unittest中了

還是以3.1的場景為例，使用unittest編寫一個測試用例

Mock步驟如下：

·導入unittest框架中的mock文件

·實例化Product對象

·mock.Mock(return_value=*)方法

·對get_product_status_by_id方法進行Mock

·調用并斷言

importunittest

fromunittestimportmock

fromunittest_mock.product_implimportProduct

classTestProduct(unittest.TestCase):

deftest_success(self):

#成功結果

mock_success_value={"id":1,"name":"蘋果","num":23}

product=Product()

product.get_product_status_by_id=mock.Mock(return_value=mock_success_value)

#調用實際函數

assertproduct.buy_product(1).get("status")==0

if__name__=="__main__":

unittest.main()

3.3pytest.mock

相比unittest，pytest由于強大的插件支持，用戶群體可能更大!

如果項目本身使用的框架是pytest，則Mock更建議使用pytest-mock這個插件

#pytest依賴

pip3installpytest

Mock步驟如下：

·使用pytest編寫測試方法，參數為mocker

·實例化Product對象

·使用mocker.patch()方法對get_product_status_by_id方法進行Mock，并設置返回值

·調用并斷言

importpytest

frompytest_mock_.product_implimportProduct

deftest_buy_product_success(mocker):

"""

購買成功Mock

:parammocker:

:return:

"""

#實例化一個產品對象

product=Product()

#對Product中的方法的返回值進行Mock

mock_value={"id":1,"name":"蘋果","num":23}

#Mock方法

#注意：需要指定方法的完整路徑

#mocker.patch的第一個參數必須是模擬對象的具體路徑，第二個參數用來指定返回值

product.get_product_status_by_id=mocker.patch("product_impl.Product.get_product_status_by_id",

return_value=mock_value)

#調用購買產品的方法

result=product.buy_product(1)

assertresult.get("status")==0

需要注意的是，mocker.patch方法第一個參數必須是Mock對象的完整路徑

4.最后

文中對Python中常見的Mock方案進行了講解，實際應用中，建議根據項目實際情況進行選型

以上內容為大家介紹了Python中的Mock，希望對大家有所幫助，如果想要了解更多Python相關知識，請關注IT培訓機構:千鋒教育。http://www.mobiletrain.org/

上一篇

5個好用的Python工具

下一篇

python 面向對象(OOP)的三大特征