單元測(cè)試 Mock不Mock？
前言
前段時(shí)間，我們團(tuán)隊(duì)就單元測(cè)試是否采用 Mock 進(jìn)行了一番交流，各有各的說法。本文就單元測(cè)試 Mock不Mock 給出我的觀點(diǎn)，歡迎各位同仁提出不同的意見，共同探討、相互交流。
單元測(cè)試沒必要?
我見過好多不寫單元測(cè)試的項(xiàng)目，大多給出的原因都是：“沒必要”、“浪費(fèi)時(shí)間”、“get不到單測(cè)的作用”，這樣的項(xiàng)目要么是小規(guī)模，要么就是頭鐵。
本人之前也有相同的觀點(diǎn)...
單元測(cè)試在軟件開發(fā)過程中還是非常重要的，除了可以提高代碼的質(zhì)量，在引入CI/CD后的自動(dòng)化測(cè)試環(huán)節(jié)可以起到快速部署、交付作用。難道每次上線都需要“點(diǎn)點(diǎn)點(diǎn)”測(cè)試？這一點(diǎn)，我想經(jīng)歷過的人都深有體會(huì)。
Mock不Mock?
那什么是Mock？
什么是Mock?
簡(jiǎn)單來(lái)說，Mock就是模擬目標(biāo)代碼的行為，在實(shí)際測(cè)試過程中代替真實(shí)的調(diào)用目標(biāo)。如下圖
圖片
這樣做的意義何在？
Mock的意義何在?
試想一下，單元測(cè)試中如果出現(xiàn)以下幾個(gè)問題應(yīng)該怎么辦？
- 涉及到的DB操作、網(wǎng)絡(luò)調(diào)用等單元測(cè)試產(chǎn)生的數(shù)據(jù)屬不屬于垃圾數(shù)據(jù)？會(huì)不會(huì)影響業(yè)務(wù)？
- 發(fā)布/部署生產(chǎn)環(huán)境的過程中，錯(cuò)誤地執(zhí)行了單元測(cè)試引起生產(chǎn)問題怎么辦？
- CI過程中的測(cè)試環(huán)節(jié)花費(fèi)時(shí)間太長(zhǎng)怎么辦？會(huì)不會(huì)影響集成交付？
以上幾個(gè)問題我想大部分開發(fā)人員都經(jīng)歷過，那如何避免這些問題？我想Mock就是最好的一種方式。
如果將涉及到的外部操作，例如DB操作、網(wǎng)絡(luò)調(diào)用等行為進(jìn)行Mock，那就不會(huì)存在垃圾數(shù)據(jù)的問題，也不用擔(dān)心環(huán)境切換帶來(lái)的問題，外部耗時(shí)的操作也可以通過Mock避免CI過程過長(zhǎng)。
個(gè)人認(rèn)為Mock只是模擬調(diào)用外部的行為，并不影響代碼邏輯。所以，不存在“Mock是不是有效的單元測(cè)試”這種說法。
如何Mock？
應(yīng)該Mock什么？
通常，我們編寫的方法(或函數(shù))都是由很多方法按照層級(jí)組成的，就像這樣
圖片
當(dāng)我們對(duì)頂層方法進(jìn)行單元測(cè)試時(shí)，應(yīng)該Mock哪些方法？
- 如果Mock方法1、2、3，那么方法4、5、6就不會(huì)被調(diào)用到，里面的邏輯不會(huì)被覆蓋到，也就不是有效的單元測(cè)試。
- 如果Mock方法4、5、6，里面的邏輯或返回值有修改，那么就要遞歸向上修改，不符合軟件工程。
但是，如果方法43、、5、6都涉及到DB或者網(wǎng)絡(luò)調(diào)用等外部不可控操作，我們就應(yīng)該對(duì)其Mock。
所以，應(yīng)該Mock一些穩(wěn)定的、不可控的方法。
Mock 編寫示例
以Python中的Mock框架為例，下面是一個(gè)示例：
class TestXxService(unittest.TestCase):
def test_init(self):
XXService.update(xx)
class XXService:
def update(xx):
......test_init函數(shù)中的....update會(huì)涉及到數(shù)據(jù)庫(kù)的操作，這里使用patch模擬這兩個(gè)函數(shù)的行為
# patch("目標(biāo)函數(shù)路徑")
patch('....update')在模擬的上下文中，XXService.update將會(huì)被模擬的函數(shù)替代執(zhí)行
def test_init(self):
with patch('....update') as mocked_update:
# 在模擬的上下文中調(diào)用業(yè)務(wù)邏輯函數(shù)
XXService.update(xx)
mocked_update.assert_called_once_with(xx)其中，assert_called_once_with 會(huì)驗(yàn)證模擬函數(shù)是否被調(diào)用了一次，并且會(huì)驗(yàn)證預(yù)期接收的參數(shù)是否匹配。
如果沒有參數(shù)，使用assert_called_once進(jìn)行驗(yàn)證是否被調(diào)用了一次。
如果模擬的函數(shù)實(shí)際被調(diào)用了多次，需要通過以下方式
# 斷言mocked_update被調(diào)用了2次
self.assertEqual(mocked_update.call_count, 2)
# 斷言mocked_update被調(diào)用了，并且參數(shù)正確
mocked_update.assert_any_call(xx)如果函數(shù)有返回值，在定義模擬函數(shù)時(shí)，添加 return_value，return_value可以是任意類型。
patch('...update',
return_value='xxx') as mocked_update在驗(yàn)證返回值時(shí)通過下面的方式
xxxx = mocked_update.return_value
self.assertEqual(xxxx, 'xxx')通過示例，我們Mock了XXService.update行為，實(shí)現(xiàn)了對(duì)XXService的隔離測(cè)試，并確保了測(cè)試的可靠性和高效性。
總結(jié)
單元測(cè)試中使用Mock有以下幾個(gè)好處：
- 隔離測(cè)試：Mock 使得測(cè)試可以專注于測(cè)試的代碼邏輯，而不必關(guān)心外部不穩(wěn)定因素。
- 提高測(cè)試速度：Mock 可以避免耗時(shí)的外部調(diào)用，從而加快測(cè)試速度。
- 提高測(cè)試的可靠性和穩(wěn)定性：通過Mock，可以避免外部變化對(duì)測(cè)試結(jié)果的影響。
