python - 我们如何确保 Mock.call_args_list 中的调用包含与调用 Mock 对象时处于相同状态的参数的调用？

Question

from mock import Mock
j = []
u = Mock()
u(j)
# At this point u.call_args_list == [call([])]
print u.call_args_list
j.append(100)
# At this point u.call_args_list == [call([100])], but I expect it to be [call([])], since it was never called when j had a value of 100 in it
print u.call_args_list

我的问题是如何确保 u.call_args_list 中的调用包含所有对象在调用 mock 时而不是在检查 mock 的参数时的状态？

我现在正在使用 mock==1.0.1。

Answer 1

这在文档部分进行了讨论 26.6.3.7. Coping with mutable arguments .

不幸的是，他们真的没有任何优雅的解决方案来解决这个问题!推荐的解决方法是使用 side_effect 从可变参数中复制元素。

If you provide a side_effect function for a mock then side_effect will be called with the same args as the mock. This gives us an opportunity to copy the arguments and store them for later assertions.

在我看来，实现起来有些困惑。如果您需要在多个地方使用该功能，您可能更愿意子类化 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])