Python单元测试与嘲笑request.get，没有引发特定的异常

Question

我有一个简单的函数，发送一个获取请求到一个API，我想能够写一个单元测试来声明，脚本打印错误，然后如果请求返回异常则退出。

def fetch_members(self): 
    try: 
     r = requests.get(api_url, auth=('user', api_key)) 
    except requests.exceptions.HTTPError as error: 
     print(error) 
     sys.exit(1) 

我已经试过以下测试，但HTTPError是从未提出

import requests 
from unittest.mock import patch 

@patch('sys.exit') 
@patch('sys.stdout', new_callable=StringIO) 
@patch('requests.get') 
def test_fetch_members_with_bad_request(self, mock_get, mock_stdout, 
             mock_exit): 
    response = requests.Response() 
    response.status_code = 400 # Bad reqeust 
    mock_get.return_value = response 
    mock_get.side_effect = requests.exceptions.HTTPError 
    with self.assertRaises(requests.exceptions.HTTPError): 
     fetch_members() 
     mock_exit.assert_called_once_with(1) 
     self.assertIsNotNone(mock_stdout.getvalue()) 

我如何去正确地修补requests.get总是提出一些类型的错误？

Answer 1

你的功能捕获并处理异常：

except requests.exceptions.HTTPError as error: 

这意味着它会永远进一步传播，所以你assertRaises()失败。只是断言sys.exit()已被调用，这就够了。

设置返回值也没有意义;该调用会引发异常。以下就足够了：

@patch('sys.exit') 
@patch('sys.stdout', new_callable=StringIO) 
@patch('requests.get') 
def test_fetch_members_with_bad_request(self, mock_get, mock_stdout, 
             mock_exit): 
    mock_get.side_effect = requests.exceptions.HTTPError 
    fetch_members() 
    mock_exit.assert_called_once_with(1) 
    self.assertIsNotNone(mock_stdout.getvalue())