1. 前言

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

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

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

2. Mock 介紹

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

因此，這個虛擬的物件是 Mock 物件，Mock 物件是真實物件在除錯期間的代替品

它的優勢包含：

前、後端並行開發模擬無法訪問的資源隔離系統，避免髒資料干擾測試結果3.1 mock

在 Python 3.3 之前使用 mock，需要先安裝依賴

# 安裝mock依賴pip3 install mock

假設 Product 類中有 2 個方法

get_product_status_by_idbuy_product

其中，get_product_status_by_id 方法還沒有實現；buy_product 方法依賴於 get_product_status_by_id 方法的返回值

# product_impl.pyclass Product(object): def __init__(self): pass def get_product_status_by_id(self, product_id): """ 透過商品id獲取產品資訊（Mock） :return: """ # 待實現查詢資料庫的業務邏輯 pass def buy_product(self, product_id): """ 購買產品（真實邏輯） :return: """ # 產品資訊 # {"id":1,"name":"蘋果","num":23} product = self.get_product_status_by_id(product_id) if product.get("num") >= 1: result = {"status": 0, "msg": "購買成功！"} else: result = {"status": 1, "msg": "購買失敗，庫存不足！"} return result

Mock 的步驟如下：

匯入使用 mock 中的 patch 方法作為測試方法的裝飾器，對 get_product_status_by_id 方法進行 Mock，方法引數為 Mock 物件測試方法中，對該 Mock 物件設定一個返回值呼叫並斷言from mock import patchfrom mock_.product_impl import Product@patch('mock_.product_impl.Product.get_product_status_by_id')def test_succuse(mock_get_product_status_by_id):    # Mock方法，指定一個返回值    mock_get_product_status_by_id.return_value = {"id": 1, "name": "蘋果", "num": 23}    product = Product()    assert product.buy_product(1).get("status") == 0需要注意的是，Mock 此方法的時候，必須制定該方法的完整路徑使用 @patch.object 同樣能完成 Mock，不同的是，@patch.object 包含 2 個引數第一個引數為該方法所在的類；第二個引數為方法名from mock import patchfrom mock_.product_impl import Product# Mock一個方法# @patch.object：物件、方法名@patch.object(Product, 'get_product_status_by_id')def test_succuse(mock_get_product_status_by_id):    # Mock方法，指定一個返回值    mock_get_product_status_by_id.return_value = {"id": 1, "name": "蘋果", "num": 23}    product = Product()    assert product.buy_product(1).get("status") == 0

3.2 unittest.mock

Python 3.3 之後，mock 作為標準庫，已經內建到 unittest 中了

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

Mock 步驟如下：

匯入 unittest 框架中的 mock 檔案例項化 Product 物件mock.Mock(return_value=*) 方法對 get_product_status_by_id 方法進行 Mock呼叫並斷言

import unittestfrom unittest import mockfrom unittest_mock.product_impl import Productclass TestProduct(unittest.TestCase): def test_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) # 呼叫實際函式 assert product.buy_product(1).get("status") == 0if __name__ == "__main__": unittest.main()

3.3 pytest.mock

相比 unittest，pytest 由於強大的外掛支援，使用者群體可能更大！

如果專案本身使用的框架是 pytest，則 Mock 更建議使用 pytest-mock 這個外掛

# pytest依賴pip3 install pytest

Mock 步驟如下：

使用 pytest 編寫測試方法，引數為 mocker例項化 Product 物件使用 mocker.patch() 方法對 get_product_status_by_id 方法進行 Mock，並設定返回值呼叫並斷言

import pytestfrom pytest_mock_.product_impl import Productdef test_buy_product_success(mocker): """ 購買成功Mock :param mocker: :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) assert result.get("status") == 0

需要注意的是，mocker.patch 方法第一個引數必須是 Mock 物件的完整路徑

4. 最後

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