def test_for_type_with_exception(self): """Testing FileAttachmentReviewUI.for_type sandboxes ReviewUI instantiation """ class BrokenReviewUI(FileAttachmentReviewUI): supported_mimetypes = ['image/broken'] def __init__(self, *args, **kwargs): raise Exception('Oh no') self.spy_on(BrokenReviewUI.__init__, owner=BrokenReviewUI) register_ui(BrokenReviewUI) try: attachment = self.create_file_attachment(self.review_request, mimetype='image/broken') review_ui = FileAttachmentReviewUI.for_type(attachment) self.assertIsNone(review_ui) self.assertTrue( BrokenReviewUI.__init__.called_with(self.review_request, attachment)) finally: unregister_ui(BrokenReviewUI)
def tearDown(self): super(SandboxTests, self).tearDown() unregister_ui(InitReviewUI) unregister_ui(SandboxReviewUI) unregister_ui(ConflictFreeReviewUI)
def shutdown(self): super(ReviewUIHook, self).shutdown() for review_ui in self.review_uis: unregister_ui(review_ui)
def tearDownClass(cls): super(FileAttachmentReviewUITests, cls).tearDownClass() unregister_ui(MyReviewUI)