如何使用Python的unittest來聲明數據庫錯誤？

Question

我正在運行Python 2.6.5，因此我使用unittest2模塊，它是2.7和3.X中unittest模塊的未來端口。我正在執行以下集成測試：

def test_hits_constraint_raise(self): 
    obj = Table1(...) 
    self.sess.add(obj) 
    self.sess.flush() 

    # assert condition raises 
    self.assertFailure(IntegrityError, sess.add, obj) 

取而代之，它遇到了錯誤。

====================================================================== 
ERROR: test_hits_constraint_raise (__main__.TestModels) 
---------------------------------------------------------------------- 
Traceback (most recent call last): 
    File "test_models.py", line 76, in test4_create_exercise_same_id_constraint_raise 
    self.sess.flush() 
    File "build/bdist.linux-i686/egg/sqlalchemy/orm/scoping.py", line 114, in do 
    return getattr(self.registry(), name)(*args, **kwargs) 
     ......... 
    File "build/bdist.linux-i686/egg/sqlalchemy/engine/default.py", line 331, in do_execute 
    cursor.execute(statement, parameters) 
IntegrityError: (IntegrityError) column exercise_id is not unique u'INSERT INTO TABLE1 (attribute1_name, attribute2_name) VALUES (?, ?)' ('value1', 'value2') 

---------------------------------------------------------------------- 
Ran 4 tests in 0.531s 

FAILED (errors=1) 

唯一的辦法是使用@unittest2.expectedFailure？那個會運行良好。 但我想知道到底發生了什麼樣的錯誤。

---------------------------------------------------------------------- 
Ran 4 tests in 0.513s 

OK (expected failures=1) 

Answer 1

您可以使用assertRaises爲with聲明背景：

with unittest.assertRaises(IntegrityError): 
    sess.add(obj) 

More details。