模拟函数引发异常以测试except块

Jesse Webb 发表于 Python

杰西·韦伯（Jesse Webb）：

我有一个foo调用另一个函数（bar）的函数（）。如果调用bar()引发一个HttpError，如果状态代码为404，我想特别处理它，否则重新引发。

我正在尝试围绕此foo函数编写一些单元测试，以模拟对的调用bar()。不幸的是，我无法得到模拟调用bar()以引发被我的代码except块捕获的异常。

这是说明我问题的代码：

import unittest
import mock
from apiclient.errors import HttpError


class FooTests(unittest.TestCase):
    @mock.patch('my_tests.bar')
    def test_foo_shouldReturnResultOfBar_whenBarSucceeds(self, barMock):
        barMock.return_value = True
        result = foo()
        self.assertTrue(result)  # passes

    @mock.patch('my_tests.bar')
    def test_foo_shouldReturnNone_whenBarRaiseHttpError404(self, barMock):
        barMock.side_effect = HttpError(mock.Mock(return_value={'status': 404}), 'not found')
        result = foo()
        self.assertIsNone(result)  # fails, test raises HttpError

    @mock.patch('my_tests.bar')
    def test_foo_shouldRaiseHttpError_whenBarRaiseHttpErrorNot404(self, barMock):
        barMock.side_effect = HttpError(mock.Mock(return_value={'status': 500}), 'error')
        with self.assertRaises(HttpError):  # passes
            foo()

def foo():
    try:
        result = bar()
        return result
    except HttpError as error:
        if error.resp.status == 404:
            print '404 - %s' % error.message
            return None
        raise

def bar():
    raise NotImplementedError()

我跟着模拟文档这不能不说您应该设置side_effect一个的Mock情况下，以一个Exception班有嘲笑功能引发错误。

我还查看了其他一些与StackOverflow相关的Q＆A，看来我在做他们正在做的相同事情，以使他们的模拟引发Exception。

为什么设置side_effect的barMock不引起预期Exception得到提升？如果我做的事情很奇怪，我应该如何在except块中测试逻辑？

马丁·彼得斯（Martijn Pieters）：

您的模拟正在引发异常，但是该error.resp.status值丢失了。而不是使用return_value，只是告诉Mock它status是一个属性：

barMock.side_effect = HttpError(mock.Mock(status=404), 'not found')

将其他关键字参数Mock()设置为结果对象的属性。

我将您的foo和bar定义放在my_tests模块中，并添加到HttpError类中，这样我也可以使用它，然后您的测试就可以成功运行：

>>> from my_tests import foo, HttpError
>>> import mock
>>> with mock.patch('my_tests.bar') as barMock:
...     barMock.side_effect = HttpError(mock.Mock(status=404), 'not found')
...     result = my_test.foo()
... 
404 - 
>>> result is None
True

您甚至可以看到print '404 - %s' % error.message生产线运行，但是我想您想在error.content那里使用它。HttpError()无论如何，这是第二个参数设置的属性。

本文收集自互联网，转载请注明来源。

如有侵权，请联系[email protected] 删除。