Python unittest模块的使用笔记:对mock.patch()的被mock函数的说明
现在假设需要对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(