1.patch裝飾器的概念

• patch裝飾器是mock第二種實現(xiàn)數(shù)據(jù)模擬的方式，patch()充當(dāng)函數(shù)裝飾器，類裝飾器或上下文管理器，可用于處理測試范圍內(nèi)的修補模塊和類級屬性；

2.patch裝飾器語法

• 語法：unittest.mock.patch（target，new = DEFAULT，spec = None，create = False，spec_set = None，autospec = None，new_callable = None，** kwarg ）；

  

  成都創(chuàng)新互聯(lián)是一家專業(yè)提供江源企業(yè)網(wǎng)站建設(shè),專注與成都網(wǎng)站建設(shè)、網(wǎng)站建設(shè)、H5高端網(wǎng)站建設(shè)、小程序制作等業(yè)務(wù)。10年已為江源眾多企業(yè)、政府機構(gòu)等服務(wù)。創(chuàng)新互聯(lián)專業(yè)網(wǎng)站制作公司優(yōu)惠進行中。

• 參數(shù)target 為’package.module.classname’格式的字符串，如果參數(shù)new被省略，那么target 被MagicMock代替；

• 如果patch()用作裝飾器并且省略new，則創(chuàng)建的mock將作為額外參數(shù)傳遞給裝飾函數(shù)，如果patch()用作上下文管理器，則上下文管理器返回創(chuàng)建的mock；

• New_Callable參數(shù)允許指定將被調(diào)用以創(chuàng)建新對象的其他類或可調(diào)用對象，默認情況下，使用magicMock。

3.patch裝飾器實現(xiàn)數(shù)據(jù)模擬

修改Test.py文件中的TestData類，在測試用例前加上一個 @mock .patch()裝飾器

from unittest import mock
from unittest import TestCase
import unittest
import function1
class TestData(TestCase):
    # patch裝飾器
    @mock.patch('function1.data_parse')
    def test_print1(self, mock_data_parse):
        mock_data_parse.return_value = {"result": "success", "reason":"null"}
        statues = function1.data_show()
        print(statues)
        self.assertEqual(statues, "data parse success")
    @mock.patch('function1.data_parse')
    def test_print2(self, data_parse):
        data_parse.return_value = {"result": "fail", "reason": "Data Error"}
        statues = function1.data_show()
        self.assertEqual(statues, "data parse failed")
if __name__ == "__main__":
    unittest.main()

我們這里模擬的是函數(shù)中的數(shù)據(jù)，如果需要mock的數(shù)據(jù)是一個類中方法的數(shù)據(jù)，寫法也是一樣的，比如function1.py模塊中的代碼是：

class DataParse:
    def data_parse(self):
        pass
class DataShow:
    def data_show(self):
        ret = DataParse().data_parse()
        try:
            if ret.get('result') == "success":
                return "data parse success"
            elif ret.get('result') == "fail":
                print("data parse failed: {}".format(ret.get('reason')))
                return "data parse failed"
            else:
                return "Unknow Reason"
        except:
            return "Server Unknow Reason"

那Test.py文件中的代碼可以這樣寫：

from unittest import mock
from unittest import TestCase
import unittest
from function1 import DataParse, DataShow
class TestData(TestCase):
    # patch裝飾器
    @mock.patch('function1.DataParse')
    def test_print1(self, mock_DataParse):
        # 先拿到類的mock實例
        dataParse = mock_DataParse.return_value
        # 在通過類的mock實例調(diào)用方法，對方法返回值進行mock
        dataParse.data_parse.return_value = {"result": "success", "reason":"null"}
        # 調(diào)用data_show
        statues = DataShow().data_show()
        print(statues)
        self.assertEqual(statues, "data parse success")
    @mock.patch('function1.DataParse')
    def test_print2(self, mock_DataParse):
        # 先拿到類的mock實例
        dataParse = mock_DataParse.return_value
        # 在通過類的mock實例調(diào)用方法，對方法返回值進行mock
        dataParse.data_parse.return_value = {"result": "fail", "reason": "Data Error"}
        # 調(diào)用data_show
        statues = DataShow().data_show()
        print(statues)
        self.assertEqual(statues, "data parse failed")
if __name__ == "__main__":
    unittest.main()
=========輸出=============
data parse success
data parse failed: Data Error
data parse failed

步驟

• 首先使用@mock.patch()裝飾器來裝飾你要使用的測試用例，@mock.patch()中需要帶一個參數(shù)，指定需要mock的函數(shù)或者類，和這個類所在的包和模塊，例如：@mock.patch(''function1.data_parse'')，表示給一個在function1模塊中的data_parse函數(shù)mock數(shù)據(jù)；

• 然后在測試用例中接受一個參數(shù)，這個參數(shù)是mock哪個函數(shù)，就在函數(shù)名前加一個mock，例如給data_parse函數(shù)mock數(shù)據(jù)，這個參數(shù)名就是mock_data_parse；

• 有了這個參數(shù)之后就可以給mock的函數(shù)的return_value賦值了。最后就可以進行斷言，運行測試了；

參考： https://www.9xkd.com/user/plan-view.html?id=1585826642