现在假设需要对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
from unittest import TestCase, mock
import unittest
from get_app import create_app
test_config = 'HHH'
class GetappTestCase(TestCase):
@mock.patch('get_app.load_yaml') # 注意这里的mock路径
def test_get_app(self, mock_load):
mock_load.return_value = test_config
self.assertEqual(create_app(), 'HHH')
if __name__ == '__main__':
unittest.main()
ii) get_app.py
from utils import load_yaml
def create_app():
ret = load_yaml()
return ret
iii) utils.py
def load_yaml():
return 'This is a config'
测试结果如下: