Python unittest: làm thế nào để kiểm tra đối số trong một ngoại lệ?

Question

Tôi đang thử nghiệm cho trường hợp ngoại lệ sử dụng unittest, ví dụ:

self.assertRaises(UnrecognizedAirportError, func, arg1, arg2) 

và mã của tôi đặt ra:

raise UnrecognizedAirportError('From') 

nào hoạt động tốt.

Làm cách nào để kiểm tra rằng đối số trong ngoại lệ là những gì tôi mong đợi?

Tôi muốn bằng cách nào đó khẳng định rằng capturedException.argument == 'From'.

Tôi hy vọng điều này là đủ rõ ràng - cảm ơn trước!

Tal.

Answer 1

Như thế này.

>>> try: 
...  raise UnrecognizedAirportError("func","arg1","arg2") 
... except UnrecognizedAirportError, e: 
...  print e.args 
... 
('func', 'arg1', 'arg2') 
>>> 

Đối số của bạn ở args, nếu bạn chỉ cần phân loại Exception.

Xem http://docs.python.org/library/exceptions.html#module-exceptions

Nếu lớp ngoại lệ được bắt nguồn từ tiêu chuẩn BaseException lớp rễ, giá trị liên quan đến hiện diện như là thuộc tính args trường hợp ngoại lệ của.

---

Sửa Ví dụ Bigger.

class TestSomeException(unittest.TestCase): 
    def testRaiseWithArgs(self): 
     try: 
      ... Something that raises the exception ... 
      self.fail("Didn't raise the exception") 
     except UnrecognizedAirportError, e: 
      self.assertEquals("func", e.args[0]) 
      self.assertEquals("arg1", e.args[1]) 
     except Exception, e: 
      self.fail("Raised the wrong exception") 

Answer 2

assertRaises là một chút đơn giản và không cho phép bạn kiểm tra chi tiết về ngoại lệ đã vượt ra ngoài nó thuộc về một lớp cụ thể. Để kiểm tra các ngoại lệ tốt hơn, bạn cần phải "cuộn riêng" với khối try/except/else (bạn có thể làm điều đó một lần và cho tất cả trong một phương thức def assertDetailedRaises mà bạn thêm vào lớp con chung của trường hợp thử nghiệm của unittest, sau đó có bài kiểm tra của bạn trường hợp tất cả kế thừa lớp con của bạn thay vì của unittest).