使用assertRaise测试异常消息

时间:2012-07-07 02:40:10

标签: python unit-testing

我试图在一个条件引发自定义异常消息的函数内断言异常。

功能:

if not Cart.objects.filter(member=member).count():
    raise CartDoesNotExist("Cart Does Not Exist for Member: %s ( %id )." % (member.email,member.id))

现在,我能够成功地产生获得加注声明所需的条件。

所以,我的测试用例看起来像这样:

def Order_CartDoesNotExist(self):        
  self.assertRaises(CartDoesNotExist,Order.objects.create_order(member=self.member2,member_slot=self.memslot,order_type="Normal"))

当我运行测试时,输出是错误。它给出了相同的错误CartDoesNotExist .....

所以我的问题是,如何提出这些例外?如何在我们的单元测试中涵盖这些情况?我不想逃避这些条件,因为它们很重要并且增加了代码覆盖率?

谢谢大家。

1 个答案:

答案 0 :(得分:4)

您的代码直接调用create_order,这会引发异常。你需要改变它的调用方式。在Python 2.7中,您可以使用:

with self.assertRaises(CartDoesNotExist):
    Order.objects.create_order(member=self.member2, member_slot=self.memslot, order_type="Normal"))

此处上下文管理器允许您直接调用代码,上下文管理器将为您处理异常。

如果您使用2.6或更低版本运行:

self.assertRaises(CartDoesNotExist, Order.objects.create_order, member=self.member2, member_slot=self.memslot, order_type="Normal")

这里你没有调用你的函数,你将它传递给assertRaises,以及它需要的参数,assertRaises将调用代码并正确处理异常。