检查unittest.mock调用参数wn.n.t.它们是作为位置参数还是关键字参数传递的

时间:2016-01-22 11:05:36

标签: python unit-testing mocking python-unittest.mock

当调用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 argumentautospeccing知道它替换的对象(可以是函数或方法),但这些对象有不同的用途(限制模拟允许的调用次数)并且不影响事后检查。

1 个答案:

答案 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)