Show More
@@ -127,7 +127,7 b' class Thread(models.Model):' | |||
|
127 | 127 | |
|
128 | 128 | query = self.replies.order_by('pub_time').prefetch_related('images') |
|
129 | 129 | if view_fields_only: |
|
130 |
query = query.defer('poster_user_agent' |
|
|
130 | query = query.defer('poster_user_agent') | |
|
131 | 131 | return query.all() |
|
132 | 132 | |
|
133 | 133 | def get_replies_with_images(self, view_fields_only=False): |
@@ -112,4 +112,25 b' class PostTests(TestCase):' | |||
|
112 | 112 | first_post = posts_in_second_page[0] |
|
113 | 113 | |
|
114 | 114 | self.assertEqual(all_threads[settings.THREADS_PER_PAGE].id, |
|
115 | first_post.id) No newline at end of file | |
|
115 | first_post.id) | |
|
116 | ||
|
117 | def test_thread_replies(self): | |
|
118 | """ | |
|
119 | Tests that the replies can be queried from a thread in all possible | |
|
120 | ways. | |
|
121 | """ | |
|
122 | ||
|
123 | tag = Tag.objects.create(name='test_tag') | |
|
124 | opening_post = Post.objects.create_post(title='title', text='text', | |
|
125 | tags=[tag]) | |
|
126 | thread = opening_post.get_thread() | |
|
127 | ||
|
128 | reply1 = Post.objects.create_post(title='title', text='text', thread=thread) | |
|
129 | reply2 = Post.objects.create_post(title='title', text='text', thread=thread) | |
|
130 | ||
|
131 | replies = thread.get_replies() | |
|
132 | self.assertTrue(len(replies) > 0, 'No replies found for thread.') | |
|
133 | ||
|
134 | replies = thread.get_replies(view_fields_only=True) | |
|
135 | self.assertTrue(len(replies) > 0, | |
|
136 | 'No replies found for thread with view fields only.') |
General Comments 0
You need to be logged in to leave comments.
Login now