CODEWITHSUNDEEP
pythonunit-testing
Arindam Roychowdhury
Arindam Roychowdhury

Reputation: 6511

Testing exception message with assertRaise

I am trying to assertRaise the exception inside a function where a condition raises a custom exception message .

Function:

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

Now , i am able to successfully produce the required condition to get to the raise statement.

So , my testcase looks like this :

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

When i run the test , the output is an Error . It gives the same error CartDoesNotExist.....

So my question is , how to raise these kind of exceptions ? How to cover these situations in our unittest? I do not want to escape these conditions as they are important and increase code coverage?

Thank you all.

Upvotes: 4

Views: 5828

Answers (1)

Ned Batchelder
Ned Batchelder

Reputation: 375584

Your code is calling create_order directly, which raises the exception. You need to change how it is called. In Python 2.7, you can use this:

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

Here the context manager allows you to call your code directly, and the context manager will handle the exception for you.

If you are running with 2.6 or below:

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

Here you aren't calling your function, you are passing it to assertRaises, along with the arguments it needs, and assertRaises will call the code and deal with the exception properly.

Upvotes: 4

Related Questions