diff options
-rw-r--r-- | yaksh/test_views.py | 28 |
1 files changed, 14 insertions, 14 deletions
diff --git a/yaksh/test_views.py b/yaksh/test_views.py index 5cbedc6..51c17f3 100644 --- a/yaksh/test_views.py +++ b/yaksh/test_views.py @@ -6351,6 +6351,7 @@ class TestPost(TestCase): username=self.student.username, password=self.student_plaintext_pass ) + self.course.students.add(self.student) response = self.client.get(reverse('yaksh:course_forum', kwargs={ 'course_id': self.course.id }), follow=True) @@ -6362,19 +6363,18 @@ class TestPost(TestCase): username=self.student.username, password=self.student_plaintext_pass ) + self.course.students.add(self.student) response = self.client.post(reverse('yaksh:course_forum', kwargs={ 'course_id': self.course.id - }), { - 'title': 'post 1', - 'description': 'post 1 description', - 'course': self.course, - 'creator': self.student + }), data={ + "title": 'Post 1', + "description": 'Post 1 description', }) self.assertEquals(response.status_code, 302) - url = response.url.split('/') - uid = url[5] - test_against = Post.objects.get(uid=uid) - self.assertEqual(test_against.title, 'post 1') + result = Post.objects.filter(title='Post 1', + creator=self.student, + course=self.course) + self.assertTrue(result.exists()) def test_open_created_post_denies_anonymous_user(self): post = Post.objects.create( @@ -6397,6 +6397,7 @@ class TestPost(TestCase): username=self.student.username, password=self.student_plaintext_pass ) + self.course.students.add(self.student) post = Post.objects.create( title='post 1', description='post 1 description', @@ -6475,6 +6476,7 @@ class TestPostComment(TestCase): username=self.student.username, password=self.student_plaintext_pass ) + self.course.students.add(self.student) response = self.client.post(reverse('yaksh:post_comments', kwargs={ 'course_id': self.course.id, 'uuid': self.post.uid @@ -6484,17 +6486,15 @@ class TestPostComment(TestCase): 'creator': self.user, }) self.assertEquals(response.status_code, 302) - url = response.url.split('/') - uid = url[5] - test_against = Comment.objects.filter(post_field__uid=uid) - comment = test_against[0] - self.assertEqual(comment.post_field, self.post) + result = Comment.objects.filter(post_field__uid=self.post.uid) + self.assertTrue(result.exists()) def test_hide_post_comment(self): self.client.login( username=self.student.username, password=self.student_plaintext_pass ) + self.course.students.add(self.student) comment = Comment.objects.create( post_field=self.post, description='post 1 comment', |