1.前言

微服務(wù)架構(gòu)下，由于各類服務(wù)開發(fā)進(jìn)度的不一致，導(dǎo)致聯(lián)調(diào)工作經(jīng)常會(huì)存在不確定性，進(jìn)而導(dǎo)致項(xiàng)目延期

在實(shí)際工作中，為了保證項(xiàng)目進(jìn)度，我們經(jīng)常需要針對(duì)部分未完成模塊及不穩(wěn)定模塊采用Mock方式，以驗(yàn)證已開發(fā)完的模塊

本篇文章將介紹Python實(shí)現(xiàn)Mock的幾種常見方式

2.Mock介紹

Mock測(cè)試：在測(cè)試驗(yàn)證過程中，對(duì)于那些尚未完成或不穩(wěn)定的對(duì)象，用一個(gè)虛擬對(duì)象來(lái)替代，以便測(cè)試的測(cè)試方法

因此，這個(gè)虛擬的對(duì)象是Mock對(duì)象，Mock對(duì)象是真實(shí)對(duì)象在調(diào)試期間的代替品

它的優(yōu)勢(shì)包含：

·前、后端并行開發(fā)

·模擬無(wú)法訪問的資源

·隔離系統(tǒng)，避免臟數(shù)據(jù)干擾測(cè)試結(jié)果

3.1mock

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

#安裝mock依賴

pip3installmock

項(xiàng)目地址：

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

假設(shè)Product類中有2個(gè)方法

·get_product_status_by_id

·buy_product

其中，get_product_status_by_id方法還沒有實(shí)現(xiàn);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獲取產(chǎn)品信息(Mock)

:return:

"""

#待實(shí)現(xiàn)查詢數(shù)據(jù)庫(kù)的業(yè)務(wù)邏輯

pass

defbuy_product(self,product_id):

"""

購(gòu)買產(chǎn)品(真實(shí)邏輯)

:return:

"""

#產(chǎn)品信息

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

product=self.get_product_status_by_id(product_id)

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

result={"status":0,"msg":"購(gòu)買成功!"}

else:

result={"status":1,"msg":"購(gòu)買失敗，庫(kù)存不足!"}

returnresult

Mock的步驟如下：

·導(dǎo)入使用mock中的patch方法

·作為測(cè)試方法的裝飾器，對(duì)get_product_status_by_id方法進(jìn)行Mock，方法參數(shù)為Mock對(duì)象

·測(cè)試方法中，對(duì)該Mock對(duì)象設(shè)置一個(gè)返回值

·調(diào)用并斷言

frommockimportpatch

frommock_.product_implimportProduct

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

deftest_succuse(mock_get_product_status_by_id):

#Mock方法，指定一個(gè)返回值

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

product=Product()

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

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

使用@patch.object同樣能完成Mock，不同的是，@patch.object包含2個(gè)參數(shù)

第一個(gè)參數(shù)為該方法所在的類;第二個(gè)參數(shù)為方法名

frommockimportpatch

frommock_.product_implimportProduct

#Mock一個(gè)方法

#@patch.object：對(duì)象、方法名

@patch.object(Product,'get_product_status_by_id')

deftest_succuse(mock_get_product_status_by_id):

#Mock方法，指定一個(gè)返回值

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作為標(biāo)準(zhǔn)庫(kù)，已經(jīng)內(nèi)置到unittest中了

還是以3.1的場(chǎng)景為例，使用unittest編寫一個(gè)測(cè)試用例

Mock步驟如下：

·導(dǎo)入unittest框架中的mock文件

·實(shí)例化Product對(duì)象

·mock.Mock(return_value=*)方法

·對(duì)get_product_status_by_id方法進(jìn)行Mock

·調(diào)用并斷言

importunittest

fromunittestimportmock

fromunittest_mock.product_implimportProduct

classTestProduct(unittest.TestCase):

deftest_success(self):

#成功結(jié)果

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

product=Product()

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

#調(diào)用實(shí)際函數(shù)

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

if__name__=="__main__":

unittest.main()

3.3pytest.mock

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

如果項(xiàng)目本身使用的框架是pytest，則Mock更建議使用pytest-mock這個(gè)插件

#pytest依賴

pip3installpytest

Mock步驟如下：

·使用pytest編寫測(cè)試方法，參數(shù)為mocker

·實(shí)例化Product對(duì)象

·使用mocker.patch()方法對(duì)get_product_status_by_id方法進(jìn)行Mock，并設(shè)置返回值

·調(diào)用并斷言

importpytest

frompytest_mock_.product_implimportProduct

deftest_buy_product_success(mocker):

"""

購(gòu)買成功Mock

:parammocker:

:return:

"""

#實(shí)例化一個(gè)產(chǎn)品對(duì)象

product=Product()

#對(duì)Product中的方法的返回值進(jìn)行Mock

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

#Mock方法

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

#mocker.patch的第一個(gè)參數(shù)必須是模擬對(duì)象的具體路徑，第二個(gè)參數(shù)用來(lái)指定返回值

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

return_value=mock_value)

#調(diào)用購(gòu)買產(chǎn)品的方法

result=product.buy_product(1)

assertresult.get("status")==0

需要注意的是，mocker.patch方法第一個(gè)參數(shù)必須是Mock對(duì)象的完整路徑

4.最后

文中對(duì)Python中常見的Mock方案進(jìn)行了講解，實(shí)際應(yīng)用中，建議根據(jù)項(xiàng)目實(shí)際情況進(jìn)行選型

以上內(nèi)容為大家介紹了Python中的Mock，希望對(duì)大家有所幫助，如果想要了解更多Python相關(guān)知識(shí)，請(qǐng)關(guān)注IT培訓(xùn)機(jī)構(gòu):千鋒教育。http://www.parentadvocate.org/