python - 第一次调用后抛出异常

Question

我有一个循环，用于将记录添加到 zip 文件。我模拟了我的 zipfile 对象，我想抛出一个异常来验证我处理大型 zipfile 的逻辑是否能正常工作。

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

Answer 1

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

Alternatively side_effect can be an exception class or instance. In this case the exception will be raised when the mock is called.

如Quick Guide所示您可以使用 side_effect 简单地通过

引发异常

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

此外，您可以在side_effect 赋值中同时使用list 和Exception。所以做你需要做的最简单的方法是这样的:

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

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