我如何检查调用参数是否会随着unittest.mock更改 [英] How can i check call arguments if they will change with unittest.mock
本文介绍了我如何检查调用参数是否会随着unittest.mock更改的处理方法,对大家解决问题具有一定的参考价值,需要的朋友们下面随着小编来一起学习吧!
问题描述
我的一个类在列表中累积值,将该列表用作另一个对象上方法的参数,并删除此列表中的某些值.像
One of my classes accumulates values in a list, uses the list as an argument to a method on another object and deletes some of the values in this list. Something like
element = element_source.get()
self.elements.append(element)
element_destination.send(elements)
self.remove_outdated_elements()
但是当我尝试测试这种行为时,我发现模拟程序不会复制其参数.
But when when i was trying to test this behavior, i've found that mocks don't copy their arguments.
>>> from unittest.mock import Mock
>>> m = Mock()
>>> a = [1]
>>> m(a)
<Mock name='mock()' id='139717658759824'>
>>> m.call_args
call([1])
>>> a.pop()
1
>>> m.assert_called_once_with([1])
Traceback (most recent call last):
File "<stdin>", line 1, in <module>
File "/usr/lib/python3.3/unittest/mock.py", line 737, in assert_called_once_with
return self.assert_called_with(*args, **kwargs)
File "/usr/lib/python3.3/unittest/mock.py", line 726, in assert_called_with
raise AssertionError(msg)
AssertionError: Expected call: mock([1])
Actual call: mock([])
有没有办法让Mock复制它的调用参数?如果没有,测试这种行为的最佳方法是什么?
Is there a way to make Mock copy it's call arguments? If not, what is the best way to test this kind of behavior?
推荐答案
There is a chapter "Coping with mutable arguments" in the documentation, which suggests several solutions to your problem.
我会选择这个:
>>> 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)
...
>>> c = CopyingMock(return_value=None)
>>> arg = set()
>>> c(arg)
>>> arg.add(1)
>>> c.assert_called_with(set())
>>> c.assert_called_with(arg)
Traceback (most recent call last):
...
AssertionError: Expected call: mock(set([1]))
Actual call: mock(set([]))
>>> c.foo
<CopyingMock name='mock.foo' id='...'>
这篇关于我如何检查调用参数是否会随着unittest.mock更改的文章就介绍到这了,希望我们推荐的答案对大家有所帮助,也希望大家多多支持IT屋!
查看全文