Python 中 Mock 到底該怎麼玩?一篇文章告訴你(超全)
1. 前言
本文的文字及圖片來源於網路,僅供學習、交流使用,不具有任何商業用途,如有問題請及時聯絡我們以作處理。
PS:如有需要Python學習資料的小夥伴可以加點選下方連結自行獲取
微服務架構下,由於各類服務開發進度的不一致,導致聯調工作經常會存在不確定性,進而導致專案延期
在實際工作中,為了保證專案進度,我們經常需要針對部分未完成模組及不穩定模組採用 Mock 方式,以驗證已開發完的模組
本篇文章將介紹 Python 實現 Mock 的幾種常見方式
2. Mock 介紹
Mock 測試:在測試驗證過程中,對於那些尚未完成或不穩定的物件,用一個虛擬物件來替代,以便測試的測試方法
因此,這個虛擬的物件是 Mock 物件,Mock 物件是真實物件在除錯期間的代替品
它的優勢包含:
- 前、後端並行開發
- 模擬無法訪問的資源
- 隔離系統,避免髒資料干擾測試結果
3.1 mock
在 Python 3.3 之前使用 mock,需要先安裝依賴
# 安裝mock依賴
pip3 install mock
專案地址:
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 class 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 patch
from 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 patch
from 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 unittest
from unittest import mock
from unittest_mock.product_impl import Product
class 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") == 0
if __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 pytest
from pytest_mock_.product_impl import Product
def 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 方案進行了講解,實際應用中,建議根據專案實際情況進行選型。