python - 如何模拟类实例属性？

Question

class MyClass:
    def __init__(self, a, b):
        self.a = a
        self.b = b
 
def myFunc():
    myClass = MyClass(1, 2)
    print(myClass.a, myClass.b)

在我的测试中，我想运行 myFunc()但里面myFunc()我要myClass返回 'test1'和 'test2'作为其 a和 b属性。

所以我这样做了:

with patch('__main__.MyClass') as MockClass:
    instance = MockClass.return_value
    instance.a.return_value = 'test1'
    instance.b.return_value = 'test2'
    myFunc()

然而，结果是<MagicMock name=MyClass().a' id='2938383'> ..... 而不是简单地 'test1'和 'test2' .

我做错了什么？我对如何使用 mock 有什么误解？

Answer 1

return_value 用于可调用对象，只写:

with patch('__main__.MyClass') as MockClass:
   instance = MockClass.return_value
   instance.a = 'test1'
   instance.b = 'test2'
   myFunc()

# test1 test2

而且你必须修补 with语句中的类，注意'__main__.MyClass'中Class的大写字母。