如何使用 MagicMock 模拟以下 python 函数?

Question

class C:
def f():
calls g
def g():
# Do something

如何在导入 C 类的测试模块中模拟 g 来测试 f？

Answer 1

你可以使用这样的路径对象来模拟该函数;

with patch.object(C, 'g', MagicMock(return_value='something')):
c.f()

这样，当你的代码调用 g 函数时，它会返回模拟响应

注意：使用前应导入补丁

您可以从下面的链接中找到详细信息：

https://docs.python.org/3/library/unittest.mock.html