斷言失敗

Question

我正在嘗試編寫一個單元測試，以確保在必要時引發HTTPException。 這是測試：

import unittest  
from requests import HTTPError
import pyport

 # Code omitted...
 def test_bad_item_type(self):
     """A bad item type should raise a HTTPError"""
     test_type = 'bad'
     test_id = 1986134
     self.assertRaises(HTTPError, pyport.get_item(test_type, test_id))

產生以下內容：

ERROR: test_bad_item_type (__main__.TestPyportFunctions) A bad item
type should raise requests.HTTPError
---------------------------------------------------------------------- 
Traceback (most   recent call last):   File "./tests.py", line 65, in
test_bad_item_type
    self.assertRaises(HTTPError, pyport.get_item(test_type, test_id))   File    "/home/sean/workspace/pyport/pyport.py", line 54, in get_item
    response.raise_for_status()   File "/usr/local/lib/python2.7/dist-packages/requests/models.py", line 741, fin raise_for_status
    raise HTTPError('%s Client Error' % self.status_code) HTTPError: 404 Client Error

引發了異常，但測試沒有捕獲到它。 這與此問題中發生的情況類似，但並不完全相同。 有人能告訴我我錯過了什么嗎？

Answer 1

它應該是：

self.assertRaises(HTTPError, pyport.get_item, test_type, test_id)

查看assertRaises的簽名：

assertRaises(exception, callable, *args, **kwds)

這是以這種方式定義的，因為如果按照自己的方式執行，Python解釋器首先調用pyport.get_item(test_type, test_id) ，然后將其結果傳遞給assertRaises。 結果是assertRaises根本沒有被調用，並且沒有捕獲到異常。 現在，如果assertRaises可以訪問函數及其參數，它可以調用函數本身並捕獲相應的異常。

Answer 2

如果您使用的是Python 2.7或更高版本， 或者您使用的是backported unittest2模塊而不是unittest，則可以避免在使用assertRaises時將其用作上下文管理器時編寫奇怪的語法：

with self.assertRaises(HTTPError):
    pyport.get_item(test_type, test_id)

http://docs.python.org/library/unittest.html#basic-example