首次调用后引发异常

Question

我有一个循环，我在其中处理将记录添加到zip文件。我已经模拟了我的 zipfile 对象，我想引发一个异常来验证我处理大型 zip 文件的逻辑是否正常工作。

有没有办法使用 MagicMocks 或普通模拟来接受第一个调用，但在第二个调用时引发异常？

Answer 1

最简单的方法是使用接受可迭代、可调用或异常（类或实例）的side_effect

或者，side_effect可以是异常类或实例。在这种情况下，调用模拟时将引发异常。

如快速指南所示，您可以使用side_effect简单地通过以下方式引发异常

>>> mock = Mock(side_effect=KeyError('foo'))
>>> mock()
Traceback (most recent call last):
 ...
KeyError: 'foo'

此外，您可以在side_effect分配中同时使用列表和异常。因此，执行所需操作的最简单方法是：

>>> m = Mock(side_effect=[1, KeyError("bar"), 3])
>>> m("a")
1
>>> m("b")
 ...
KeyError: 'bar'
>>> m("c")
3

另一种方法是使用可调用对象来编写模拟应该如何反应的逻辑。在这种情况下，您可以根据参数或测试的状态自由选择。

Answer 2

，带有生成器和模拟side_effect

from unittest.mock import Mock
def sample_generator():
    yield 1
    yield 2
    raise Exception()
gen = sample_generator()
def sideeffect():
    global gen
    for x in gen:
        return x

m = Mock(side_effect=sideeffect)
m() #1
m() #2
m() #3

您可以获得：

File "test.py", line 22, in <module>
   m() #3
   ....
   raise Exception()
Exception

我相信你可以通过一些努力让它稍微干净一些，但它应该解决你的基本问题