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 _register_review_uis(**kwargs):
"""Registers all bundled review UIs."""
from reviewboard.reviews.ui.base import register_ui
from reviewboard.reviews.ui.image import ImageReviewUI
from reviewboard.reviews.ui.markdownui import MarkdownReviewUI
register_ui(ImageReviewUI)
register_ui(MarkdownReviewUI)
def _register_review_uis(**kwargs):
"""Registers all bundled review UIs."""
from reviewboard.reviews.ui.base import register_ui
from reviewboard.reviews.ui.image import ImageReviewUI
from reviewboard.reviews.ui.markdownui import MarkdownReviewUI
register_ui(ImageReviewUI)
register_ui(MarkdownReviewUI)
def setUp(self):
super(SandboxTests, self).setUp()
register_ui(InitReviewUI)
register_ui(SandboxReviewUI)
register_ui(ConflictFreeReviewUI)
self.factory = RequestFactory()
filename = os.path.join(settings.STATIC_ROOT, 'rb', 'images',
'trophy.png')
with open(filename, 'r') as f:
self.file = SimpleUploadedFile(f.name,
f.read(),
content_type='image/png')
self.user = User.objects.get(username='******')
self.review_request = ReviewRequest.objects.create(self.user, None)
self.file_attachment1 = FileAttachment.objects.create(
mimetype='image/jpg', file=self.file)
self.file_attachment2 = FileAttachment.objects.create(
mimetype='image/png', file=self.file)
self.file_attachment3 = FileAttachment.objects.create(
mimetype='image/gif', file=self.file)
self.review_request.file_attachments.add(self.file_attachment1)
self.review_request.file_attachments.add(self.file_attachment2)
self.review_request.file_attachments.add(self.file_attachment3)
self.draft = ReviewRequestDraft.create(self.review_request)
def setUp(self):
super(SandboxTests, self).setUp()
register_ui(InitReviewUI)
register_ui(SandboxReviewUI)
register_ui(ConflictFreeReviewUI)
self.factory = RequestFactory()
filename = os.path.join(settings.STATIC_ROOT,
'rb', 'images', 'trophy.png')
with open(filename, 'r') as f:
self.file = SimpleUploadedFile(f.name, f.read(),
content_type='image/png')
self.user = User.objects.get(username='******')
self.review_request = ReviewRequest.objects.create(self.user, None)
self.file_attachment1 = FileAttachment.objects.create(
mimetype='image/jpg',
file=self.file)
self.file_attachment2 = FileAttachment.objects.create(
mimetype='image/png',
file=self.file)
self.file_attachment3 = FileAttachment.objects.create(
mimetype='image/gif',
file=self.file)
self.review_request.file_attachments.add(self.file_attachment1)
self.review_request.file_attachments.add(self.file_attachment2)
self.review_request.file_attachments.add(self.file_attachment3)
self.draft = ReviewRequestDraft.create(self.review_request)
def setUp(self):
super(SandboxTests, self).setUp()
register_ui(InitReviewUI)
register_ui(SandboxReviewUI)
register_ui(ConflictFreeReviewUI)
self.factory = RequestFactory()
self.user = User.objects.get(username='******')
self.review_request = self.create_review_request()
self.png_file = self.create_file_attachment(self.review_request)
self.png_file.mimetype = 'image/png'
self.jpg_file = self.create_file_attachment(self.review_request)
self.jpg_file.mimetype = 'image/jpg'
self.gif_file = self.create_file_attachment(self.review_request)
self.gif_file.mimetype = 'image/gif'
def setUp(self):
super(SandboxTests, self).setUp()
register_ui(InitReviewUI)
register_ui(SandboxReviewUI)
register_ui(ConflictFreeReviewUI)
self.factory = RequestFactory()
self.user = User.objects.get(username='******')
self.review_request = self.create_review_request()
self.png_file = self.create_file_attachment(self.review_request)
self.png_file.mimetype = 'image/png'
self.jpg_file = self.create_file_attachment(self.review_request)
self.jpg_file.mimetype = 'image/jpg'
self.gif_file = self.create_file_attachment(self.review_request)
self.gif_file.mimetype = 'image/gif'
def __init__(self, extension, review_uis):
super(ReviewUIHook, self).__init__(extension)
self.review_uis = review_uis
for review_ui in self.review_uis:
register_ui(review_ui)
def __init__(self, extension, review_uis):
super(ReviewUIHook, self).__init__(extension)
self.review_uis = review_uis
for review_ui in self.review_uis:
register_ui(review_ui)
def setUpClass(cls):
super(FileAttachmentReviewUITests, cls).setUpClass()
register_ui(MyReviewUI)
