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 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)
def tearDown(self):
super(SandboxTests, self).tearDown()
unregister_ui(InitReviewUI)
unregister_ui(SandboxReviewUI)
unregister_ui(ConflictFreeReviewUI)
