您如何测试函数引发异常？

Question

我下面有一个函数，想编写一个单元测试来检查我的代码是否使用模拟库捕获了ConnectionError

def get_foo():
    try:
        return requests.get("http://www.bongani.com")
    except requests.exceptions.ConnectionError:
        print("Error")

我有的：

import unittest
import mock


class MyTestCase(unittest.TestCase):
    @mock.patch('requests.get')
    def test_foo(self, mock_requests_get):
        mock_requests_get.side_effect = requests.exceptions.ConnectionError()
        with self.assertRaises(requests.exceptions.ConnectionError):
            get_foo()

if __name__ == '__main__':
    unittest.main()

我收到此错误：

Traceback (most recent call last):
File "/usr/local/lib/python2.7/site-packages/mock/mock.py", line 1305 , in patched
  return func(*args, **keywargs)
File "<ipython-input-24-15266e4f708a>", , in test_foo
  get_foo()
AssertionError: ConnectionError not raised

我想模拟return requests.get("http://www.bongani.com")行，以便在调用时引发异常

Answer 1

异常不会冒出来（即被抑制），因为您抓住了它，然后又没有重新引发它。 反过来，单元测试将失败，因为您正在测试函数在抑制异常时会引发该异常。

您可以根据需要的行为来更改功能或功能的单元测试。

假设测试正确，则您的函数需要重新引发如下异常：

def get_foo():
    try:
        return requests.get("http://www.bongani.com")
    except requests.exceptions.ConnectionError:
        print("Error")
        raise  # <---------------- add this line

另一种情况，功能正确，测试错误。 您将测试修复为如下所示：

class MyTestCase(unittest.TestCase):
    @mock.patch('requests.get')
    def test_foo(self, mock_requests_get):
        mock_requests_get.side_effect = requests.exceptions.ConnectionError()
        self.assertIsNone(get_foo())  # <----- no more assertRaises

Answer 2

确保修补正确的路径

@mock.patch('path.to.get_foo.requests.get')

您不想实例化异常。

mock_requests_get.side_effect = requests.exceptions.ConnectionError