什么是mock?
mock在翻譯過來有模擬的意思爹梁。這里要介紹的mock是輔助單元測(cè)試的一個(gè)模塊。它允許您用模擬對(duì)象替換您的系統(tǒng)的部分狐蜕，并對(duì)它們已使用的方式進(jìn)行斷言煞聪。
在Python2.x 中 mock是一個(gè)單獨(dú)模塊，需要單獨(dú)安裝膊夹。
> pip install -U mock
在Python3.x中衬浑，mock已經(jīng)被集成到了unittest單元測(cè)試框架中，所以放刨，可以直接使用嚎卫。
什么情況下使用mock
在項(xiàng)目的單元測(cè)試過程中，會(huì)遇到：
1宏榕、接口的依賴
2拓诸、外部接口調(diào)用
3、測(cè)試環(huán)境非常復(fù)雜
單元測(cè)試應(yīng)該只針對(duì)當(dāng)前單元進(jìn)行測(cè)試, 所有的內(nèi)部或外部的依賴應(yīng)該是穩(wěn)定的, 已經(jīng)在別處進(jìn)行測(cè)試過的.使用mock 就可以對(duì)外部依賴組件實(shí)現(xiàn)進(jìn)行模擬并且替換掉, 從而使得單元測(cè)試將焦點(diǎn)只放在當(dāng)前的單元功能麻昼。
解決測(cè)試依賴
例如奠支，我們要測(cè)試A模塊，然后A模塊依賴于B模塊的調(diào)用抚芦。但是倍谜，由于B模塊的改變迈螟，導(dǎo)致了A模塊返回結(jié)果的改變，從而使A模塊的測(cè)試用例失敗尔崔。其實(shí)答毫，對(duì)于A模塊，以及A模塊的用例來說季春，并沒有變化洗搂，不應(yīng)該失敗才對(duì)。
這個(gè)時(shí)候就是mock發(fā)揮作用的時(shí)候了载弄。通過mock模擬掉影響A模塊的部分（B模塊）耘拇。至于mock掉的部分（B模塊）應(yīng)該由其它用例來測(cè)試。
# function.py
def add_and_multiply(x, y):
addition = x + y
multiple = multiply(x, y)
return (addition, multiple)
def multiply(x, y):
return x * y
然后宇攻，針對(duì) add_and_multiply()函數(shù)編寫測(cè)試用例惫叛。func_test.py
import unittest
import function
class MyTestCase(unittest.TestCase):
def test_add_and_multiply(self):
x = 3
y = 5
addition, multiple = function.add_and_multiply(x, y)
self.assertEqual(8, addition)
self.assertEqual(15, multiple)
if __name__ == "__main__":
unittest.main()
運(yùn)行結(jié)果：
> python3 func_test.py
.
----------------------------------------------------------------------
Ran 1 test in 0.000s
OK
目前運(yùn)行一切正確常，然而逞刷，add_and_multiply()函數(shù)依賴了multiply()函數(shù)的返回值嘉涌。如果這個(gè)時(shí)候修改multiply()函數(shù)的代碼。
def multiply(x, y):
return x * y + 3
這個(gè)時(shí)候夸浅，multiply()函數(shù)返回的結(jié)果變成了x*y加3洛心。再次運(yùn)行測(cè)試，則會(huì)報(bào)錯(cuò)题篷。然而词身，add_and_multiply()函數(shù)以及它的測(cè)試用例并沒有做任何修改，罪魁禍?zhǔn)资莔ultiply()函數(shù)引起的番枚，我們應(yīng)該把 multiply()函數(shù)mock掉法严。
import unittest
from unittest.mock import patch
import function
class MyTestCase(unittest.TestCase):
@patch("function.multiply") #這是個(gè)裝飾器，其作用了解一下
def test_add_and_multiply2(self, mock_multiply):
x = 3
y = 5
mock_multiply.return_value = 15 #重點(diǎn)：比之前多了這一行
addition, multiple = function.add_and_multiply(x, y)
mock_multiply.assert_called_once_with(3, 5) #比之前多了這一行
self.assertEqual(8, addition)
self.assertEqual(15, multiple)
if __name__ == "__main__":
unittest.main()
@patch("function.multiply")
patch()裝飾/上下文管理器可以很容易地模擬類或?qū)ο笤谀K測(cè)試葫笼。在測(cè)試過程中深啤，您指定的對(duì)象將被替換為一個(gè)模擬（或其他對(duì)象），并在測(cè)試結(jié)束時(shí)還原路星。這里模擬function.py文件中multiply()函數(shù)溯街。
def test_add_and_multiply2(self, mock_multiply):
在定義測(cè)試用例中，將mock的multiply()函數(shù)（對(duì)象）重命名為 mock_multiply對(duì)象洋丐。
mock_multiply.return_value = 15
設(shè)定mock_multiply對(duì)象的返回值為固定的15呈昔。
ock_multiply.assert_called_once_with(3, 5)
檢查ock_multiply方法的參數(shù)是否正確。
再次友绝，運(yùn)行測(cè)試用例堤尾，通過！
