def test_steal_attachments(self): """middleware validates if attachments are already assigned to other posts""" other_post = testutils.reply_thread(self.thread) attachments = [ self.mock_attachment(post=other_post), self.mock_attachment(), ] middleware = AttachmentsMiddleware( request=RequestMock( {'attachments': [attachments[0].pk, attachments[1].pk]}), mode=PostingEndpoint.EDIT, user=self.user, post=self.post, ) serializer = middleware.get_serializer() self.assertTrue(serializer.is_valid()) middleware.save(serializer) # only unassociated attachment was associated with post self.assertEqual(self.post.update_fields, ['attachments_cache']) self.assertEqual(self.post.attachment_set.count(), 1) self.assertEqual( Attachment.objects.get(pk=attachments[0].pk).post, other_post) self.assertEqual( Attachment.objects.get(pk=attachments[1].pk).post, self.post)
def test_edit_attachments(self): """middleware removes and adds attachments to post""" attachments = [ self.mock_attachment(post=self.post), self.mock_attachment(post=self.post), self.mock_attachment(), ] middleware = AttachmentsMiddleware( request=RequestMock( {'attachments': [attachments[0].pk, attachments[2].pk]}), mode=PostingEndpoint.EDIT, user=self.user, post=self.post, ) serializer = middleware.get_serializer() self.assertTrue(serializer.is_valid()) middleware.save(serializer) # attachments were associated with post self.assertEqual(self.post.update_fields, ['attachments_cache']) self.assertEqual(self.post.attachment_set.count(), 2) attachments_filenames = [ attachments[2].filename, attachments[0].filename ] self.assertEqual([a['filename'] for a in self.post.attachments_cache], attachments_filenames)
def test_add_attachments(self): """middleware adds attachments to post""" attachments = [ self.mock_attachment(), self.mock_attachment(), ] middleware = AttachmentsMiddleware(request=RequestMock( {'attachments': [a.pk for a in attachments]}), mode=PostingEndpoint.EDIT, user=self.user, post=self.post) serializer = middleware.get_serializer() self.assertTrue(serializer.is_valid()) middleware.save(serializer) # attachments were associated with post self.assertEqual(self.post.update_fields, ['attachments_cache']) self.assertEqual(self.post.attachment_set.count(), 2) attachments_filenames = list( reversed([a.filename for a in attachments])) self.assertEqual([a['filename'] for a in self.post.attachments_cache], attachments_filenames)
def test_get_new_attachments(self): """get_initial_attachments returns list of attachments already existing on post""" middleware = AttachmentsMiddleware( request=RequestMock(), mode=PostingEndpoint.EDIT, user=self.user, post=self.post, ) serializer = middleware.get_serializer() attachments = serializer.get_new_attachments(middleware.user, [1, 2, 3]) self.assertEqual(attachments, []) attachment = self.mock_attachment() attachments = serializer.get_new_attachments(middleware.user, [attachment.pk]) self.assertEqual(attachments, [attachment]) # only own orphaned attachments may be assigned to posts other_user_attachment = self.mock_attachment(user=False) attachments = serializer.get_new_attachments( middleware.user, [other_user_attachment.pk]) self.assertEqual(attachments, [])
def test_use_this_middleware(self): """use_this_middleware returns False if we can't upload attachments""" middleware = AttachmentsMiddleware(user=self.user) self.override_acl({'max_attachment_size': 0}) self.assertFalse(middleware.use_this_middleware()) self.override_acl({'max_attachment_size': 1024}) self.assertTrue(middleware.use_this_middleware())
def test_middleware_is_optional(self): """middleware is optional""" INPUTS = ({}, {'attachments': []}) for test_input in INPUTS: middleware = AttachmentsMiddleware(request=RequestMock(test_input), mode=PostingEndpoint.START, user=self.user, post=self.post) serializer = middleware.get_serializer() self.assertTrue(serializer.is_valid())
def get_middleware_for_testing(**kwargs): mock_kwargs = { 'prefix': 'test', 'mode': 0, 'request': None, 'user': None, 'datetime': None, 'parsing_result': None, 'thread': None, 'post': None, } mock_kwargs.update(kwargs) return AttachmentsMiddleware(**mock_kwargs)
def test_middleware_validates_ids(self): """middleware validates attachments ids""" INPUTS = ('none', ['a', 'b', 123], range(settings.MISAGO_POST_ATTACHMENTS_LIMIT + 1)) for test_input in INPUTS: middleware = AttachmentsMiddleware(request=RequestMock( {'attachments': test_input}), mode=PostingEndpoint.START, user=self.user, post=self.post) serializer = middleware.get_serializer() self.assertFalse(serializer.is_valid(), "%r shouldn't validate" % test_input)
def test_get_initial_attachments(self): """get_initial_attachments returns list of attachments already existing on post""" middleware = AttachmentsMiddleware(request=RequestMock(), mode=PostingEndpoint.EDIT, user=self.user, post=self.post) serializer = middleware.get_serializer() attachments = serializer.get_initial_attachments( middleware.mode, middleware.user, middleware.post) self.assertEqual(attachments, []) attachment = self.mock_attachment(post=self.post) attachments = serializer.get_initial_attachments( middleware.mode, middleware.user, middleware.post) self.assertEqual(attachments, [attachment])
def test_cant_delete_attachment(self): """middleware validates if we have permission to delete other users attachments""" self.override_acl({ 'max_attachment_size': 1024, 'can_delete_other_users_attachments': False }) attachment = self.mock_attachment(user=False, post=self.post) self.assertIsNone(attachment.uploader) serializer = AttachmentsMiddleware(request=RequestMock( {'attachments': []}), mode=PostingEndpoint.EDIT, user=self.user, post=self.post).get_serializer() self.assertFalse(serializer.is_valid())