mock.MagicMock(spec=SomeClass) 和mock.create_autospec(SomeClass) 有什么区别？

Question

我试图了解这两个模拟构造之间的区别以及何时适合使用其中任何一个。我在解释器中测试了它，例如：

>>> mm = mock.MagicMock(spec=list)
>>> ca = mock.create_autospec(list)
>>> mm
<MagicMock spec='list' id='140372375801232'>
>>> mm()
<MagicMock name='mock()' id='140372384057808'>
>>> mm.append()
<MagicMock name='mock.append()' id='140372375724720'>
>>> mm().append()
<MagicMock name='mock().append()' id='140372375753104'>
>>> ca
<MagicMock spec='list' id='140372384059248'>
>>> ca()
<NonCallableMagicMock name='mock()' spec='list' id='140372384057040'>
>>> ca.append()
<MagicMock name='mock.append()' id='140372375719744'>
>>> ca().append()
<MagicMock name='mock().append()' id='140372375796848'>
>>> 

但我不明白为什么“构建”使用创建的模拟create_autospec给了我 aNonCallableMagicMock并且MagicMock给了我更多MagicMock。该文档没有多大帮助。

Answer 1

spec使用参数和使用之间的主要区别create_autospec是递归性。在第一种情况下，对象本身是指定的，而被调用的对象不是：

>>> mm = mock.MagicMock(spec=list)
>>> mm
<MagicMock spec='list' id='2868486557120'>
>>> mm.foo
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
  File "c:\Python\Python38\lib\unittest\mock.py", line 635, in __getattr__
    raise AttributeError("Mock object has no attribute %r" % name)
AttributeError: Mock object has no attribute 'foo'
>>> mm.append
<MagicMock name='mock.append' id='2868486430240'>
>>> mm.append.foo
<MagicMock name='mock.append.foo' id='2868486451408'>

在第二种情况下，被调用的对象也被指定（惰性地）：

>>> ca = mock.create_autospec(list)
>>> ca
<MagicMock spec='list' id='2868486254848'>
>>> ca.foo
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
  File "c:\Python\Python38\lib\unittest\mock.py", line 635, in __getattr__
    raise AttributeError("Mock object has no attribute %r" % name)
AttributeError: Mock object has no attribute 'foo'
>>> ca.append
<MagicMock name='mock.append' spec='method_descriptor' id='2868486256336'>
>>> ca.append.foo
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
  File "c:\Python\Python38\lib\unittest\mock.py", line 635, in __getattr__
    raise AttributeError("Mock object has no attribute %r" % name)
AttributeError: Mock object has no attribute 'foo'

有一个警告，如您的示例代码所示。如果您create_autospec按此处所示使用，它的行为就好像该对象是一个类，而不是一个实例，因此您可以调用它（创建一个实例）：

>>> ca = mock.create_autospec(list)
>>> ca()
<NonCallableMagicMock name='mock()' spec='list' id='2868485877280'>

如果你想让它像一个实例一样运行，你必须使用instance=True：

>>> ca = mock.create_autospec(list, instance=True)
>>> ca
<NonCallableMagicMock spec='list' id='2868485875024'>
>>> ca()
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
TypeError: 'NonCallableMagicMock' object is not callable

请注意，使用mock.patchwithautospec=True创建的模拟行为与使用创建的模拟类似mock.create_autospec，如文档中所述。

另请注意，无论实际调用的返回值如何，调用的返回值始终为 a 。MagicMock因此，即使函数返回None，例如list.append，如果从模拟调用该方法，也会返回模拟，无论规范如何。