文章目录
- 前言
- 单元测试没必要?
- Mock不Mock?
- 什么是Mock?
- Mock的意义何在?
- 如何Mock？
- 应该Mock什么？
- Mock 编写示例
- 总结
前言
前段时间，我们团队就单元测试是否采用 Mock 进行了一番交流，各有各的说法。本文就单元测试 Mock不Mock 给出我的观点，欢迎各位同仁提出不同的意见，共同探讨、相互交流。
单元测试没必要?
我见过好多不写单元测试的项目，大多给出的原因都是：“没必要”、“浪费时间”、“get不到单测的作用”，这样的项目要么是小规模，要么就是头铁。
本人之前也有相同的观点…
单元测试在软件开发过程中还是非常重要的，除了可以提高代码的质量，在引入CI/CD后的自动化测试环节可以起到快速部署、交付作用。难道每次上线都需要“点点点”测试？这一点，我想经历过的人都深有体会。
Mock不Mock?
那什么是Mock？
什么是Mock?
简单来说，Mock就是模拟目标代码的行为，在实际测试过程中代替真实的调用目标。如下图
这样做的意义何在？
Mock的意义何在?
试想一下，单元测试中如果出现以下几个问题应该怎么办？
- 涉及到的DB操作、网络调用等单元测试产生的数据属不属于垃圾数据？会不会影响业务？
- 发布/部署生产环境的过程中，错误地执行了单元测试引起生产问题怎么办？
- CI过程中的测试环节花费时间太长怎么办？会不会影响集成交付？
以上几个问题我想大部分开发人员都经历过，那如何避免这些问题？我想Mock就是最好的一种方式。
如果将涉及到的外部操作，例如DB操作、网络调用等行为进行Mock，那就不会存在垃圾数据的问题，也不用担心环境切换带来的问题，外部耗时的操作也可以通过Mock避免CI过程过长。
个人认为Mock只是模拟调用外部的行为，并不影响代码逻辑。所以，不存在“Mock是不是有效的单元测试”这种说法。
如何Mock？
应该Mock什么？
通常，我们编写的方法(或函数)都是由很多方法按照层级组成的，就像这样
当我们对顶层方法进行单元测试时，应该Mock哪些方法？
- 如果Mock方法1、2、3，那么方法4、5、6就不会被调用到，里面的逻辑不会被覆盖到，也就不是有效的单元测试。
- 如果Mock方法4、5、6，里面的逻辑或返回值有修改，那么就要递归向上修改，不符合软件工程。
但是，如果方法43、、5、6都涉及到DB或者网络调用等外部不可控操作，我们就应该对其Mock。
所以，应该Mock一些稳定的、不可控的方法。
Mock 编写示例
以Python中的Mock框架为例，下面是一个示例：
class TestXxService(unittest.TestCase):
def test_init(self):
XXService.update(xx)
class XXService:
def update(xx):
......
test_init函数中的....update会涉及到数据库的操作，这里使用patch模拟这两个函数的行为
# patch("目标函数路径")
patch('....update')
在模拟的上下文中，XXService.update将会被模拟的函数替代执行
def test_init(self):
with patch('....update') as mocked_update:
# 在模拟的上下文中调用业务逻辑函数
XXService.update(xx)
mocked_update.assert_called_once_with(xx)
其中，assert_called_once_with 会验证模拟函数是否被调用了一次，并且会验证预期接收的参数是否匹配。
如果没有参数，使用assert_called_once进行验证是否被调用了一次。
如果模拟的函数实际被调用了多次，需要通过以下方式
# 断言mocked_update被调用了2次
self.assertEqual(mocked_update.call_count, 2)
# 断言mocked_update被调用了，并且参数正确
mocked_update.assert_any_call(xx)
如果函数有返回值，在定义模拟函数时，添加 return_value，return_value可以是任意类型。
patch('...update',
return_value='xxx') as mocked_update
在验证返回值时通过下面的方式
xxxx = mocked_update.return_value
self.assertEqual(xxxx, 'xxx')
通过示例，我们Mock了XXService.update行为，实现了对XXService的隔离测试，并确保了测试的可靠性和高效性。
总结
单元测试中使用Mock有以下几个好处：
- 隔离测试：Mock 使得测试可以专注于测试的代码逻辑，而不必关心外部不稳定因素。
- 提高测试速度：Mock 可以避免耗时的外部调用，从而加快测试速度。
- 提高测试的可靠性和稳定性：通过Mock，可以避免外部变化对测试结果的影响。
