现在假设需要对get_app模块内的create_app函数做单元测试，同时create_app函数调用了另一模块utils的load_yaml函数。由于模块utils的load_yaml函数可能处于开发阶段或是需要通过网络传输数据，从而导致测试的不便。这时就需要对load_yaml做一个mock.patch，即伪造一个load_yaml的返回值来完成对其调用者的单元测试。
这里的关键在于mock.patch()中load_yaml的路径写法：从调用者(create_app)路径引入load_yaml而不是直接从load_yaml自身所在的模块路径引入。
演示文件共3个，均位于同一目录下。
i) test_get_app.py
1from unittest import TestCase, mock 2import unittest 3from get_app import create_app 4 5test_config = 'HHH' 6 7class GetappTestCase(TestCase): 8 9 @mock.patch('get_app.load_yaml') # 注意这里的mock路径 10 def test_get_app(self, mock_load): 11 mock_load.return_value = test_config 12 self.assertEqual(create_app(), 'HHH') 13 14if __name__ == '__main__': 15 unittest.main()
ii) get_app.py
1from utils import load_yaml 2 3def create_app(): 4 ret = load_yaml() 5 return ret
iii) utils.py
1def load_yaml(): 2 return 'This is a config'
测试结果如下：
