def test_basic(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)
comments = api.get_comments(ct.id, pk)
self.assertEqual(len(comments), 1)
comment = api.get_comment(c.id)
self.assertTrue(comment is not None)
self.assertEqual(comment.id, c.id)
comment = api.get_comment(-1)
self.assertTrue(comment is None)
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)
