这在文档部分进行了讨论26.6.3.7。应对可变参数 https://docs.python.org/3.5/library/unittest.mock-examples.html#coping-with-mutable-arguments.
不幸的是,他们确实没有任何优雅的解决方案来解决这个问题!推荐的解决方法是使用以下方法从可变参数中复制元素side_effect
.
如果您为模拟提供 side_effect 函数,则将使用与模拟相同的参数调用 side_effect。这使我们有机会复制参数并将其存储以供以后断言。
在我看来,实施起来有点混乱。如果您需要在多个地方使用该功能,您可能更喜欢子类化Mock
并直接添加该功能:
from copy import deepcopy
class CopyingMock(MagicMock):
def __call__(self, *args, **kwargs):
args = deepcopy(args)
kwargs = deepcopy(kwargs)
return super(CopyingMock, self).__call__(*args, **kwargs)
2017:现在可以在第三方发行版中使用(pip install copyingmock
).
>>> from copyingmock import CopyingMock
>>> mock = CopyingMock()
>>> list_ = [1,2]
>>> mock(list_)
<CopyingMock name='mock()' id='4366094008'>
>>> list_.append(3)
>>> mock.assert_called_once_with([1,2])
>>> mock.assert_called_once_with(list_)
AssertionError: Expected call: mock([1, 2, 3])
Actual call: mock([1, 2])