当调用unittest.mock.Mock
对象时,我可以检查具有该调用的确切签名的参数值:
from unittest.mock import Mock
m = Mock() # creation of mock
m('foo', bar='baz') # call to the mock
m.assert_called_once_with('foo', bar='baz') # check call arguments
检查具有相同值的其他签名将失败。例如,如果我们将'baz'
作为位置参数而不是命名参数进行检查,则断言将失败:
m.assert_called_once_with('foo', 'baz')
# AssertionError: Expected call: mock('foo', 'baz')
# Actual call: mock('foo', bar='baz')
必须。如果由m
替换的函数是
def actual_fu(foo, bar):
# do something
然后调用将是等效的,但如果它是
def a_different_actual_fu(foo, *args, bar='some default'):
# do something
那么这些电话就不一定了。 Mock
不知道实际函数的签名,因此它不能依赖于我们在第一种情况下的等价性。
是否有办法通过让Mock(或断言助手函数或类似函数)知道有关的函数来检查那些不确定它们是通过位置传递还是作为关键字参数的调用参数值。实际的功能被mock替换了?
Mock
对象可以通过可选的spec
argument或autospeccing知道它替换的对象(可以是函数或方法),但这些对象有不同的用途(限制模拟允许的调用次数)并且不影响事后检查。
答案 0 :(得分:4)
可以使用可选的spec参数或
autospeccing
使Mock对象知道它替换的对象(可以是函数或方法),但这些对象有不同的用途。
这正是Issue 17015: mock could be smarter and inspect the spec's signature改进问题的关键所在。 spec
实际上是非常相关的,现在使模拟功能 - 签名感知。
当我们断言使用关键字参数调用mock时,看看mock是如何失败的 - 不让它知道实际的函数签名:
>>> from unittest.mock import Mock
>>>
>>> def actual_fu(foo, bar):
... pass
>>>
>>> m = Mock()
>>> m('foo', bar='baz')
<Mock name='mock()' id='4356741496'>
>>>
>>> m.assert_called_once_with(foo='foo', bar='baz')
Traceback (most recent call last):
File "<stdin>", line 1, in <module>
File "/Library/Frameworks/Python.framework/Versions/3.5/lib/python3.5/unittest/mock.py", line 803, in assert_called_once_with
return self.assert_called_with(*args, **kwargs)
File "/Library/Frameworks/Python.framework/Versions/3.5/lib/python3.5/unittest/mock.py", line 792, in assert_called_with
raise AssertionError(_error_message()) from cause
AssertionError: Expected call: mock(bar='baz', foo='foo')
Actual call: mock('foo', bar='baz')
现在,如果我们提供spec
:
>>> m = Mock(spec=actual_fu)
>>> m('foo', bar='baz')
<Mock name='mock()' id='4356249528'>
>>>
>>> m.assert_called_once_with(foo='foo', bar='baz')
>>> m.assert_called_once_with('foo', bar='baz')
>>> m.assert_called_once_with(bar='baz', foo='foo')
>>> m.assert_called_once_with('foo', 'baz')
>>>
(使用Python 3.5.1)