python - 重用Mock创建属性模拟unittest.patch

Question

from unittest.mock import patch


class A:
    def b(self):
        return 'HAHAHA A'
a = A()

with patch('__main__.A') as a_mock:
    a_mock.b.return_value = 'not working'
    print(a.b())

HAHAHA A
>>>

为什么它不打印'notworking'？那么 a_mock 是做什么用的呢？ ______________

Answer 1

您用补丁替换了整个类，而不是现有类上的方法。 a = A() 在替换类之前创建了 A 的实例，因此 a.__class__ 仍然引用实际类，而不是模拟类.

模拟一次只能替换一个引用，而不能替换所引用的对象。在补丁之前，名称 A 和属性 a.__class__ 都是对类对象的引用。然后，您仅修补 A 引用，将 a.__class__ 保留在原处。

换句话说，a.__class__ 没有被修补，只有 A 和 A().b() 会打印不工作。

您必须仅仅修补类上的方法，以便a.__class__仍然引用A，并且 a.b 将解析为修补后的 A.b 模拟:

with patch('__main__.A.b') as b_mock:
    b_mock.return_value = 'working as long as you patch the right object'
    print(a.b())

演示:

>>> with patch('__main__.A.b') as b_mock:
...     b_mock.return_value = 'working as long as you patch the right object'
...     print(a.b())
...
working as long as you patch the right object

不幸的是，你不能用Mock来修补a.__class__引用； Python 只允许您使用该属性的实际类。

>>> with patch('__main__.a.__class__') as a_class_mock:
...     a_class_mock.b.return_value = 'working as long as you patch the right object'
...     print(a.b())
...
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
  File "/Users/mjpieters/Development/Library/buildout.python/parts/opt/lib/python3.5/unittest/mock.py", line 1312, in __enter__
    setattr(self.target, self.attribute, new_attr)
TypeError: __class__ must be set to a class, not 'MagicMock' object