python - 使用 assertRaise 测试异常消息

Question

我正在尝试在条件引发自定义异常消息的函数内 assertRaise 异常。

功能:

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

现在，我能够成功地产生到达 raise 语句所需的条件。

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

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

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

所以我的问题是，如何引发此类异常？如何在我们的单元测试中涵盖这些情况？我不想逃避这些条件，因为它们很重要并且会增加代码覆盖率？

谢谢大家

Answer 1

您的代码直接调用了 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 将调用代码并正确处理异常。