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)