python - 模拟通过实例使用的类方法

Question

我正尝试使用 mock 修补类方法，如所述 in the documentation . Mock 对象本身工作正常，但它的方法不工作:例如，它们的属性如 call_count 没有更新，即使类 的 method_calls 属性>Mock 对象是。更重要的是，它们的 return_value 属性被忽略了:

class Lib:
    """In my actual program, a module that I import"""
    def method(self):
        return "real"

class User:
    """The class I want to test"""
    def run(self):
        l = Lib()
        return l.method()

with patch("__main__.Lib") as mock:
    #mock.return_value = "bla" # This works
    mock.method.return_value = "mock"
    u = User()
    print(u.run())

>>> 
mock
<MagicMock name='Lib().method()' id='39868624'>

我在这里做错了什么？

编辑:通过构造函数传递类 Mock 也不起作用，因此这与 patch 函数没有真正的关系。

Answer 1

我发现了我的错误:为了配置模拟实例的方法，我必须使用 mock().method 而不是 mock.method。

class Lib:
    """In my actual program, a module that I import"""
    def method(self):
        return "real"

class User:
    """The class I want to test"""
    def run(self):
        l = Lib()
        return l.method()

with patch("__main__.Lib") as mock:
    #mock.return_value = "bla" # This works
    mock().method.return_value = "mock"
    u = User()
    print(u.run())