什么是mock?
Mock，顧名思義，模擬，在我們日常生活中或者影視作品中見得最多的可能就是預備飛行員的模擬訓練，印象比較深的是電影《薩利機長》中的模擬器，經過幾千次模擬，人們得出機長薩利在飛機遇到鳥群撞擊后，作出了最符合當時情況的最正確的迫降決定，創造了奇跡。這里面涉及到的模擬就和我們今天所要講的模擬類似，即用一個虛擬的對象來完成某些不容易構造或者不容易獲取的對象或者場景的構造。
為什么要mock
本文所講的mock是mock測試，是輔助單元測試的一個模塊。
實際生產中的項目是非常復雜的，對其進行單元測試的時候，可能會遇到以下問題：
- 接口的依賴
- 外部接口調用
- 測試環境非常復雜
單元測試應該只針對當前單元進行測試, 所有的內部或外部的依賴應該是穩定的, 已經在別處進行測試過的.使用mock 就可以對外部依賴組件實現進行模擬並且替換掉, 從而使得單元測試將焦點只放在當前的單元功能。
安裝使用
在Python2.x中 mock是一個單獨模塊，需要單獨安裝。在Python3中 它已經被集成到了unitest模塊可以直接會用from unittest import mock進行導入。當然，我們也可以使用如下命令獨立安裝。
pip install -U mock
簡單的例子
我們先從最簡單例子開始。
#modular.py
class Count():
def add(self):
pass
這里要實現一個Count計算類，add()方法要實現兩數相加。但這個功能我還沒有完成。這時就可以借助mock對其進行測試。
# mock_demo01.py
from unittest import mock
import unittest
from modular import Count
# test Count class
class TestCount(unittest.TestCase):
def test_add(self):
count = Count()
count.add = mock.Mock(return_value=13)
result = count.add(8,5)
self.assertEqual(result,13)
if __name__ == '__main__':
unittest.main()
首先實例化一個對象，調用被測試類Count() 。
count = Count()
count.add = mock.Mock(return_value=7)
通過Mock類模擬被調用的方法add()方法，return_value 定義add()方法的返回值。
result = count.add(2,5)
接下來，相當於在正常的調用add()方法，傳兩個參數2和5，然后會得到相加的結果7。然后，7的結果是我們在上一步就預先設定好的。
self.assertEqual(result,7)
最后，通過assertEqual()方法斷言，返回的結果是否是預期的結果7。
運行測試結果：
> python3 mock_demo01.py
.
----------------------------------------------------------------------
Ran 1 test in 0.000s
OK
這樣一個用例就在mock的幫助下編寫完成，並且測試通過了。
完成功能測試
再接下來完成module.py文件中add()方法。
#module.py
class Count():
def add(self, a, b):
return a + b
然后，修改測試用例：
from unittest import mock
import unittest
from module import Count
class MockDemo(unittest.TestCase):
def test_add(self):
count = Count()
count.add = mock.Mock(return_value=13, side_effect=count.add)
result = count.add(8, 8)
print(result)
count.add.assert_called_with(8, 8)
self.assertEqual(result, 16)
if __name__ == '__main__':
unittest.main()
count.add = mock.Mock(return_value=13, side_effect=count.add)
side_effect參數和return_value是相反的。它給mock分配了可替換的結果，覆蓋了return_value。簡單的說，一個模擬工廠調用將返回side_effect值，而不是return_value。
所以，設置side_effect參數為Count類add()方法，那么return_value的作用失效。
result = count.add(8, 8)
print(result)
這次將會真正的調用add()方法，得到的返回值為16（8+8）。通過print打印結果。
assert_called_with(8,8)
檢查mock方法是否獲得了正確的參數。
解決測試依賴
前面的例子，只為了讓大家對mock有個初步的印象。再接來，我們看看如何mock方法的依賴。
例如，我們要測試A模塊，然后A模塊依賴於B模塊的調用。但是，由於B模塊的改變，導致了A模塊返回結果的改變，從而使A模塊的測試用例失敗。其實，對於A模塊，以及A模塊的用例來說，並沒有變化，不應該失敗才對。
這個時候就是mock發揮作用的時候了。通過mock模擬掉影響A模塊的部分（B模塊）。至於mock掉的部分（B模塊）應該由其它用例來測試。
# 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
然后，針對 add_and_multiply()函數編寫測試用例。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()
運行結果：
> python3 func_test.py
.
----------------------------------------------------------------------
Ran 1 test in 0.000s
OK
目前運行一切正確常，然而，add_and_multiply()函數依賴了multiply()函數的返回值。如果這個時候修改multiply()函數的代碼。
……
def multiply(x, y):
return x * y + 3
這個時候，multiply()函數返回的結果變成了x*y加3。
再次運行測試：
> python3 func_test.py
F
======================================================================
FAIL: test_add_and_multiply (__main__.MyTestCase)
----------------------------------------------------------------------
Traceback (most recent call last):
File "fun_test.py", line 19, in test_add_and_multiply
self.assertEqual(15, multiple)
AssertionError: 15 != 18
----------------------------------------------------------------------
Ran 1 test in 0.000s
FAILED (failures=1)
測試用例運行失敗了，然而，add_and_multiply()函數以及它的測試用例並沒有做任何修改，罪魁禍首是multiply()函數引起的，我們應該把 multiply()函數mock掉。
import unittest
from unittest.mock import patch
import function
class MyTestCase(unittest.TestCase):
@patch("function.multiply")
def test_add_and_multiply2(self, mock_multiply):
x = 3
y = 5
mock_multiply.return_value = 15
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()裝飾/上下文管理器可以很容易地模擬類或對象在模塊測試。在測試過程中，您指定的對象將被替換為一個模擬（或其他對象），並在測試結束時還原。
這里模擬function.py文件中multiply()函數。def test_add_and_multiply2(self, mock_multiply):在定義測試用例中，將mock的multiply()函數（對象）重命名為 mock_multiply對象。mock_multiply.return_value = 15設定mock_multiply對象的返回值為固定的15。mock_multiply.assert_called_once_with(3, 5) 檢查ock_multiply方法的參數是否正確。
再次，運行測試用例，通過！