from django.test import TestCase import boards from boards.models import Post class BoardTests(TestCase): def create_post(self): return Post.objects.create_post(title = 'title', text = 'text') def test_post_add(self): post = self.create_post() self.assertIsNotNone(post) self.assertEqual(boards.models.NO_PARENT, post.parent) def test_delete_post(self): post = self.create_post() Post.objects.delete_post(post) self.assertIsNone(post) def test_delete_posts_by_ip(self): post = self.create_post() Post.objects.delete_posts_by_ip('0.0.0.0') self.assertIsNone(post)