使用 assertRaise 测试异常消息

Question

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

功能：

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