def test_remove(self): ct = ContentType.objects.get_for_model(self.user1) pk = self.user1.pk c = api.post_comment(content_type_id=ct.id, object_pk=pk, user_id=pk, comment="Root message") self.assertTrue(c is not None) c = api.remove_comment(c.id, self.user1) self.assertTrue(c is not None) self.assertTrue(c.is_removed) removed = api.get_comments_removed( content_type_id=ct.id, object_pk=pk) self.assertEqual(len(removed), 1) # restore_comment with wrong user x = api.restore_comment(c.id, self.user2) self.assertTrue(x is None) c = api.restore_comment(c.id, self.user1) self.assertTrue(c is not None) self.assertFalse(c.is_removed) # non-exising comments should return None x = api.restore_comment(-1, self.user1) self.assertEqual(x, None) x = api.remove_comment(-1, self.user1) self.assertEqual(x, None) # remove_comment with wrong user orgid = c.id c = api.remove_comment(orgid, self.user2) self.assertTrue(c is None) c = api.get_comment(orgid) self.assertFalse(c.is_removed) removed = api.get_comments_removed( content_type_id=ct.id, object_pk=pk) self.assertEqual(len(removed), 0)
def test_remove(self): ct = ContentType.objects.get_for_model(self.user1) pk = self.user1.pk c = api.post_comment(content_type_id=ct.id, object_pk=pk, user_id=pk, comment="Root message") self.assertTrue(c is not None) c = api.remove_comment(c.id, self.user1) self.assertTrue(c is not None) self.assertTrue(c.is_removed) removed = api.get_comments_removed(content_type_id=ct.id, object_pk=pk) self.assertEqual(len(removed), 1) # restore_comment with wrong user x = api.restore_comment(c.id, self.user2) self.assertTrue(x is None) c = api.restore_comment(c.id, self.user1) self.assertTrue(c is not None) self.assertFalse(c.is_removed) # non-exising comments should return None x = api.restore_comment(-1, self.user1) self.assertEqual(x, None) x = api.remove_comment(-1, self.user1) self.assertEqual(x, None) # remove_comment with wrong user orgid = c.id c = api.remove_comment(orgid, self.user2) self.assertTrue(c is None) c = api.get_comment(orgid) self.assertFalse(c.is_removed) removed = api.get_comments_removed(content_type_id=ct.id, object_pk=pk) self.assertEqual(len(removed), 0)
def restore(request, comment_id): comment = api.restore_comment(comment_id, request.user) if comment: return HttpResponseRedirect(comment.get_absolute_url()) raise Http404()