# HG changeset patch # User neko259 # Date 2013-03-23 20:43:00 # Node ID b62ef70fbdc5a568210fe9da911d77e576acd98b # Parent bb33ad66a71396fc8cdccfc21cfc5185a0c38c39 Fixed the tests. Added exists() method for the post. diff --git a/boards/models.py b/boards/models.py --- a/boards/models.py +++ b/boards/models.py @@ -45,6 +45,11 @@ class PostManager(models.Manager): return [opening_post, replies] + def exists(self, post_id): + posts = Post.objects.filter(id = post_id) + + return len(posts) == 0 + class Tag(models.Model): """ @@ -54,6 +59,7 @@ class Tag(models.Model): name = models.CharField(max_length = 100) + class Post(models.Model): """A post is a message.""" diff --git a/boards/tests.py b/boards/tests.py --- a/boards/tests.py +++ b/boards/tests.py @@ -3,6 +3,7 @@ import boards from boards.models import Post + class BoardTests(TestCase): def create_post(self): @@ -17,14 +18,16 @@ class BoardTests(TestCase): def test_delete_post(self): post = self.create_post() + post_id = post.id Post.objects.delete_post(post) - self.assertIsNone(post) + self.assertTrue(Post.objects.exists(post_id)) def test_delete_posts_by_ip(self): post = self.create_post() + post_id = post.id Post.objects.delete_posts_by_ip('0.0.0.0') - self.assertIsNone(post) + self.assertTrue(Post.objects.exists(post_id))