模拟side_effect迭代器在耗尽后可以重置吗？

Question

mock.reset_mock()将不会重置副作用迭代器。有没有办法做到这一点，而不是再次创建模拟？

>>> from mock import MagicMock 
>>> mock = MagicMock(side_effect = [1,2]) 
>>> mock(), mock() 
(1, 2) 
>>> mock() 

Traceback (most recent call last): 
    File "<pyshell#114>", line 1, in <module> 
    mock() 
    File "C:\Python27\Lib\site-packages\mock.py", line 955, in __call__ 
    return _mock_self._mock_call(*args, **kwargs) 
    File "C:\Python27\Lib\site-packages\mock.py", line 1013, in _mock_call 
    result = next(effect) 
StopIteration 
>>> mock.reset_mock() 
>>> mock() 

Traceback (most recent call last): 
    ... 
StopIteration 
>>> mock = MagicMock(side_effect = [1,2]) 
>>> mock(), mock() 
(1, 2) 
>>> 

这样做的目的是重新使用在随后的测试模拟，但我怀疑，就像一台发电机，它不能重启。

所以（迟到总比不到好）被指出在我看了mock.py，发现side_effect是一个迭代对象（不能为复位一旦累死了）正确的方向后：

def __set_side_effect(self, value): 
    value = _try_iter(value) 
    ... 

def _try_iter(obj): 
    ... 
    try: 
     return iter(obj) 
    except TypeError: 
     # XXXX backwards compatibility 
     # but this will blow up on first call - so maybe we should fail early? 
     return obj 

和def reset_mock()没有解决侧面效应。

Answer 1

由于user2357112评论，重新分配side_effect将解决您的问题。

>>> from mock import MagicMock 
>>> 
>>> lst = [1, 2] 
>>> mock = MagicMock(side_effect=lst) 
>>> mock(), mock() 
(1, 2) 
>>> mock.side_effect = lst # <------- 
>>> mock(), mock() 
(1, 2)